揭示大模型在线策略蒸馏的三大失效机制并提出改进方法
The Many Faces of On-Policy Distillation: Pitfalls, Mechanisms, and Fixes

- 通过实证分析发现教师选择与损失设计影响数学推理性能
- 当教师信息为实例专属时,自蒸馏会因分布错配导致效果下降
- 提出梯度截断、强化学习适配教师等方案提升稳定性
在线策略蒸馏(OPD)和自蒸馏(OPSD)作为大语言模型的后训练方法,能提供来自模型自身策略采样轨迹的密集词元级监督。然而现有研究结果不一:虽在系统提示和知识内化方面表现良好,近期也报告存在不稳定与性能退化。本文开展全面实证研究,揭示OPD与OPSD在何时有效、何时失败及其原因。发现数学推理任务中OPD对教师选择与损失形式高度敏感;而测试时缺乏实例专属特权信息(PI)导致OPSD在多数场景失效。但当PI为共享隐含规则(如系统提示或对齐偏好)时,OPSD有效。识别出三个失败机制:(1) 学生生成前缀导致师生分布错配;(2) 偏向TopK反KL梯度引发优化不稳定;(3) 自蒸馏特有缺陷——学生学得无PI的聚合策略,无法应对实例专属信息。进一步验证停止梯度的TopK目标、适应RLVR的教师、以及经SFT稳定的学生可缓解上述问题。
原文摘要 · Abstract (English)
On-policy distillation (OPD) and on-policy self-distillation (OPSD) have emerged as promising post-training methods for large language models, offering dense token-level supervision on trajectories sampled from the model's own policy. However, existing results on their effectiveness remain mixed: while OP(S)D has shown promise in system prompt and knowledge internalization, recent studies also report instability and degradation. In this work, we present a comprehensive empirical study of when OPD and OPSD work, when they fail, and why. We find that OPD on mathematical reasoning is highly sensitive to teacher choice and loss formulation, whereas OPSD fails in our tested settings due to test-time absence of instance-specific privileged information (PI). In contrast, OPSD is effective when PI represents a shared latent rule, such as a system prompt or alignment preference. We identify three failure mechanisms: (1) distribution mismatch between teacher and student caused by conditioning on student-generated prefixes, (2) optimization instability from biased TopK reverse-KL gradients, and (3) an OPSD-specific limitation where the student learns a PI-free policy that aggregates PI-conditioned teachers, which is insufficient when PI is instance-specific. We further show that stop-gradient TopK objectives, RLVR-adapted teachers, and SFT-stabilized students mitigate these failures.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。