← 返回 2026-01-09

GDPO:基于组奖励解耦归一化的多奖励强化学习策略优化方法 GDPO: Group reward-Decoupled Normalization Policy Optimization for Multi-reward RL Optimization

Shih-Yang Liu, Xin Dong, Ximing Lu, Shizhe Diao, Peter Belcak, Mingjie Liu, Min-Hung Chen, Hongxu Yin, Yu-Chiang Frank Wang, Kwang-Ting Cheng, Yejin Choi, Jan Kautz, Pavlo Molchanov 📅 2026-01-08 👍 235 2026-07-13 08:35
GRPO LLM对齐 多奖励优化 强化学习 策略优化

通过解耦多奖励归一化解决GRPO信号坍塌问题,提升RL对齐效果

前置知识

GRPO (Group Relative Policy Optimization)

GRPO是一种用于大语言模型强化学习训练的策略优化算法,由DeepSeek团队提出。与传统的PPO不同,GRPO不需要单独训练一个价值网络(critic model),而是通过组内相对比较来估计优势函数。具体做法是:对同一个问题采样一组(group)回答,计算每个回答的奖励,然后在组内进行标准化(减去均值、除以标准差),得到相对优势值。这种方法简单高效,已成为当前LLM RL训练的主流选择。

本文的核心贡献是对GRPO在多奖励场景下的改进,因此理解GRPO的基本原理是读懂本文的前提。

多奖励强化学习 (Multi-reward RL)

在LLM对齐训练中,单一奖励往往无法全面衡量模型输出质量。多奖励RL通过同时优化多个奖励信号来引导模型行为,例如同时优化回答的正确性、格式合规性、长度约束、安全性等。每个奖励捕获不同的用户偏好维度,共同指导模型产生更符合人类期望的行为。

本文正是针对多奖励RL场景中GRPO的局限性提出改进,理解多奖励RL的动机和挑战是理解本文贡献的关键。

优势函数 (Advantage Function)

优势函数 $A(s,a) = Q(s,a) - V(s)$ 衡量在状态 $s$ 下采取动作 $a$ 相比平均水平的好坏程度。在LLM RL中,优势函数用于指导策略更新:正优势的动作会被强化,负优势的动作会被抑制。在GRPO中,优势函数通过组内奖励标准化来近似计算。

本文发现GRPO在多奖励场景下的优势函数计算存在信息压缩问题,这是理解论文核心问题的关键。

奖励坍塌 (Reward Collapse)

本文提出的新概念,指在GRPO的多奖励设置中,不同的奖励组合在组内标准化后会被映射到相同的优势值,导致训练信号的区分度下降。例如,奖励组合 (0,1) 和 (0,2) 在标准化后都变成相同的优势值 (−0.7071, 0.7071),尽管它们代表不同的模型表现水平。

这是本文发现的核心问题,GDPO的提出正是为了解决这一奖励坍塌现象。

研究动机

随着大语言模型能力的提升,用户期望模型不仅能提供准确的回答,还能在多种场景下展现出符合多元人类偏好的行为。为实现这一目标,强化学习流水线开始引入多个奖励信号,每个奖励捕获不同的偏好维度。然而,近期工作在多奖励设置中默认使用Group Relative Policy Optimization (GRPO)作为优化方法,却未检验其在多奖励场景下的适用性。本文揭示了一个关键问题:直接将GRPO应用于多个奖励的求和会导致奖励信号坍塌。具体来说,当使用两个二值奖励 $r_1, r_2 \in \{0,1\}$ 采样两个rollout时,总奖励可能取值 $\{0, 1, 2\}$。虽然理论上存在6种不同的奖励组合,但经过GRPO的组内标准化后,仅产生2个不同的优势值组。例如,奖励组合 (0,1)、(0,2) 和 (1,2) 都被映射到相同的优势值 (−0.7071, 0.7071),而 (0,0)、(1,1) 和 (2,2) 都映射到 (0,0)。这种过度压缩消除了奖励维度间的重要区分,导致不准确的策略更新、次优的奖励收敛,甚至在某些情况下导致训练早期失败。

