Rethinking On-Policy Distillation of Large Language Models: Phenomenology, Mechanism, and Recipe

AK's Picks LLM SFT Other LLM
2026年04月14日
在线策略蒸馏(OPD)已成为大语言模型后训练阶段的一项核心技术,但其训练动力学机制迄今仍缺乏深入理解。本文对OPD的动力学行为与内在机制开展了系统性研究。我们首先发现,OPD能否成功取决于两个关键条件:(i)学生模型与教师模型需具备相容的思维模式;(ii)即便二者思维模式一致且教师模型在各项指标上得分更高,该教师仍须提供学生在原始训练过程中未曾接触过的、真正新颖的能力。我们通过“由弱至强”的逆向蒸馏实验验证了上述结论:同一家族的1.5B与7B参数量教师模型,在学生模型视角下其输出分布彼此不可区分。进一步深入到词元(token)层面的机制分析,我们发现,成功的OPD过程呈现出一种渐进式对齐现象——即在学生模型曾访问过的状态上,双方对高概率词元的预测逐步趋于一致;而这种对齐集中于一个规模较小、却承载绝大部分概率质量(97%–99%)的共享词元集合。在此基础上,我们提出了两种切实可行的策略,以挽救失败的OPD过程:一是采用离线策略的“冷启动”(off-policy cold start),二是基于教师模型输出特征进行提示词(prompt)筛选(teacher-aligned prompt selection)。最后,我们指出:OPD看似以密集的词元级奖励形式提供了“免费午餐”,实则暗含代价——这进而引发一个根本性问题:OPD是否具备可扩展性,能否适用于长视野(long-horizon)场景下的蒸馏任务?
On-policy distillation (OPD) has become a core technique in the post-training of large language models, yet its training dynamics remain poorly understood. This paper provides a systematic investigation of OPD dynamics and mechanisms. We first identify that two conditions govern whether OPD succeeds or fails: (i) the student and teacher should share compatible thinking patterns; and (ii) even with consistent thinking patterns and higher scores, the teacher must offer genuinely new capabilities beyond what the student has seen during training. We validate these findings through weak-to-strong reverse distillation, showing that same-family 1.5B and 7B teachers are distributionally indistinguishable from the student's perspective. Probing into the token-level mechanism, we show that successful OPD is characterized by progressive alignment on high-probability tokens at student-visited states, a small shared token set that concentrates most of the probability mass (97%-99%). We further propose two practical strategies to recover failing OPD: off-policy cold start and teacher-aligned prompt selection. Finally, we show that OPD's apparent free lunch of dense token-level reward comes at a cost, raising the question of whether OPD can scale to long-horizon distillation.