DRPO:用平滑散度正则化改进LLM强化学习稳定性

Rethinking the Divergence Regularization in LLM RL

精选理由

做LLM RL后训练的团队终于有了更优雅的信任区域控制方案——DRPO用连续正则化替代硬裁剪,解决了长尾词汇下梯度丢弃问题,训练更稳且收敛更快,做RLHF或GRPO优化的建议直接读论文。

AI 摘要

该论文指出,在LLM后训练中常用的PPO和GRPO方法依赖重要性比率裁剪来约束策略更新,但在长尾词汇分布下,比率无法准确反映分布偏移。DPPO虽改用散度掩码,但硬掩码会直接丢弃越界token的梯度,导致信息损失。作者提出DRPO,用平滑的优势加权二次正则项替代硬掩码,在保持相同信任区域几何的同时,提供连续梯度权重,既能抑制发散更新,又能在边界外提供修正信号。实验表明,DRPO在不同模型规模、架构和精度设置下均提升了训练稳定性和效率。

AI 翻译 · 中文

该论文指出,在LLM后训练中常用的PPO和GRPO方法依赖重要性比率裁剪来约束策略更新,但在长尾词汇分布下,比率无法准确反映分布偏移。DPPO虽改用散度掩码,但硬掩码会直接丢弃越界token的梯度,导致信息损失。作者提出DRPO,用平滑的优势加权二次正则项替代硬掩码,在保持相同信任区域几何的同时,提供连续梯度权重,既能抑制发散更新,又能在边界外提供修正信号。实验表明,DRPO在不同模型规模、架构和精度设置下均提升了训练稳定性和效率。

arXiv cs.LGReinforcement 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-regi