用平滑正则化改进大模型强化学习的稳定性和效率
Rethinking the Divergence Regularization in LLM RL

- 用连续可导的正则项替代硬性裁剪,实现梯度平滑修正
- 在不同模型规模和精度下均提升训练稳定性和收敛速度
- 适合追求高效稳定训练的大模型强化学习研究者
强化学习已成为大语言模型后训练的关键环节。由于训练与推理不一致及策略停滞问题,大模型RL通常为离线策略,需信任区域控制以保证优化稳定性。主流方法如PPO和GRPO通过比例裁剪近似控制,但在长尾词汇分布下,重要性比可能无法准确反映分布偏移。近期方法DPPO改用基于散度的掩码,定义以采样词绝对概率变化为边界的信任区域。然而DPPO仍采用硬掩码:一旦词越过有害方向的信任边界,其梯度即被丢弃。为此,本文提出分散正则化策略优化(DRPO),将硬掩码替换为对策略偏移的平滑优势加权二次正则项。该方法保持与DPPO相同的信任区域几何结构,同时引入有界且连续的梯度权重,可衰减发散更新并提供边界外的纠正信号。跨模型规模、架构和精度设置的实验表明,DRPO显著提升了大模型强化学习训练的稳定性与效率。
原文摘要 · Abstract (English)
Reinforcement learning (RL) has become a key component of post-training large language models (LLMs). In practice, LLM RL is often off-policy because of training-inference mismatch and policy staleness, making trust-region control essential for stable optimization. Mainstream methods such as PPO and GRPO approximate this control with a ratio-clipping mechanism, but the importance ratio can be a poor proxy for distributional shift in long-tailed vocabularies. Recent work such as DPPO addresses this mismatch by replacing ratio-based clipping with a divergence-based mask, yielding a trust region defined by the sampled token's absolute probability shift. However, DPPO still relies on a hard mask: once a token crosses the trust-region boundary in a harmful direction, its gradient is discarded rather than corrected. To address this, we propose Divergence Regularized Policy Optimization (DRPO), which replaces the hard mask with a smooth advantage-weighted quadratic regularizer on policy shift. DRPO preserves the same trust-region geometry as DPPO while inducing bounded, continuous gradient weights that attenuate diverging updates and provide corrective signals beyond the boundary. Experiments across model scales, architectures, and precision settings show that DRPO improves the stability and efficiency of LLM RL training.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。