arXiv:2607.29209cs.LGcs.AI2026-07被引 1

解决强化学习中优势融合的稳定性问题,提升模型性能与训练稳定

SAF-OPD: Stable Advantage Fusion for On-Policy Distillation

论文配图:SAF-OPD: Stable Advantage Fusion for On-Policy Distillation
图 1 · 摘自论文原文
  • 通过分阶段机制控制优势信号强度和时间跨度,避免熵崩溃
  • 在7个数学推理与代码生成任务上,性能提升0.51%至2.70%
  • 适合追求训练稳定性和超越教师表现的强化学习研究者

基于可验证奖励的强化学习(RLVR)向每个标记广播单一响应级奖励,而在线策略蒸馏(OPD)对每个标记评分以获得密集优势,但受限于教师质量且抑制探索。二者互补性使融合有前景,但固定系数融合引发熵崩溃:因幅度不匹配,标记级OPD优势可能远超有界的RLVR优势,淹没其信号;因时间不匹配,持续高强度的OPD会持续拉拽学生模型趋近教师,限制超越探索。本文提出SAF框架,通过轻量四阶段流水线仅作用于OPD优势,分别实现幅度控制(稀疏-压缩)与时间控制(预热-退火),各阶段独立可开关,开销极小。以GRPO实例化RLVR,评估SAF在七项数学推理与代码生成基准上,覆盖Qwen3-1.7B/4B/8B模型:显著避免熵崩溃,一致优于固定系数融合,在所有六种模型-领域设置下提升综合得分0.51%-2.70%,并实现更稳定训练。

原文摘要 · Abstract (English)

Reinforcement learning with verifiable rewards (RLVR) broadcasts a single response-level reward to every token, while on-policy distillation (OPD) scores each token against a stronger teacher for a dense advantage but caps performance at teacher quality and discourages exploration beyond it. Their complementarity makes combining RLVR and OPD promising, but we find that fusing the two advantages with a fixed coefficient triggers entropy collapse from two miscalibrations: a magnitude mismatch, where token-level OPD advantages can spike far beyond the bounded RLVR advantage and erase its signal, and a temporal mismatch, where sustained full-strength OPD keeps pulling the student toward the teacher and limits exploration needed to surpass it. We propose SAF, a Stable Advantage Fusion framework that resolves both issues via a lightweight, four-stage pipeline applied only to the OPD advantage: a sparsify-then-compress mechanism for magnitude control paired with a warm-up-then-anneal mechanism for temporal control, with each stage independently switchable and adding negligible overhead. Instantiating RLVR with GRPO, we evaluate SAF across seven mathematical reasoning and code generation benchmarks with Qwen3-1.7B/4B/8B: SAF avoids entropy collapse and consistently outperforms fixed-coefficient GRPO+OPD fusion, improving the aggregate score by 0.51-2.70% across all six model-domain settings while achieving more stable training.

强化学习蒸馏训练稳定代码生成

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