arXiv:2605.21125cs.LG2026-05中稿 · ICML被引 4

解决大模型强化学习中奖励趋同导致的训练失效问题。

Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation

论文配图:Advantage Collapse in Group Relative Policy Optimization: Diagnosis and Mitigation
图 1 · 摘自论文原文
  • 引入新指标ACR诊断奖励趋同现象,量化梯度失效比例。
  • 提出AVSPO方法,在不增加采样成本下提升各规模模型准确率4-6个百分点。
  • 适合关注大模型推理能力提升与强化学习稳定性的研究者。

Group Relative Policy Optimization (GRPO) 是强化学习从可验证奖励(RLVR)框架中的主流算法,在提升大语言模型(LLM)推理能力方面表现优异。然而,GRPO 易出现优势崩溃(advantage collapse):当组内奖励趋同(如全对或全错)时,优势值趋近零,导致梯度消失。为此,我们提出首个诊断指标——优势崩溃率(ACR),量化训练批次中无效梯度的比例。在0.5B至14B参数的大模型上,数学推理基准测试显示,ACR能有效预测训练停滞与最终性能。进一步提出轻量级扩展算法AVSPO,通过实时监控ACR注入虚拟奖励样本,使模型能在无额外模型回滚的情况下从同质组中学习。AVSPO相较GRPO降低58%-63%的优势崩溃率,并在所有模型尺度上实现4-6个百分点的稳定准确率提升,同时保持跨领域任务泛化能力。代码与数据集见https://github.com/hexixiang/Advantage-Collapse-Rate。

原文摘要 · Abstract (English)

Group Relative Policy Optimization (GRPO), a prominent algorithm within the Reinforcement Learning from Verifiable Rewards (RLVR) framework, has achieved strong results in improving the reasoning capabilities of large language models (LLMs). However, GRPO is prone to advantage collapse, a failure mode where homogeneous rewards within a group (e.g., all correct or all incorrect answers) yield near-zero advantages and vanishing gradients. To address this, we introduce the Advantage Collapse Rate (ACR), the first diagnostic metric quantifying the proportion of training batches with ineffective gradients. Across models from 0.5B to 14B parameters on mathematical reasoning benchmarks, we show that ACR strongly predicts training stagnation and final performance. We then propose Adaptive Virtual Sample Policy Optimization (AVSPO), a lightweight extension of GRPO that injects virtual reward samples, guided by real-time ACR monitoring, to enable learning from homogeneous groups without additional model rollouts. AVSPO reduces advantage collapse by 58-63% relative to GRPO and yields consistent accuracy gains of 4-6 percentage points across all model scales, while maintaining generalization on the evaluated out-of-domain task. Code and datasets are available at https://github.com/hexixiang/Advantage-Collapse-Rate.

强化学习大模型推理梯度优化奖励设计

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