← 返回 2026-02-13

揭示隐式优势对称性:为何GRPO在探索和难度适应上表现挣扎 Unveiling Implicit Advantage Symmetry: Why GRPO Struggles with Exploration and Difficulty Adaptation

Zhiqi Yu, Zhangquan Chen, Mengting Liu, Heye Zhang, Liangqiong Qu 📅 2026-02-05 👍 13 2026-07-13 08:35
GRPO LLM推理 优势估计 强化学习 数学推理

GRPO的隐式优势对称性导致探索不足和难度适应差,A-GRAE通过非对称机制解决

前置知识

GRPO (Group Relative Policy Optimization)

GRPO是DeepSeek-R1等先进系统中使用的强化学习算法,它通过组内相对优势估计(Group Relative Advantage Estimation, GRAE)来替代PPO中的价值模型。具体来说,对于每个查询q,GRPO从旧策略πθold采样G个响应{o1,...,oG},通过规则验证器获得奖励{r1,...,rG},然后计算每个响应的相对优势Ai = (ri - mean(r)) / std(r)。这种方法避免了训练一个与策略模型同等规模的critic模型,大大简化了训练流程。

本文的核心论点是GRPO存在隐式优势对称性问题,理解GRPO的基本机制是理解本文贡献的前提。

Pass@k 评估指标

Pass@k是衡量模型推理能力的重要指标,表示在k次采样中至少有一次正确的概率。与Pass@1(贪婪解码准确率)不同,Pass@k能够反映模型的探索能力和推理边界。当k增大时,Pass@k应该单调递增,但如果模型发生能力边界收缩,Pass@k可能会在某个k值后低于基线模型,这表明模型的推理多样性受到了损害。

本文发现GRPO的Pass@k在大k值时会低于基线模型,这是论证GRPO存在探索问题的关键证据。

RLVR (Reinforcement Learning with Verifiable Rewards)

RLVR是一种利用可验证奖励进行强化学习的范式,特别适用于数学推理等有明确正确答案的任务。在这种范式中,奖励ri由规则验证器提供(如答案是否正确),而不是由学习的奖励模型提供。这避免了奖励模型可能带来的偏差和不稳定,使得训练信号更加可靠。RLVR已经成为激活基础模型推理能力的标准方法。

RLVR是本文研究的背景框架,本文在RLVR范式下分析GRPO的优势对称性问题。

优势对称性 (Advantage Symmetry)

这是本文提出的核心概念,指GRAE中存在的一种隐式性质。在组级别,正确轨迹的优势权重之和严格等于错误轨迹的优势权重之和,即∑|Ai| (i∈Gpos) = ∑|Ai| (i∈Gneg),这导致零和性质∑Ai = 0。在样本级别,中等难度样本(成功概率p=0.5)的绝对优势之和最大,呈现对称的倒U形曲线。这种对称性限制了模型的探索能力和难度适应能力。

这是本文的核心发现和贡献,理解优势对称性是理解A-GRAE设计动机的关键。

熵崩溃 (Entropy Collapse)

熵崩溃是指在强化学习训练过程中,模型输出分布的熵急剧下降的现象。当模型过度强调某些轨迹时,输出分布会变得过于尖锐,导致采样多样性降低。在本文中,Positive-Dominant组(放大正确轨迹权重)表现出最急剧的熵下降,这直接导致了模型探索能力的丧失。熵崩溃是能力边界收缩的直接原因之一。

本文通过熵动态分析揭示了过度利用已知正确路径的负面后果,这是设计A-GRAE组级别非对称机制的重要依据。

研究动机

GRPO作为DeepSeek-R1和OpenAI-o1等先进系统中的标准强化学习算法,虽然在激活LLM推理能力方面取得了显著成功,但存在两个根本性限制。第一个问题是能力边界收缩:实验表明GRPO的Pass@k在k值增大时甚至会低于基线模型,在AMC23和MATH数据集上,GRPO的Pass@256分别从基线模型的100%和96.3%下降到92.5%和95.0%,这意味着GRPO虽然提高了正确路径的采样概率,但未能发现基线模型原始采样支持之外的新解,导致推理边界实际上收缩了。第二个问题是难度适应不足:GRPO的奖励机制对所有任务一视同仁,不考虑任务的内在复杂性或模型的当前能力,这往往导致在简单任务上灾难性过拟合或在困难任务上学习不足。