本文的目标是本文的具体目标是提出一种新的策略优化方法,能够在多奖励强化学习设置中更好地保留不同奖励组合之间的区分度,产生更精确的优势估计,从而实现更准确的多奖励优化和更稳定的训练过程。作者希望该方法能够广泛适用于工具调用、数学推理、代码推理等多种任务场景,并在不同数量的奖励信号下都表现出色。

与已有工作不同的是,本文的独特切入角度是从优势函数计算的信息论视角分析GRPO在多奖励场景下的根本缺陷。与以往专注于设计新奖励函数的工作不同,本文首次系统性地揭示了GRPO的组内标准化机制在多奖励设置中会导致信息丢失。作者通过理论分析和实验验证,证明了即使移除GRPO中的标准差归一化项(如Dr.GRPO和DeepSeek-v3.2的做法),也只能略微增加不同优势值组的数量,无法根本解决问题。基于这一洞察,本文提出了通过解耦各奖励的归一化过程来保留跨奖励差异的创新方案。

核心方法

GDPO的核心思想是将多奖励优化中的归一化过程解耦,即对每个奖励分别进行组内标准化,然后再聚合。直觉上,如果把GRPO比作先把所有科目的成绩加起来再算排名,那么GDPO就是先对每个科目单独排名,再综合排名。这样做的好处是保留了每个奖励维度内的相对差异,避免了因直接求和导致的信息压缩。具体技术路线是:首先对每个奖励 $r_k$ 在组内独立计算标准化优势 $A_k$,然后将所有奖励的标准化优势加权求和得到 $A_{sum}$,最后对 $A_{sum}$ 进行批次级别的标准化以保持数值稳定性。

GDPO与已有方法的本质区别在于归一化的位置和粒度。GRPO是先将所有奖励求和 $r_{sum} = r_1 + \cdots + r_n$,再对 $r_{sum}$ 做组内标准化,这导致不同奖励组合被压缩到相同的优势值。GDPO则反转了这一顺序:先对每个奖励 $r_k$ 独立做组内标准化得到 $A_k$,再将标准化后的优势相加。这种解耦归一化的设计确保了每个奖励维度的相对差异被完整保留。论文通过图2的示例清楚展示了这一差异:在两个二值奖励、两个rollout的设置下,GRPO只产生2个不同的优势值组,而GDPO产生3个,更好地反映了奖励组合的真实差异。

方法步骤详情

