arXiv:2605.25582cs.LGcs.AI2026-05

通过分阶段策略蒸馏,提升大模型强化学习的样本效率与性能上限。

Extreme Region Policy Distillation

论文配图:Extreme Region Policy Distillation
图 1 · 摘自论文原文
  • 先在固定数据上激进优化,再用信任域约束蒸馏有效信号。
  • 相比传统方法,性能相当或更优,但KL散度显著降低。
  • 兼容强弱教师,对训练信号不足场景仍有效。

大语言模型的强化学习面临样本效率与最终性能之间的根本矛盾:严格在线方法在一次更新后丢弃轨迹,而离线方法重用数据会引入分布偏移,现有信任域技术主要通过保守优化缓解,常导致丰富训练信号未被充分利用。我们对固定数据进行大量离线更新实验,发现激进的多步优化带来快速初期提升,但过度更新导致轨迹概率偏离、熵崩溃,性能过早平台化;收紧KL约束仅降低上限,无法解决退化问题。为此提出极端区域策略蒸馏(ERPD),采用两阶段框架,解耦样本效率与KL效率。第一阶段在固定数据上进行弱约束离线优化,最大化提取训练信号,生成词级别监督。第二阶段将这些信号在信任域约束下蒸馏至基础策略,过滤有害漂移同时保留有效信号。蒸馏后策略性能相当或更优,且KL散度大幅下降,表明第一阶段的大部分偏差用于无意义漂移而非真实改进。关键优势在于,即使激进优化未能产生更强策略,退化的教师也能通过替代信号构造策略提供有效监督。我们在数学推理任务上验证了ERP,对强基线模型实现突破性提升,对弱教师也表现出稳定改进。

原文摘要 · Abstract (English)

Reinforcement learning for large language models faces a fundamental trade-off between sample efficiency and asymptotic performance: strictly on-policy methods discard trajectories after a single update, while off-policy reuse introduces distribution mismatch that existing trust-region techniques mitigate primarily by enforcing conservative optimization, often leaving rich training signals underutilized. To investigate this, we perform extensive off-policy updates on fixed data. Our experiments reveal that aggressive multi-step optimization brings rapid initial gains, but excessive updates cause trajectory probabilities to deviate and entropy to collapse, with performance plateauing early. Tightening KL constraints merely lowers the ceiling without resolving the degradation. This motivates Extreme Region Policy Distillation (ERPD), a two-stage framework that decouples sample efficiency from KL efficiency. The first stage performs weakly constrained off-policy optimization on fixed data to maximally extract training signals. The resulting policy provides token-level supervision. In the second stage, we distill these signals into the base policy under trust-region constraints, filtering harmful drift while preserving useful signals. The distilled policy achieves comparable or better performance with substantially smaller KL divergence, indicating that much of the first-stage divergence was spent on unnecessary drift rather than genuine improvement. Crucially, ERPD accommodates both strong and weak teachers: when aggressive optimization yields no stronger policy, even degenerate teachers provide effective supervision via alternative signal construction strategies. We validate ERPD on mathematical reasoning, showing gains for strong base models where on-policy training plateaus, and reliable improvements with weak teachers.

强化学习策略蒸馏大模型训练

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