arXiv:2605.07804cs.LGcs.AI2026-05被引 15

动态剪枝强化推理模型训练,提升长序列任务效率与可靠性

Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning

论文配图:Prune-OPD: Efficient and Reliable On-Policy Distillation for Long-Horizon Reasoning
图 1 · 摘自论文原文
  • 实时检测学生与教师思维偏差,动态调整训练轨迹
  • 在思维漂移时减少37.6%~68.0%训练时间,性能不降反升
  • 适合长序列推理、资源受限场景下的高效模型蒸馏

在线策略蒸馏(OPD)利用密集教师奖励来增强推理模型。然而,将OPD扩展到长时序任务时暴露关键缺陷:随着学生生成的前缀逐渐偏离教师思维路径,教师的密集奖励失去局部可利用性。继续在这些‘漂移’轨迹上生成和评估令牌不仅降低奖励质量,还造成巨大计算浪费。为此,我们提出Prune-OPD框架,动态对齐训练预算与监督质量。通过持续监测学生与教师预测的局部兼容性(如采用top-k重叠),Prune-OPD实时检测前缀漂移事件。一旦检测到严重漂移,即单调降低后续不可靠奖励权重,并触发动态回溯截断。这使训练过程能及时停止无效生成,将计算严格分配给可靠教师监督。在多种教师-学生组合下,Prune-OPD始终实现计算与监督可靠性对齐。当前缀漂移导致密集教师奖励不可靠时,训练时间减少37.6%–68.0%,且在挑战性基准(AMC, AIME, HMMT)上性能保持甚至提升;当学生-教师兼容性高时,自动扩展训练窗口以保留长上下文监督。结果表明,Prune-OPD并非盲目缩短轨迹,而是将计算重新分配至局部可利用的教师奖励。

原文摘要 · Abstract (English)

On-policy distillation (OPD) leverages dense teacher rewards to enhance reasoning models. However, scaling OPD to long-horizon tasks exposes a critical flaw: as the student's generated prefix inevitably diverges from the teacher's thought process, the teacher's dense reward loses local exploitability. Continuing to generate and evaluate tokens on these ``drifted'' trajectories not only degrades reward quality but also incurs massive computational waste. To address this, we introduce \textbf{Prune-OPD}, a framework that dynamically aligns training budgets with supervision quality. By continuously monitoring the local compatibility between student and teacher predictions (e.g., via top-$k$ overlap), Prune-OPD detects prefix-drift events in real time. Upon detecting severe drift, it monotonically down-weights subsequent unreliable rewards and triggers dynamic rollout truncation. This allows the training process to halt futile generation and reallocate compute strictly to reliable teacher supervision. Across diverse teacher-student combinations, Prune-OPD consistently aligns computation with supervision reliability. When prefix drift makes dense teacher rewards unreliable, it reduces training time by 37.6\%--68.0\% while preserving, and often improving, performance on challenging benchmarks (AMC, AIME, HMMT). When student-teacher compatibility remains high, it automatically preserves long-context supervision by expanding the training window. These results suggest that Prune-OPD improves OPD not by blindly shortening rollouts, but by reallocating computation toward locally exploitable teacher rewards.

模型蒸馏长序列推理高效训练动态剪枝

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