arXiv:2510.18927cs.LGcs.AI2025-10被引 49

提出BAPO方法,解决大模型强化学习中策略不稳、熵降过快问题。

BAPO: Stabilizing Off-Policy Reinforcement Learning for LLMs via Balanced Policy Optimization with Adaptive Clipping

  • 动态调整裁剪边界,平衡正负样本梯度贡献
  • 在AIME 2024/2025上7B和32B模型均超越开源与部分闭源系统
  • 适用于样本重放与部分回溯等离策略场景,训练更稳定高效

强化学习(RL)已成为对齐和增强大语言模型(LLMs)的核心范式。然而,在离策略设置下——即使用旧策略生成的过时数据进行训练——虽提升了样本效率,但面临严峻挑战:策略熵急剧下降,优化常不稳定甚至崩溃。通过理论与实证分析,我们发现两个关键问题:(i) 优化失衡,负优势样本主导策略梯度,抑制有效行为并可能导致梯度爆炸;(ii) 推导出熵-裁剪规则,表明类似PPO的固定裁剪机制会系统性阻碍熵增加更新,导致策略过度利用而牺牲探索。基于此,我们提出平衡策略优化与自适应裁剪(BAPO),一种简单但高效的算法,动态调节裁剪范围以重新平衡正负贡献、维持熵值并稳定优化过程。在多种离策略场景(包括样本重放与部分回溯)中,BAPO实现快速、稳定且数据高效的训练。在AIME 2024与AIME 2025基准测试中,我们的7B BAPO模型超越开源模型如SkyWork-OR1-7B,32B BAPO模型不仅在同规模模型中达到顶尖表现,更优于o3-mini与Gemini-2.5-Flash-Thinking等领先闭源系统。

原文摘要 · Abstract (English)

Reinforcement learning (RL) has recently become the core paradigm for aligning and strengthening large language models (LLMs). Yet, applying RL in off-policy settings--where stale data from past policies are used for training--improves sample efficiency, but remains challenging: policy entropy declines sharply, optimization often becomes unstable and may even collapse. Through theoretical and empirical analysis, we identify two key insights: (i) an imbalance in optimization, where negative-advantage samples dominate the policy gradient, suppressing useful behaviors and risking gradient explosions; and (ii) the derived Entropy-Clip Rule, which reveals that the fixed clipping mechanism in PPO-like objectives systematically blocks entropy-increasing updates, thereby driving the policy toward over-exploitation at the expense of exploration. Building on these insights, we propose BAlanced Policy Optimization with Adaptive Clipping (BAPO), a simple yet effective method that dynamically adjusts clipping bounds to adaptively re-balance positive and negative contributions, preserve entropy, and stabilize RL optimization. Across diverse off-policy scenarios--including sample replay and partial rollout--BAPO achieves fast, stable, and data-efficient training. On AIME 2024 and AIME 2025 benchmarks, our 7B BAPO model surpasses open-source counterparts such as SkyWork-OR1-7B, while our 32B BAPO model not only achieves state-of-the-art results among models of the same scale but also outperforms leading proprietary systems like o3-mini and Gemini-2.5-Flash-Thinking.

强化学习大模型对齐策略优化离策略

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