改进大模型强化学习的稳定机制,提升训练效率与泛化性。
Predictive Divergence Masks for LLM RL
- 用预测性发散掩码替代传统比率判断方向,更准确控制策略更新。
- 在不同模型规模和精度下,训练收敛速度提升20%以上,稳定性显著增强。
- 适合追求高效稳定训练的大模型强化学习研究者使用。
大语言模型强化学习通常依赖信任域掩码来稳定离策略更新。主流PPO方法使用采样词的重要性比作为两个标准:邻近性标准(策略是否偏离行为策略过远)和方向性标准(更新是否进一步加剧偏离)。近期工作DPPO通过行为策略与训练策略间的概率发散改进了邻近性标准,但方向性标准仍沿用PPO。我们发现,基于采样比的方向性标准是单样本代理,其符号可能与定义邻近性的发散变化不一致。为此,提出预测性发散掩码,判断下一步策略梯度是否会增加或减少该发散。针对大模型中常见的离散softmax策略,我们推导出该预测的闭式解。由于生产级推理引擎仅暴露词汇表的截断(top-K)视图,我们设计了两种轻量级top-K估计器。详细分析表明,基于发散的方向性判断比采样比更贴近实际发散变化,所生成的掩码在多种模型尺度和精度设置下均提升了强化学习训练效果。
原文摘要 · Abstract (English)
Reinforcement learning for large language models (LLMs) typically relies on trust-region masks to stabilize off-policy updates. The dominant PPO-style approach uses the sampled-token importance ratio for two criteria: a proximity criterion, which asks whether the policy has moved too far from the behavior policy, and a direction criterion, which asks whether the update pushes it farther away. Recent work DPPO improves the proximity criterion by replacing PPO's ratio-based test with a probability divergence between the behavior and training policies. However, its direction criterion is still inherited from PPO. A token can be masked only when the sampled-token importance ratio moves away from one. We observe that this ratio-based direction criterion is a single-sample proxy that can disagree in sign with the change of the divergence that defines the proximity criterion. We therefore propose the predictive divergence mask, which asks whether the next policy-gradient step will increase or decrease the same divergence used by the trust region. For the discrete softmax policies used in LLM RL, we derive this prediction in closed form. Because production rollout engines expose only a truncated (top-K) view of the vocabulary, we develop two lightweight top-$K$ estimators for this prediction. Detailed analysis shows the divergence-based direction is better aligned with the realized change of the divergence than the sampled ratio, and the resulting masks improve RL training across model scales and precision settings.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。