本文的目标是本文的核心目标是深入理解GRPO失败的根本原因,并在此基础上提出一种能够同时解决探索不足和难度适应问题的改进方案。具体来说,作者希望:(1)识别并形式化GRAE中隐式的'优势对称性'性质;(2)通过控制实验验证这种对称性是次优的;(3)基于实验洞察设计一种能够动态调节探索激励和样本难度焦点的非对称优势估计方法。

与已有工作不同的是,本文的独特切入角度是将GRPO的梯度优化重新表述为一种带权重的SFT(监督微调)变体,从而揭示了之前被忽视的'优势对称性'性质。虽然近期工作如Dr.GRPO和DAPO改进了GRAE范式,但它们仍然保留了这种固有的对称性,因此未能从根本上解决边界收缩和难度适应问题。一些方法通过负学习隐式打破组级别对称性,另一些通过强调高熵token或困难样本隐式打破样本级别对称性,但这些方法只在其中一个层面上隐式处理对称性,无法同时提高准确性和多样性。本文首次同时在两个层面系统性地分析和解决对称性问题。

核心方法

本文的方法论遵循'分析-验证-设计'的三阶段路线。首先,作者将GRPO的梯度优化重新表述为SFT的加权变体,其中优势值Ai作为权重因子。通过这种视角,作者发现GRAE存在两个层面的隐式对称性:组级别上,正确和错误轨迹的权重严格相等;样本级别上,中等难度样本获得最大更新量。然后,作者设计了两组控制实验来验证打破这些对称性的因果效应。最后,基于实验洞察,作者提出A-GRAE框架,通过两个关键机制打破对称性:(1)在组级别衰减正确轨迹的优势以鼓励探索;(2)在样本级别实现从易到难的动态课程学习。整个方法的核心直觉是:对称性虽然简洁,但在训练动态中是次优的,适度的非对称性能够更好地平衡探索与利用。

A-GRAE的核心创新在于同时在两个层面打破优势对称性,并引入动态调节机制。与已有方法的本质区别体现在三个方面:第一,这是首个同时处理组级别和样本级别对称性的方法,而之前的工作只隐式处理其中一个层面;第二,A-GRAE引入了基于训练进度的动态调节,使用批量平均奖励ωs作为训练状态的代理指标,实现从探索到利用的平滑过渡,而不是使用静态的权重策略;第三,在组级别,A-GRAE采用衰减抑制策略而非简单的抑制或放大,即A* = α·min(1, ωs)·Ai(当Ai > 0时),这在早期鼓励探索的同时在后期保持稳定性。这种设计解决了之前实验中发现的'持续熵增长导致训练不稳定'的问题。

方法步骤详情

A-GRAE的实现包含以下步骤:首先,计算批量平均奖励ωs = (∑ri) / B作为训练状态的代理指标,其中B是批量中的总轨迹数,ωs越高表示模型能力越强。然后,在样本级别实现动态注意力转移:Ai = (ωs·ri / (2·std·√p)) + ((1-ωs)·ri / (2·std·√(1-p))),其中p是采样成功率。这个公式的直觉是:当ωs较低(训练早期)时,简单样本(高p)获得更多权重;当ωs较高(训练后期)时,困难样本(低p)获得更多权重。在组级别,对正优势进行衰减抑制:当Ai > 0时,A* = α·min(1, ωs)·Ai;当Ai ≤ 0时,A* = Ai,其中α ≤ 1是缩放参数。最后,将计算得到的A*代入GRPO的目标函数进行策略优化。

技术新颖性

