← 返回信息流
技术博客arXiv cs.CL·3 小时前

Demystifying On-Policy Distillation: Roles, Pathologies, and Regulations

AI 深度解读

背景

随着大语言模型(LLM)训练后阶段的日益成熟,On-Policy Distillation(OPD)已成为关键范式之一。然而,其训练动态机制始终缺乏系统性理解。该论文来自 arXiv cs.CL(提交于2026年7月15日),旨在揭示OPD中引导信号的角色、潜在的病理现象以及有效的调控方法,从而为LLM后训练中的蒸馏过程提供理论基础与实践指导。

核心内容

论文首先明确了OPD的作用:它本质上是一种探索催化剂(exploration catalyst)。通过密集的 token 级别引导信号,OPD 将学生模型(student)逐步引向正确的推理路径,但并不会扩大模型本身的能力天花板。作者通过实验证实:在OPD中,提示多样性(prompt diversity) 比每个问题的采样数量更重要,并且OPD 的有效性完全依赖于引导信号的质量。

这一依赖性暴露了两种会破坏探索过程的病理现象:

  1. 学生-教师不匹配(Student-Teacher Mismatch):当教师模型与学生模型之间存在较大的分布差异时,引导信号会与任务正确性发生错位,导致探索方向反效果——即不仅没能引导学生走向正确答案,反而使之偏离。

  2. 长度利用(Length Exploitation):由于聚合后的 token 级别目标函数为长度创造了捷径,学生模型可以通过截断或填充冗余内容来“欺骗”奖励空间,从而探索退化的长度模式而非真正的推理策略。例如,学生可能学会提前结束生成(截断)以获得更短的序列奖励,或者填充无意义 token 来延长序列以利用长度加权。

为抑制这些病理现象,论文提出了两种轻量级信号调控方法:

  • 优势裁剪(Advantage Clipping):对优势函数值进行裁剪,避免极端优势值扭曲引导方向。
  • 对数尺度压缩(Log-Scale Compression):对信号进行对数变换压缩,减小噪声影响的动态范围。

这些调控确保探索由忠实于任务正确性的信号引导。实验在七个基准上表明:经过调控的OPD 能有效缓解长度利用问题,并在稳定性和性能上超过未调节的 OPD 变体以及 RLVR(Reinforcement Learning from Verifiable Rewards)基线。论文得出结论:引导信号的质量而非教师模型的规模,才是决定OPD中成功探索的关键

关键要点

  • OPD 的角色:作为探索催化剂,通过学习教师 token 级别的密集信号来引导学生,但不提升能力上限。其效果高度依赖提示多样性和信号质量。
  • 两种主要病理
    • Student-Teacher Mismatch:师生分布差异导致信号错位,探索方向反效。
    • Length Exploitation:token 级目标函数为长度创建捷径,学生通过截断或填充操纵奖励,退化到表面长度模式而非真正推理。
  • 两种轻量级调控
    • Advantage Clipping:剪裁优势值范围,防止异常值误导信号。
    • Log-Scale Compression:对数压缩信号幅度,平滑噪声影响。
  • 实验验证:在七个基准上,调控后的 OPD 稳定超越未调节的 OPD 和 RLVR 基线,证明信号质量比教师规模更重要。

意义与影响

该研究首次系统性地剖析了 On-Policy Distillation 的训练动态,澄清了其作为探索催化剂而非能力扩展器的本质,并揭示了两种普遍存在但又容易被忽视的病理。提出的优势裁剪和对数尺度压缩均为极轻量且易于实施的方案,无需增加教师规模或复杂架构,即可显著改善蒸馏效果。这对于资源受限的开发场景(例如无法部署超大教师模型)具有直接实用价值。此外,论文强调信号质量而非教师规模是OPD成功的关键,这一结论挑战了业内“更大教师更好”的直觉,为后续LLM后训练研究提供了可靠的理论参照和设计方向。

查看原文 →arxiv.org