GDPO的方法包含三个主要步骤。第一步是解耦的组内标准化:对每个问题 $i$ 的第 $k$ 个奖励,计算其在组内的标准化优势 $$A_{k}^{(i,j)} = \frac{r_{k}^{(i,j)} - \text{mean}\{r_{k}^{(i,1)}, \ldots, r_{k}^{(i,G)}\}}{\text{std}\{r_{k}^{(i,1)}, \ldots, r_{k}^{(i,G)}\}}$$ 其中 $G$ 是组大小。第二步是加权聚合:将各奖励的标准化优势加权求和 $$A_{sum}^{(i,j)} = w_1 A_1^{(i,j)} + \cdots + w_n A_n^{(i,j)}$$ 权重 $w_k$ 用于编码不同目标的优先级。第三步是批次级标准化:对聚合后的优势进行批次级别的标准化 $$\hat{A}_{sum}^{(i,j)} = \frac{A_{sum}^{(i,j)} - \text{mean}\{A_{sum}^{(i',j')}\}}{\text{std}\{A_{sum}^{(i',j')\} + \epsilon}$$ 确保优势的数值范围不随奖励数量增加而增长,提高训练稳定性。最后使用标准的PPO-clip目标函数进行策略更新。

技术新颖性

GDPO的技术新颖性体现在多个层面。首先,它首次从信息论角度系统分析了GRPO在多奖励场景下的信号坍塌问题,这是以往工作未曾关注的。其次,解耦归一化的思想虽然简单,但非常有效,且理论上有明确的解释:通过对每个奖励独立标准化,GDPO保留了 $O(G^n)$ 级别的不同优势值组,而GRPO只有 $O(G)$ 级别。论文的图3量化了这一差异:随着rollout数量或奖励数量增加,GDPO产生的好处值组数量远超GRPO和GRPO w/o std。此外,论文还系统研究了奖励优先级调整策略,提出了通过条件化奖励函数来解决奖励难度差异导致的优化偏差问题,这在实际应用中具有重要价值。

GDPO方法概览和奖励趋势对比
Figure 1: GDPO方法概览和奖励趋势对比
不同rollout数量和奖励数量下的优势值组数量对比
Figure 3: 不同rollout数量和奖励数量下的优势值组数量对比
不同长度奖励权重下的准确率和长度超限率
Figure 6: 不同长度奖励权重下的准确率和长度超限率
使用条件化长度奖励时的训练曲线
Figure 7: 使用条件化长度奖励时的训练曲线

实验结果

本文在三类任务上进行了全面实验,结果一致表明GDPO优于GRPO。在工具调用任务中,使用Qwen2.5-Instruct-1.5B训练时,GDPO相比GRPO在Live任务上提升约5%,在Non-Live任务上提升约3%,整体平均准确率提升2.7%(从30.18%到32.81%),正确格式率提升超过4%(从76.33%到80.66%)。在数学推理任务中,GDPO在AIME基准上为DeepSeek-R1-1.5B和Qwen3-4B-Instruct分别带来6.3%和2.3%的准确率提升。特别值得注意的是,GRPO在训练约400步后出现正确性奖励下降的训练崩溃现象,而GDPO持续改善。在代码推理任务中,GDPO在三个奖励设置下(pass率、长度约束、bug率)在所有基准上都实现了更好的平衡,例如在CodeContests上将pass率提升2.6%的同时bug率从14.1%降至13.2%。论文还发现,简单移除GRPO的标准差归一化项(GRPO w/o std)虽然略微增加了优势值组数量,但在格式奖励上完全失败,正确格式率降至0%,表明这种修改会引入训练不稳定性。

GDPO与GRPO在工具调用任务上的BFCL-v3评估结果
Table 1: GDPO与GRPO在工具调用任务上的BFCL-v3评估结果
GRPO、GRPO w/o std和GDPO在工具调用任务上的对比
Table 2: GRPO、GRPO w/o std和GDPO在工具调用任务上的对比
GRPO与GDPO在数学推理基准上的对比
Table 3: GRPO与GDPO在数学推理基准上的对比
条件化长度奖励的效果对比
Table 4: 条件化长度奖励的效果对比
GRPO与GDPO在代码推理基准上的对比
Table 5: GRPO与GDPO在代码推理基准上的对比
工具调用任务上GDPO、GRPO、GRPO w/o std的奖励曲线对比
Figure 4: 工具调用任务上GDPO、GRPO、GRPO w/o std的奖励曲线对比
DeepSeek-R1-1.5B数学推理任务上的训练行为对比
Figure 5: DeepSeek-R1-1.5B数学推理任务上的训练行为对比
查看结构化数据
任务指标本文基线提升
工具调用 (BFCL-v3) 平均准确率 / 正确格式率 GDPO: 32.81% / 80.66% (1.5B), 40.87% / 82.23% (3B) GRPO: 30.18% / 76.33% (1.5B), 39.20% / 81.64% (3B) +2.63% / +4.33% (1.5B), +1.67% / +0.59% (3B)
数学推理 (AIME) Pass@1准确率 / 长度超限率 GDPO: 29.4% / 6.5% (1.5B), 53.1% / 0.2% (7B), 56.9% / 0.1% (4B) GRPO: 23.1% / 10.8% (1.5B), 50.2% / 2.1% (7B), 54.6% / 2.5% (4B) +6.3% / -4.3% (1.5B), +2.9% / -1.9% (7B), +2.3% / -2.4% (4B)
代码推理 (CodeContests) Pass率 / Bug率 GDPO2-obj: 65.8% / 13.2%, GDPO3-obj: 65.6% / 2.5% GRPO2-obj: 63.2% / 14.1%, GRPO3-obj: 65.6% / 3.9% +2.6% / -0.9% (2obj), +0% / -1.4% (3obj)
数学推理 (MATH) Pass@1准确率 GDPO: 86.2% (1.5B), 93.9% (7B), 93.9% (4B) GRPO: 83.6% (1.5B), 94.1% (7B), 93.9% (4B) +2.6% (1.5B), -0.2% (7B), 0% (4B)

局限与改进

本文存在若干局限性。首先,实验主要在1.5B到7B规模的模型上进行,未验证在更大规模模型(如70B、400B+)上的效果,考虑到大规模模型的训练成本极高,GDPO的可扩展性仍需进一步验证。其次,论文仅探索了最多三个奖励的设置,在更多奖励维度(如安全性、公平性、风格等)下的表现尚不清楚。第三,GDPO需要对每个奖励单独维护组内统计量,计算开销略高于GRPO,尽管论文未详细分析这一额外成本。第四,实验中的奖励函数都是基于规则的,未探索使用学习到的奖励模型(reward model)时的表现,而实际应用中复杂的偏好往往难以用规则精确表达。最后,论文承认在某些情况下(如Qwen3-4B-Instruct在MATH上),GDPO与GRPO的差距很小,说明GDPO的优势在奖励差异明显时更为显著。

独立分析的弱点

本文存在几个值得改进的弱点。第一,理论分析较为初步,仅通过枚举小规模示例说明奖励坍塌问题,缺乏对GDPO优势值组数量的更严格数学刻画,例如给出不同奖励数量 $n$ 和组大小 $G$ 下GDPO与GRPO优势值组数量的精确公式。第二,批次级标准化的引入虽然提高了训练稳定性,但可能引入跨批次的不一致性,特别是在奖励分布随训练变化时,论文未深入分析这一潜在问题。第三,奖励优先级调整部分的实验主要在DeepSeek-R1-7B上进行,未验证在其他模型架构上的泛化性。第四,条件化奖励函数(如将长度奖励条件化在正确性上)虽然有效,但这种设计需要人工根据任务特点进行调整,缺乏自动化的优先级适应机制。第五,论文未提供GDPO与更广泛的基线方法(如PPO、ReMax、RLOO等)的比较,限制了对GDPO相对优势的全面理解。

未来方向

基于本文成果,未来研究可以在多个方向展开。首先,可以探索GDPO在更大规模模型和更多奖励维度下的效果,特别是在同时优化正确性、安全性、公平性、风格等多个维度时的表现。其次,可以研究自动化的奖励优先级适应机制,例如通过元学习或在线调整来动态确定奖励权重和条件化策略,减少人工设计的需求。第三,将GDPO与学习到的奖励模型结合使用是一个重要方向,因为实际应用中的偏好往往是隐式的、难以用规则表达的。第四,可以探索GDPO在其他RL算法框架(如PPO、ReMax)中的应用,验证解耦归一化的思想是否具有更广泛的适用性。第五,研究GDPO在在线学习和持续学习场景下的表现,特别是在奖励分布随时间变化时的适应能力。

复现评估

本文的复现性较好。作者在论文中声明已在HF-TRL、verl和Nemo-RL三个框架中实现了GDPO,这大大降低了复现门槛。论文提供了详细的超参数配置(见附录D和E),包括batch size(512)、rollout数量(4或16)、最大响应长度(1024或8000)、学习率等关键设置。训练数据方面,工具调用任务使用公开的ToolACE、Hammar和xLAM数据集,数学推理使用DeepScaleR-Preview数据集,代码推理使用Eurus-2-RL数据集,均为公开可用。评估基准(BFCL-v3、AIME、MATH等)也都是标准测试集。算力需求方面,论文未明确说明训练所需的GPU资源,但基于使用的模型规模(1.5B-7B)和训练步数(100-500步),估计需要多块A100或H100 GPU。总体而言,具备相应算力的研究团队应该能够较为顺利地复现本文结果。