本文的技术新颖性体现在多个层面。首先,理论贡献是首次识别并形式化了GRAE的优势对称性性质,这为理解GRPO及其变体的行为提供了统一的解释框架。其次,Theorem 1(行为空间中的logits更新)和Theorem 2(更新幅度与样本难度的关系)提供了严格的数学分析,其中Theorem 2揭示了∑|Ai| = 2|G|√(p(1-p))的关键公式,解释了为何中等难度样本获得最大更新。第三,实验设计的创新在于通过系统性的控制实验(放大vs抑制正确轨迹、硬聚焦vs软聚焦)验证对称性的因果效应,这种'反事实'实验设计在RLVR领域是新颖的。最后,A-GRAE的动态调节机制(基于ωs的自适应权重)比静态策略更加符合训练动态的需求。

训练集上正确采样响应的批量内计数
Figure 5: 训练集上正确采样响应的批量内计数

实验结果

本文在七个基准测试上进行了全面实验,涵盖文本推理和多模态推理任务。在Qwen2.5-Math-7B上的主要发现包括:(1)GRPO虽然显著提高了Pass@1准确率(MATH从63.4%提升到76.5%),但在大k值时Pass@k会低于基线模型(MATH Pass@256从96.3%下降到95.0%,AMC23从100%下降到92.5%),证实了能力边界收缩问题。(2)放大正确轨迹权重(Positive-Dominant)导致熵崩溃,在AIME2025和MATH数据集上k>16时性能显著下降。(3)抑制正确轨迹权重(Negative-Dominant)虽然提高了Pass@k(AIME2025 Pass@256从46.7%提升到56.7%),但存在训练不稳定风险。(4)A-GRAE在所有GRPO变体上都实现了持续改进:GRPO+A-GRAE在MATH上Pass@1从76.5%提升到78.3%,Pass@256从95.0%提升到96.5%;DAPO+A-GRAE在AIME2025上Pass@1从12.0%提升到13.3%,Pass@256从53.3%提升到60.0%。(5)多模态实验也验证了A-GRAE的普适性,Dr.GRPO+A-GRAE在Geo3K上从44.9%提升到46.8%,在Xray300上从69.5%提升到72.0%。

控制实验总结
Table 1: 控制实验总结
MATH、AIME 2025和AMC23上的Pass@k结果(Qwen2.5-Math-7B)
Table 2: MATH、AIME 2025和AMC23上的Pass@k结果(Qwen2.5-Math-7B)
MATH、AIME 2025和AMC23上的Pass@k结果(带置信区间)
Table 3: MATH、AIME 2025和AMC23上的Pass@k结果(带置信区间)
多模态基准上的性能对比(Qwen2.5-VL-3B-Instruct)
Table 4: 多模态基准上的性能对比(Qwen2.5-VL-3B-Instruct)
打破组级别对称性的实验结果(Qwen2.5-Math-7B)
Figure 2: 打破组级别对称性的实验结果(Qwen2.5-Math-7B)
实验I中三个组在训练集上的熵动态
Figure 3: 实验I中三个组在训练集上的熵动态
打破样本级别对称性的实验结果(Qwen2.5-Math-7B)
Figure 4: 打破样本级别对称性的实验结果(Qwen2.5-Math-7B)
查看结构化数据
任务指标本文基线提升
MATH (数学推理) Pass@1 GRPO+A-GRAE: 78.3%, Dr.GRPO+A-GRAE: 78.6% GRPO: 76.5%, Dr.GRPO: 77.2% +1.8% (GRPO), +1.4% (Dr.GRPO)
MATH (数学推理) Pass@256 GRPO+A-GRAE: 96.5%, Dr.GRPO+A-GRAE: 96.9% GRPO: 95.0%, Dr.GRPO: 95.0% +1.5% (GRPO), +1.9% (Dr.GRPO)
AIME 2025 (高难度数学) Pass@1 GRPO+A-GRAE: 11.3%, DAPO+A-GRAE: 13.3% GRPO: 10.3%, DAPO: 12.0% +1.0% (GRPO), +1.3% (DAPO)
AIME 2025 (高难度数学) Pass@256 GRPO+A-GRAE: 56.7%, DAPO+A-GRAE: 60.0% GRPO: 46.7%, DAPO: 53.3% +10.0% (GRPO), +6.7% (DAPO)
AMC23 (数学竞赛) Pass@1 GRPO+A-GRAE: 62.6%, DAPO+A-GRAE: 63.3% GRPO: 60.2%, DAPO: 62.0% +2.4% (GRPO), +1.3% (DAPO)
AMC23 (数学竞赛) Pass@256 GRPO+A-GRAE: 97.5%, DAPO+A-GRAE: 100.0% GRPO: 92.5%, DAPO: 97.5% +5.0% (GRPO), +2.5% (DAPO)
Geo3K (多模态几何) Pass@1 Dr.GRPO+A-GRAE: 46.8% Dr.GRPO: 44.9% +1.9%
Xray300 (医学影像) Pass@1 Dr.GRPO+A-GRAE: 72.0% Dr.GRPO: 69.5% +2.5%

