简介:近期行业前沿在长程任务中重新采用PPO算法,引发了对大模型强化学习策略选型的广泛探讨。本文以早期经典RLHF常用的PPO与某创新架构采用的GRPO为切入点,深入剖析两者的数学模型、架构差异与核心优缺点,帮助开发者与技术架构师在不同业务场景下做出合理的算法选型与迁移决策。
在大语言模型(LLM)的对齐与强化学习微调(RLHF)阶段,算法的选择直接决定了模型在复杂指令遵循和长程推理任务中的表现。近期,行业内某前沿模型在长程任务中未采用近期备受关注的GRPO,而是回归了早期的PPO算法,这一决策引起了技术社区的广泛讨论。为了探寻这背后的技术逻辑,本文将对经典PPO与当前备受关注的GRPO进行深度对比分析,从问题建模、损失函数、架构差异到适用场景进行全面拆解,帮助技术团队在模型训练方案选型时提供客观依据。
在RLHF的工程实践中,如何高效、稳定地更新策略模型一直是核心难题。本文对比的对象是两种主流的强化学习优化算法:PPO(近端策略优化)与GRPO(分组相对策略优化)。
PPO算法:作为早期经典RLHF流程(如业界知名模型早期版本)的标准配置,PPO通过引入价值网络来估计状态价值,从而计算优势函数,并配合裁剪机制和KL惩罚来约束策略更新步长。
GRPO算法:作为近期被某创新大模型架构广泛采用的策略优化方法,GRPO旨在降低训练的资源开销与工程复杂度,其核心思路是通过组内采样和相对比较来估计优势,省去了庞大的价值网络。
尽管两者在架构实现上有显著差异,但它们在大模型对齐的技术目标和底层逻辑上存在诸多共同点:
理解两者的差异是进行技术选型的关键。我们从架构、损失函数、计算成本等维度展开深入分析。
PPO的核心特征是依赖一个额外的价值网络来估计基线,从而计算优势函数。在PPO的问题建模中,语言模型策略为$\pi\theta(y{t+1} \mid x, y{\leq t})$,状态定义为$s_t=(x, y{t})$。PPO需要训练一个价值网络$V_\phi(s_t)$来逼近真实价值,进而计算优势$A_t = R_t - V(S_t)$。
GRPO则放弃了价值网络的显式估计。它通过对同一个Prompt进行多次采样生成一组回复,利用这组回复的相对奖励(如减去组内均值)来作为优势的估计。这种基于组内相对排序的方法从根本上消除了价值网络引入的估计偏差。
原版PPO的常见最终损失函数是一个复合结构,完整公式如下:
$\mathcal{L}{\text{PPO}}(\theta) = \mathcal{L}{\text{policy}}(\theta) + cv \mathcal{L}{\text{value}}(\theta) - ce \mathcal{L}{\text{entropy}}(\theta) + \beta \mathcal{L}_{\text{KL}}(\theta)$
其中,策略损失部分旨在让模型生成高优势的token,公式为:
$\mathcal{L}_{\text{policy}}(\theta) = - \mathbb{E}_t \left[ \min \left( r_t(\theta) A_t, \operatorname{clip}(r_t(\theta), 1-\epsilon, 1+\epsilon) A_t \right) \right]$
逐token的优势函数计算依赖于指数衰减奖励:
$Rt = \sum{k=t}^{T} \gamma^{k-t} r_k, \quad A_t=R_t - V(S_t)$
GRPO的损失函数在形式上保留了类似的策略损失裁剪项,但其核心区别在于$A_t$的来源。GRPO的优势函数直接来源于组内归一化后的奖励,而不需要经过价值网络反向传播计算。这使得GRPO的梯度更新路径更短,数值稳定性更好。
这是两者在工程实践中最显著的差异:
| 对比维度 | PPO算法 | GRPO算法 |
|---|---|---|
| 核心架构 | 策略网络 + 价值网络 + 奖励网络 + 参考网络 | 策略网络 + 奖励网络 + 参考网络 |
| 优势估计 | 依赖价值网络$V(s)$减去基线计算 | 依赖同一Prompt下多次采样的组内相对奖励计算 |
| 工程复杂度 | 高,需平衡策略与价值网络的学习率及收敛步调 | 较低,无需训练价值网络,调参变量减少 |
| 计算与显存开销 | 极高,额外需要价值网络的显存与反向传播算力 | 较低,但需要增加单Prompt的采样批次大小 |
| 长程任务表现 | 较好,逐token的精细价值估计有助于分配信用 | 存在挑战,组内相对优势在极长上下文中可能信噪比降低 |
| 数值稳定性 | 易受价值网络估计误差影响,训练容易波动 | 相对稳定,基于排序的优势减少了绝对偏差的影响 |
1. 适用PPO的场景
如果您的业务场景是长程推理任务(如多步代码生成、复杂数学证明、长文本连贯创作),PPO可能更为合适。在长程任务中,奖励往往非常稀疏(只在最后给出一个整体评价)。PPO通过价值网络可以对中间步骤进行逐token的信用分配,精确指导模型在长序列中哪些动作是关键的。
2. 适用GRPO的场景
如果您的团队算力资源有限,且主要任务是短文本生成、对话对齐或常规指令遵循,GRPO是更优的选择。它通过增加采样数量换取对策略梯度的准确估计,不仅降低了显存门槛,还能加快迭代速度。在团队缺乏深厚强化学习调参经验时,优先考虑GRPO可以大幅降低项目失败风险。
如果技术团队计划从PPO方案迁移至GRPO,或因长程任务需求从GRPO回退至PPO,需要重点评估以下问题:
PPO与GRPO并非绝对的替代关系,而是大模型强化学习在不同工程约束下的技术选择。PPO凭借价值网络在长程与复杂信用分配任务中占据精度优势,但工程极其沉重;GRPO通过组内相对优势估计,极大地平民化了RLHF流程,但在稀疏奖励的长序列任务中可能面临信噪比下降的挑战。技术选型没有银弹,理解算法背后的数学直觉与系统架构边界,结合自身业务的上下文长度、算力规模和团队工程能力,才能做出最合理的决策。