Mechanism

Decouples offline data generation (Grow step: sample a large dataset of trajectories from the current policy πθg\pi_{\theta_g}) from iterative offline policy optimization (Improve step: filter the grown dataset with an increasing sequence of reward thresholds τ1<τ2<⋯<τM\tau_1 < \tau_2 < \dots < \tau_M and train for multiple offline loss steps on each thresholded subset).

References

  1. Reinforced Self-Training (ReST) for Language Modeling ↗

    Caglar Gulcehre, Tom Le Paine, Srivatsan Srinivasan, Ksenia Konyushkova, Lotte Weerts, Abhishek Sharma et al. · 2023

    All 14 authors

    Caglar Gulcehre, Tom Le Paine, Srivatsan Srinivasan, Ksenia Konyushkova, Lotte Weerts, Abhishek Sharma, Aditya Siddhant, Alex Ahern, Miaosen Wang, Chenjie Gu, Wolfgang Macherey, Arnaud Doucet, Orhan Firat, Nando de Freitas