arXiv:2605.27028cs.LGcs.AI2026-05

通过限制生成前几步,提升强化学习蒸馏的稳定性与效率

Less is More: Early Stopping Rollout for On-Policy Distillation

论文配图:Less is More: Early Stopping Rollout for On-Policy Distillation
图 1 · 摘自论文原文
  • 仅用学生模型前几步生成轨迹进行蒸馏
  • 在多种模型和任务上超越完整轨迹蒸馏,且训练更稳定
  • 适合跨模型家族场景,显存占用更低

基于策略的蒸馏(OPD)近期成为序列级模仿学习的替代方案,通过教师模型对学生的自采样轨迹打分来训练学生。然而我们观察到‘离策略教师衰减’问题:对于后续词元,学生早期轨迹作为上下文已偏离教师策略,导致教师修正能力下降,可能退化为预训练阶段的词元补全行为。我们实证验证了该问题,并提出早期停止采样(ESR)策略:简单地将采样过程限制在响应的前几个词元。实验表明,ESR在不同模型规模、类型、任务及训练设置下均优于完整轨迹蒸馏,且具备更高的GPU效率与训练稳定性,尤其在跨模型家族场景中表现显著。进一步分析发现ESR存在‘级联对齐’与‘子模式锁定’效应,可能解释其为何能超越教师性能。此外,该位置选择策略无法仅用KL散度或熵信号解释。

原文摘要 · Abstract (English)

On-policy distillation has recently emerged as a promising alternative to standard sequence-level imitation, training a student by scoring its own rollouts with a teacher model. However, we observe ``Off-policy Teacher Decay'' problem in this paradigm: for the later tokens, with student's earlier trajectory as context that is off-policy to the teacher, the teacher's ability to produce a corrective score would decay, and may fall back to token-completion behavior learned in the pre-training stage. We empirically verify this problem, and we propose Early Stopping Rollout (ESR) to fix it: a simple yet effective distillation strategy that simply restricts the rollout generation to the first response tokens. We show that ESR both surpasses the full rollout OPD performance across model size, family, tasks and training regime, and exhibit much higher GPU efficiency and training stability, especially under cross model family scenarios. We further investigate the mechanism behind this surprising performance and discovered "Cascading Alignment" and "Sub-mode Commitment" effect of ESR that may explain why it works effectively and even sometimes exceeding the teacher model performance. Besides, we show that this position-based token selection strategy cannot be fully explainable by KL divergence and entropy signals.

强化学习模型蒸馏序列生成训练优化

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