Mechanism
Proves that offline SFT or naive single-turn RL on self-correction tasks fails due to distribution shift and mode collapse (the model either outputs the final answer on Turn 1 and learns nothing about Turn 2 correction, or intentionally sabotages Turn 1 to earn a high improvement delta on Turn 2). SCoRe solves this via a two-stage on-policy RL pipeline:
- Stage I: Train Turn 2 to maximize while strongly KL-anchoring Turn 1 to so the model cannot sabotage Turn 1.
- Stage II: Jointly optimize with a progress-shaping bonus.
References
- Training Language Models to Self-Correct via Reinforcement Learning ↗
All 18 authors
Aviral Kumar, Vincent Zhuang, Rishabh Agarwal, Yi Su, John D Co-Reyes, Avi Singh, Kate Baumli, Shariq Iqbal, Colton Bishop, Rebecca Roelofs, Lei M Zhang, Kay McKinney, Disha Shrivastava, Cosmin Paduraru, George Tucker, Doina Precup, Feryal Behbahani, Aleksandra Faust