局限与改进

尽管A-GRAE取得了显著改进,但仍存在一些局限性。首先,方法引入了额外的超参数α(缩放参数),虽然作者建议α ≤ 1,但最优值可能因任务和模型而异,需要额外的调参工作。其次,使用批量平均奖励ωs作为训练状态的代理指标可能不够精确,特别是在奖励稀疏或分布不均的情况下。第三,本文的实验主要集中在数学推理和医学影像任务上,对于其他类型的任务(如代码生成、创意写作等),A-GRAE的有效性尚未验证。第四,论文未详细讨论计算开销,虽然A-GRAE的计算复杂度应该与原始GRPO相当,但在大规模训练中的实际效率差异需要进一步评估。第五,作者观察到Negative-Dominant组在后期训练中会出现不稳定(突然增加的完全未解决问题),虽然A-GRAE通过动态调节缓解了这个问题,但未提供严格的理论保证。

独立分析的弱点

本文存在几个值得深入探讨的弱点。首先,Theorem 2的推导假设了二元奖励(ri ∈ {0, 1}),但对于更复杂的奖励函数(如部分正确的连续奖励),优势对称性的表现可能不同,这限制了理论的普适性。改进方向是将分析扩展到连续奖励空间。其次,动态调节机制依赖于批量平均奖励ωs,这可能受到批量大小和采样策略的影响,在小批量或高度不平衡的采样中可能不稳定。可以考虑使用指数移动平均或其他更鲁棒的状态估计方法。第三,组级别的衰减抑制策略(α·min(1, ωs)·Ai)中,α的选择缺乏理论指导,可以探索自适应的α调整策略。第四,论文未充分讨论不同α值对训练稳定性的影响,消融实验中只展示了α=10的结果,缺乏更细致的超参数敏感性分析。

未来方向

基于本文的成果,可以从多个方向进行延伸研究。首先,可以将A-GRAE扩展到更广泛的RLVR任务中,如代码生成、工具使用、多轮对话等,验证优势对称性问题在这些领域是否同样存在。其次,可以探索更精细的动态调节策略,例如基于课程学习理论的自适应难度调度,或者基于元学习的自动超参数调整。第三,可以将优势对称性的分析与其他改进GRPO的方法(如token级别的优势估计、过程奖励模型)相结合,探索是否存在协同效应。第四,可以进一步研究优势对称性与模型规模的关系,即在更大规模的模型(如70B、400B参数)中,对称性问题是否更加严重或有所缓解。第五,可以探索在分布式训练环境中A-GRAE的实现和优化,因为当前的分析主要基于单进程设置。

复现评估

本文在复现性方面提供了较好的支持。作者在GitHub上开源了代码(https://github.com/HKU-HealthAI/A-GRAE),这大大降低了复现门槛。实验使用了公开可用的模型(Qwen2.5-Math-7B、Llama-3.2-3B-Instruct、Qwen2.5-VL-3B-Instruct)和数据集(MATH、AMC23、AIME2025等),这些都是广泛使用的基准。训练框架verl也是开源的。超参数设置相对明确(batch size=1024, G=8, mini-batch=256, lr=1e-6)。然而,完整复现可能需要相当的计算资源(7B模型的RL训练通常需要多GPU),且论文未提供详细的硬件配置和训练时间。此外,A-GRAE引入的超参数α需要额外的调参工作,可能影响复现的一致性。