arXiv:2605.16826cs.LGcs.AI2026-05被引 8

解耦大模型蒸馏中的KL与轨迹选择,统一解释多种训练范式。

Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation

论文配图:Decoupling KL and Trajectories: A Unified Perspective for SFT, DAgger, Offline RL, and OPD in LLM Distillation
图 1 · 摘自论文原文
  • 拆解序列KL为前向/反向与教师/学生前缀的组合,揭示四种新目标
  • 实验证明前向KL提升准确率但降低多样性,反向KL增强鲁棒性
  • 提出混合KL与熵门控长度策略,显著提升生成质量并压缩长度

知识蒸馏在大模型后训练中至关重要,但其设计空间仍不清晰,尤其与强化学习(RL)结合时。我们发现主流范式——离线蒸馏与在线蒸馏(OPD)——隐式耦合了两个独立选择:前缀来源与词级KL方向。通过分解自回归响应分布的序列级KL,可得:前向KL将教师前缀与词级前向KL配对,反向KL将学生前缀与词级反向KL配对。我们主张这种耦合非本质:解耦两轴可得四个有效目标。梯度层面的身份关系表明,前向KL等价于带有教师软标签的SFT交叉熵匹配,反向KL则等价于基于教师-学生对数似然比的密集奖励的强化学习策略梯度,从而连接到离线SFT、DAgger式在线SFT、离线强化学习型蒸馏和OPD。我们在数学推理任务上开展全面受控实验,评估四种目标作为独立方法或后续强化学习的初始化。结果揭示三个权衡:KL方向带来准确率-熵权衡,前缀来源引出质量-算力权衡,训练长度导致准确率-稳定性权衡。基于此,我们提出KL混合与熵门控长度课程。KL混合显示长序列蒸馏需较大前向KL权重以防止熵坍缩与长度膨胀而不损失准确率。熵门控长度课程使平均@k和通过率@k分别提升3.6和最高5.8点,且平均响应长度减少约3倍,优于固定长视野训练。

原文摘要 · Abstract (English)

Knowledge distillation is central to LLM post-training, yet its design space remains poorly understood, especially alongside reinforcement learning (RL). We show that the prevailing paradigms, off-policy distillation and on-policy distillation (OPD), implicitly couple two orthogonal choices: prefix source and token-level KL direction. This follows from decomposing sequence-level KL over autoregressive response distributions: forward KL pairs teacher prefixes with token-level forward KL, and reverse KL pairs student prefixes with token-level reverse KL. We argue this coupling is not intrinsic: decoupling the two axes yields four valid objectives. We establish gradient-level identities showing forward KL gives SFT-style cross-entropy matching with teacher soft targets, whereas reverse KL gives an RL-style policy-gradient objective with a dense teacher-student log-ratio reward, connecting them to off-policy SFT, DAgger-style on-policy SFT, offline-RL-style distillation, and OPD. We conduct an extensive controlled study on math reasoning, evaluating the four objectives both as standalone methods and as initializations for subsequent RL. The results reveal three tradeoffs: KL direction induces an accuracy-entropy tradeoff, prefix source a quality-compute tradeoff, and training length an accuracy-stability tradeoff. Motivated by these findings, we propose KL mixing and an entropy-gated length curriculum. KL mixing shows long-sequence distillation requires substantial forward-KL weight to prevent entropy collapse and length inflation without sacrificing accuracy. The entropy-gated length curriculum improves Avg@k and Pass@k by 3.6 and up to 5.8 points, and cuts average response length by roughly 3x versus fixed long-horizon training. Our results provide a framework and practical methods for designing reasoning distillation objectives that balance accuracy, diversity, compute, and RL behavior.

大模型蒸馏强化学习知识蒸馏序列建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。