arXiv:2505.23585cs.LGcs.CL2025-05被引 33

提出OPO算法,让大模型强化学习更稳定高效

On-Policy RL with Optimal Reward Baseline

  • 采用精确的在线策略训练,提升稳定性
  • 优化奖励基线降低梯度方差,无需额外模型
  • 适合需要稳定生成多样结果的大模型对齐任务

强化学习是实现大语言模型与人类偏好对齐及增强推理能力的基础。然而,现有算法常因宽松的在线策略约束导致训练不稳定,且依赖辅助模型造成计算效率低下。本文提出一种新型简化强化学习算法OPO(On-Policy RL with Optimal Reward Baseline),强调精确在线策略训练,实证上可稳定训练过程并促进探索。同时,OPO引入可实践的最优奖励基线公式,有效降低梯度方差。在数学推理基准测试中,OPO表现优异,训练稳定,无需额外模型或正则化项。此外,其政策偏移更低、输出熵更高,推动生成结果更丰富且少重复。该方法为大语言模型对齐与推理任务中的稳定高效强化学习提供了新方向。实现已合并至verl库:https://verl.readthedocs.io/en/latest/algo/opo.html。

原文摘要 · Abstract (English)

Reinforcement learning algorithms are fundamental to align large language models with human preferences and to enhance their reasoning capabilities. However, current reinforcement learning algorithms often suffer from training instability due to loose on-policy constraints and computational inefficiency due to auxiliary models. In this work, we propose On-Policy RL with Optimal reward baseline (OPO), a novel and simplified reinforcement learning algorithm designed to address these challenges. OPO emphasizes the importance of exact on-policy training, which empirically stabilizes the training process and enhances exploration. Moreover, OPO integrates a practically feasible formulation of the optimal reward baseline that minimizes gradient variance. We evaluate OPO on mathematical reasoning benchmarks. The results demonstrate its superior performance and training stability without additional models or regularization terms. Furthermore, OPO achieves lower policy shifts and higher output entropy, encouraging more diverse and less repetitive responses. These results highlight OPO as a promising direction for stable and effective reinforcement learning in large language model alignment and reasoning tasks. The implementation is merged into the verl library at https://verl.readthedocs.io/en/latest/algo/opo.html.

强化学习大模型对齐训练稳定奖励基线

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