Gains and Collapse in On-Policy Distillation:A Reinforcement Learning Perspective
Abstract
On-policy distillation (OPD) has become an important approach to language model post-training. However, despite its performance gains, OPD can also collapse into excessively long and repetitive generation, and the mechanism underlying these divergent outcomes remains poorly understood. We explain these outcomes through a reinforcement learning perspective: the teacher implicitly rewards student behaviors, even those it rarely exhibits itself. From this perspective, our experiments show that OPD improves performance without expanding the student's capabilities. When the implicit reward model is reliable, OPD makes correct responses easier to sample. In contrast, when the preference misaligns with quality, reward hacking happens: the implicit reward model amplifies overlong, repetitive student rollouts, even though it rarely generates such text itself. Guided by this diagnosis, we find that masking unhealthy responses during training and using SFT initialization can each effectively mitigate the collapse. Together, these findings show that OPD amplifies student behaviors favored by the teacher's implicit feedback, shifting the focus from how well the teacher generates to how reliably it evaluates student rollouts. Our code is available at https://github.com/HancCui/opd_hacking.
Community
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- When Teacher Guidance Misleads: Reward-Aligned On-Policy Distillation (2026)
- Towards Understanding On-Policy Distillation through the Lens of Test-Time Scaling (2026)
- Understanding the Synergy between SFT, RLVR, and OPD in LLM Post-Training (2026)
- Reward-Aligned Reweighting for On-Policy Distillation (2026)
- When EOS Tokens Disagree: Understanding Length Inflation in On-Policy Distillation (2026)
- Recursive Self-Improvement via On-Policy Distillation for Reasoning (2026)
- Teaching a Moving Student: Rethinking the Curriculum of On-Policy Distillation (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper