arXiv:2607.05184cs.AIcs.LG2026-07被引 11

强思维模型用特权信息自蒸馏反而会降低长推理表现。

Rethinking On-Policy Self-Distillation for Thinking Models

  • 用特权信息做自蒸馏,让模型在长推理中更易出错。
  • 长推理时准确率最高可下降17%,且越长越严重。
  • 适合关注推理过程细节的模型优化研究者。

自蒸馏是语言模型自我提升的有力方法。在该设定中,模型可利用如数学题解法等特权信息充当自身教师。这在思维模型中尤为吸引人,因其可在测试时推理吸收特权信息。然而,我们发现特权自蒸馏会损害思维模型在长推理轨迹上的表现:在 AIME24、AIME25、HMMT25 上评估的五个 Qwen3 与 OLMo 思维模型中,特权上下文蒸馏导致 avg@16 准确率相对下降高达17%。该退化随隐藏的特权上下文增多而加剧,且在长回溯预算下最为明显,此时思维模型本可获得最大收益。这一失效现象不仅限于自蒸馏:有策略蒸馏(OPD)能提升思维模型,但使用特权的 OPD 反而逆转增益。诊断显示,问题源于特权教师上下文重塑了高熵分叉点处的学习机制,在这些位置多个后续路径仍合理,可能引发不同推理路径。特权上下文降低了思维模型回溯中的分叉率,但对指令模型无此影响。这形成有趣反差:特权上下文有助于指令调优模型,却损害更强的思维模型。当学生启动自我修正分支时,特权 OPD 会惩罚原始 OPD 支持的重新考虑标记。经长度归一化后,使用特权教师训练的思维模型仍生成更少验证、回溯和犹豫标记。结果表明,强思维模型的自蒸馏需关注校正与推理步骤中的细粒度信号。

原文摘要 · Abstract (English)

Self-distillation is a promising recipe for self-improvement in language models. In this setting, a model can serve as its own teacher when given privileged information, such as a solution to a math problem. This seems especially appealing for thinking models, which can use test-time reasoning to absorb the privileged information. Surprisingly, we show that privileged self-distillation degrades thinking models on long reasoning traces: across five Qwen3 and OLMo thinking models evaluated on AIME24, AIME25, and HMMT25, privileged-context distillation causes a relative drop of up to 17% in avg@16 accuracy. The degradation scales with the amount of privileged context withheld from the student and is most pronounced at long rollout budgets, where thinking models otherwise obtain their largest gains. This failure mode is not specific to self-distillation: on-policy distillation (OPD) improves thinking models, but privileged OPD reverses these gains. Our diagnostics link this failure mode to how privileged teacher context reshapes learning at high-entropy forking positions, where multiple continuations remain plausible and may lead to different reasoning paths. Privileged context lowers fork rates in thinking-model rollouts but not in instruction-model rollouts. This leads to an interesting dichotomy, where privileged context can help instruction-tuned models but hurts stronger thinking models. The effect is visible when the student begins a self-correction branch, where privileged OPD penalizes sampled reconsideration tokens that vanilla OPD supports. Thinking models trained with a privileged teacher produce fewer verification, backtracking, and hedging markers, even after length normalization. These findings indicate that self-distillation for strong thinking models requires attention to token-level signal, especially around correction and reasoning steps.

思维模型自蒸馏推理优化强化学习

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