arXiv:2503.06639cs.LGstat.ML2025-03被引 121

GRPO通过奖励校准实现推理增强,能有效提升大模型成功率。

Reinforcement Learning with Verifiable Rewards: GRPO's Effective Loss, Dynamics, and Success Amplification

  • 用前一策略生成合成数据构建对比损失,优化策略更新
  • 成功率随迭代收敛至高于参考策略的固定点
  • 适合需要可靠推理的LLM强化学习场景

最近提出的组相对策略优化(GRPO)用于在可验证(二元)奖励下提升大语言模型的推理能力。我们发现,奖励的均值与方差校准会诱导出一种加权对比损失,其中对比样本来自前一策略生成的合成数据。尽管原始GRPO采用裁剪以保持更新靠近旧策略,本文分析了不同奖励归一化方式(仅均值或均值+方差)以及使用KL散度正则化更新的变体:惩罚与前一模型的差异(镜像)、惩罚与固定参考模型$π_{\mathrm{ref}}$的差异,或两者结合。对每种情况,最优策略$π_n$均可显式表示为二元奖励及$π_{n-1}$下奖励的一阶和二阶统计量,以及$π_{n-1}$和$π_{\mathrm{ref}}$的函数。迭代后,策略序列$\{π_n\}$的成功概率(PoS)遵循简单递推关系,收敛至由参考策略成功概率和正则化强度决定的固定点。我们进一步证明该固定点高于参考值,表明GRPO能放大策略的成功概率。

原文摘要 · Abstract (English)

Group Relative Policy Optimization (GRPO) was introduced and used recently for promoting reasoning in LLMs under verifiable (binary) rewards. We show that the mean + variance calibration of these rewards induces a weighted contrastive loss in which the contrastive samples are synthetic data drawn from the previous policy. While GRPO was originally paired with clipping to keep updates near the old policy, we analyze variants that differ in reward normalization (mean-only vs mean + variance) and in how they regularize updates using KL divergence: either penalizing divergence from the previous model (mirror), penalizing divergence from a fixed reference model $π_{\mathrm{ref}}$, or combining both forms of regularization. For each, the optimal policy $π_n$ admits an explicit form in terms of the binary reward and the first and second order statistics of the reward under $π_{n-1}$, as well as the policies $π_{n-1}$ and $π_{\mathrm{ref}}$. Iterating results in a sequence $\{π_n\}$ whose probability of success (PoS) obeys a simple recurrence that converges to a fixed point determined by the reference PoS and the regularization strength. We further show that this fixed point exceeds the reference, demonstrating that GRPO amplifies the policy's probability of success.

强化学习大模型推理策略优化奖励校准

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