iGRPO:自我反馈驱动的LLM推理 iGRPO: Self-Feedback-Driven LLM Reasoning
通过两阶段迭代自我反馈机制提升LLM数学推理能力的强化学习算法
前置知识
GRPO(组相对策略优化)
GRPO是DeepSeek-Math提出的一种无价值函数的策略优化算法,是PPO的高效变体。它通过组内相对奖励计算优势值,避免了训练单独critic模型的开销。具体来说,给定一个提示,GRPO采样G个候选输出,计算每个输出的奖励分数,然后在组内进行归一化得到优势值:$$\hat{A}_i = \frac{R_i - \text{mean}(\{R_1,...,R_G\})}{\text{std}(\{R_1,...,R_G\})}$$。这种设计保持了PPO的稳定性(通过重要性采样和裁剪机制),同时显著降低了计算成本。
iGRPO是GRPO的直接扩展,理解GRPO的组内归一化优势估计和裁剪目标函数是理解本文创新的基础
强化学习用于推理
近年来,强化学习被广泛应用于提升大语言模型的推理能力。典型的流程是:给定数学问题作为提示,让LLM生成解决方案,然后通过规则验证(如答案是否正确)或奖励模型评估输出质量,最后用RL算法更新模型参数。DeepSeek-R1提出的'零配置'方法直接在基础模型上进行RL训练,无需先进行监督微调,这启发了大量后续工作。
本文的方法属于这一范式,通过改进RL算法本身来提升推理性能
自我反馈与迭代改进
人类在解决复杂问题时很少一次性完成,而是通过迭代草稿、识别错误、基于内部反馈进行修正。自我反馈机制利用模型自身的输出作为改进信号,让LLM能够'反思'并优化自己的解决方案。现有方法如Self-Rewarding Language Models、Reflexion等尝试实现这一思路,但如何将其有效融入RL训练流程仍是开放问题。
iGRPO的核心创新就是将自我反馈机制系统性地融入GRPO训练,填补了人类问题解决方式与LLM训练之间的差距
研究动机
当前的强化学习算法在训练LLM进行复杂推理时存在根本性缺陷:它们不包含任何形式的自我反馈或反思机制。现有的GRPO、PPO等方法将每次生成视为独立事件,忽略了模型自身生成过程中的潜在有价值信息。这与人类解决问题的方式形成鲜明对比——人类很少一次性解决非平凡问题,通常会迭代初始草稿、识别错误并基于内部反馈进行修正(Flower & Hayes, 1981; Simon, 2012; Polya, 2014)。虽然有证据表明自我反馈机制可以增强多步推理和错误纠正能力(Madaan et al., 2023; Shinn et al., 2023),但现有RL框架未能充分利用这一迭代改进过程,在人类自然问题解决方式与LLM典型训练方式之间留下了关键差距。
本文的目标是本文旨在填补这一差距,提出迭代组相对策略优化(iGRPO),一种通过动态自我条件化实现引导式策略改进的强化学习算法。具体目标包括:1)在保持GRPO计算效率的同时引入自我反馈机制;2)创建一个闭合的改进循环,让策略能够系统性地改进自身的最佳尝试;3)在匹配的rollout预算下验证方法的有效性,确保额外的计算开销可忽略不计;4)在多个模型规模和架构上展示一致的性能提升。
与已有工作不同的是,本文的独特切入角度在于'动态自我条件化'的概念。与传统的上下文学习(ICL)使用固定的静态示例不同,iGRPO的条件信号是由策略自身生成的,并与学习过程共同进化。具体来说,iGRPO通过两阶段机制实现:Stage 1进行探索性草稿生成并选择最高奖励的草稿,Stage 2基于这个'最佳草稿'进行条件化精炼。这种设计创建了一个引导式学习动态:随着策略改进,草稿质量也提高,从而为后续迭代提供越来越有信息量的条件化信号。与现有的Self-Verification和Critique-GRPO等方法不同,iGRPO不需要模型生成额外的验证或评论文本,而是直接将最佳草稿作为高信号支架,训练策略在此基础上进行改进。
核心方法
iGRPO的方法整体思路可以概括为'探索-选择-精炼'的两阶段闭环。直觉上,这类似于人类解题时先快速写出一个初步解法(草稿),然后检查这个解法是否正确,最后基于这个初步尝试进行改进。技术路线上,Stage 1从当前策略采样N个候选草稿,使用相同的标量奖励信号选择最高奖励的草稿作为'自我反馈';Stage 2将这个最佳草稿附加到原始提示后,形成增强提示,然后在此基础上采样G个精炼输出,进行标准的GRPO风格更新。关键洞察是:通过将探索(Stage 1)与条件化利用(Stage 2)在每个优化步骤中耦合,模型获得了可泛化的自我改进能力,这种能力在训练中会累积增强。
iGRPO的核心创新是'动态自我条件化',这与已有方法有本质区别。传统的上下文学习使用固定的静态示例作为条件信号,这个信号在整个训练过程中保持不变,无法适应不断进化的策略能力。而iGRPO引入了不同的范式:条件信号是由策略自身生成的,并与学习共同进化。数学上,标准GRPO目标$$\mathcal{J}_{GRPO}(\theta)$$将每次生成视为独立;而iGRPO的目标为$$\mathcal{J}_{iGRPO}(\theta) = \mathbb{E}_{q \sim P(Q)} \mathbb{E}_{o \sim \pi_\theta(\cdot|q'_\theta(q))}[R_\phi(o)]$$,其中$$q'_\theta(q) = \text{Concat}(q, \hat{d}_\theta(q))$$,$$\hat{d}_\theta(q)$$是当前策略生成的最佳草稿。这创建了一个耦合动力系统:随着$$\pi_\theta$$改进,$$\hat{d}_\theta$$的质量也提高,从而为后续生成提供越来越有信息量的条件化信号。与Self-Verification和Critique-GRPO等方法不同,iGRPO不需要模型执行额外的验证或评论任务,而是保持生成过程与奖励信号的清晰分离,同时仍然利用从模型自身输出学习的自我改进原则。
方法步骤详情
iGRPO方法的具体步骤如下:首先,对于每个训练提示$$q$$,Stage 1从当前策略快照$$\pi_{\theta_{old}}$$采样N个候选草稿:$$d_i \sim \pi_{\theta_{old}}(\cdot | q), i=1,...,N$$。然后使用奖励函数$$R_\phi$$评估每个草稿,选择最高奖励的草稿:$$\hat{d} = \arg\max_{i \in \{1,...,N\}} R_\phi(d_i)$$。接下来,Stage 2构造增强提示$$q' = \text{Concat}(q, \hat{d})$$,即在原始提示后附加最佳草稿。然后从$$\pi_{\theta_{old}}$$采样G个精炼输出:$$o_j \sim \pi_{\theta_{old}}(\cdot | q'), j=1,...,G$$。这些输出获得奖励后,通过组内归一化计算优势值:$$\hat{A}_j = \frac{R_\phi(o_j) - \text{mean}(\{R_\phi(o_1),...,R_\phi(o_G)\})}{\text{std}(\{R_\phi(o_1),...,R_\phi(o_G)\})}$$。最后,使用裁剪的替代目标和KL惩罚仅对Stage 2的输出进行策略更新。关键的是,Stage 1不产生梯度,仅作为自适应探索机制塑造优化景观。
技术新颖性
iGRPO的技术新颖性体现在几个方面。首先,它引入了'动态自我条件化'的概念,将自我反馈系统性地融入基于组的RL优化中,这在现有框架中是缺失的。其次,iGRPO建立了'引导式策略改进'的理论框架:在二元奖励下,所选最佳草稿的期望奖励为$$\mathbb{E}[R_\phi(\hat{d}_\theta(q))] = 1 - (1 - V_\theta(q))^N$$,其中$$V_\theta(q)$$是策略的成功概率,这单调递增于$$V_\theta(q)$$,从而建立了引导效应的数学保证。第三,iGRPO的两阶段设计是正交于现有GRPO改进的:它改变的是优化器看到的数据分布,而不是底层的优化目标本身,这使得它可以与DAPO、GSPO等其他变体结合使用。最后,iGRPO通过延迟熵崩溃改变了学习动态,在训练中期维持更高的探索性,从而避免了过早的模式坍缩。
实验结果
本文的核心发现在多个层面上验证了iGRPO的有效性。首先,在控制实验中(匹配的rollout预算),iGRPO在所有模型规模(7B、8B、14B)和架构上一致优于vanilla GRPO。对于Nemotron-H-8B-Base-8K,GRPO将宏平均从29.65%提升到41.08%,而iGRPO进一步提升到45.04%,比GRPO高3.96个百分点,比最强的自我改进基线(Critique-GRPO的43.39%)高1.65个百分点。对于DeepSeek-R1-Distill-Qwen-7B,iGRPO达到69.87%的平均准确率,比GRPO的68.29%高1.58个百分点。在14B模型上,DeepSeek-R1-Distill-Qwen-14B的宏平均从71.29%提升到73.02%,OpenMath-Nemotron-14B从76.73%提升到78.00%。其次,在更难的泛化设置中,使用OpenReasoning-Nemotron-7B在AceReason-Math上训练,iGRPO在AIME24和AIME25上分别达到85.62%和79.64%的新SOTA,并且将收益转移到数学之外的任务(GPQA +1.84%,MMLU-Pro +0.91%)。第三,消融研究表明iGRPO的精炼接口可以推广到其他group-based PPO变体(DAPO平均提升+1.19%,GSPO +1.11%),与生成式评判器兼容(GPT-5评判器平均提升+0.94%),并且通过熵分析显示iGRPO延迟了过早的熵崩溃(训练中期熵值更高,最终收敛到相似水平)。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| AIME25(数学竞赛) | Pass@1准确率(64次平均) | iGRPO: 40.16% (7B), 65.57% (14B), 79.64% (SOTA) | GRPO: 38.90% (7B), 64.53% (14B) | +1.26% (7B), +1.04% (14B), +1.78% (SOTA vs GRPO基线) |
| AIME24(数学竞赛) | Pass@1准确率(64次平均) | iGRPO: 56.30% (7B), 76.72% (14B), 85.62% (SOTA) | GRPO: 55.00% (7B), 74.79% (14B) | +1.30% (7B), +1.93% (14B), +1.52% (SOTA vs GRPO基线) |
| MATH500(数学问题) | Pass@1准确率 | iGRPO: 93.80% (7B), 96.70% (14B) | GRPO: 93.25% (7B), 96.00% (14B) | +0.55% (7B), +0.70% (14B) |
| AMC23(数学竞赛) | Pass@1准确率 | iGRPO: 95.00% (7B), 97.50% (14B) | GRPO: 90.00% (7B), 95.00% (14B) | +5.00% (7B), +2.50% (14B) |
| GSM8K(小学数学) | Pass@1准确率 | iGRPO: 92.42% (7B), 94.69% (14B) | GRPO: 92.12% (7B), 94.40% (14B) | +0.30% (7B), +0.29% (14B) |
| Minerva Math(科学数学) | Pass@1准确率 | iGRPO: 41.54% (7B), 36.76% (14B) | GRPO: 40.44% (7B), 35.70% (14B) | +1.10% (7B), +1.06% (14B) |
局限与改进
尽管iGRPO取得了显著成果,但仍存在一些局限性。首先,虽然iGRPO的峰值内存使用与GRPO几乎相同(差异约0.0063 GB),但训练吞吐量有所下降(0.34 vs 0.41 samples/sec),整体训练时间增加约13%(94.1 vs 83.3 GPU小时)。这个开销源于两阶段的顺序解码,虽然不需要更多GPU或额外内存容量,但在大规模部署时仍需考虑。其次,本文的验证主要集中在数学推理任务上,对于其他领域(如代码生成、自然语言理解、多模态推理)的泛化能力尚未充分探索,尽管Figure 2显示了在GPQA和MMLU-Pro上的积极转移。第三,Stage 1的选择机制相对简单(仅选择最高奖励的草稿),可能错失从多个高质量草稿中学习的机会。第四,虽然熵分析显示iGRPO延迟了过早的熵崩溃,但具体的机制和最优的熵维持策略仍需进一步研究。最后,实验中使用的奖励函数是简单的二元规则奖励(答案正确性),对于更复杂的奖励信号(如过程奖励、生成式评判器)的兼容性和效果需要更全面的评估。
独立分析的弱点
从独立分析的角度,iGRPO存在几个可以改进的弱点。首先,Stage 1的选择策略是纯贪心的(选择最高奖励的草稿),这可能导致探索不足。改进方向包括:引入多样性感知的选择机制(例如选择多个高质量但不同的草稿),或者使用基于不确定性的选择(选择模型最不确定的草稿以增加学习信号)。其次,两阶段的计算分配是固定的(如N=8,G=8),但最优分配可能随训练阶段变化。可以研究自适应分配策略,根据训练进度动态调整Stage 1和Stage 2的预算比例。第三,当前方法在推理时退化为标准的单次生成,没有利用训练阶段学到的精炼能力。可以探索推理时的迭代精炼机制,例如多次采样并选择最佳输出。第四,虽然论文显示了与DAPO和GSPO的兼容性,但具体的结合方式(简单包装vs深度融合)和最优配置尚未充分研究。最后,对于长序列推理任务,当前的组内归一化可能受到序列长度的影响,需要考虑长度感知的优势估计。
未来方向
基于本文的成果,可以延伸出几个有前景的未来研究方向。首先,将iGRPO的自我反馈机制扩展到其他可验证任务,如代码生成(通过单元测试验证)、定理证明(通过形式化验证)、多步问答(通过事实一致性检查)。其次,研究更复杂的自我反馈形式,例如使用模型自身的置信度估计作为额外信号,或者引入自然语言的自我评论(结合Critique-GRPO的思想)。第三,探索多轮迭代精炼,即在Stage 2的输出上再次应用iGRPO,形成更深的改进层次。第四,研究iGRPO与其他训练技术的结合,如课程学习(逐步增加问题难度)、数据增强(从模型生成中扩充训练集)。第五,将动态自我条件化的思想扩展到多智能体设置,例如让多个模型互相提供反馈。最后,理论分析可以进一步深化,研究在非二元奖励、连续奖励空间下的引导效应,以及收敛速度和样本复杂度的理论保证。
复现评估
从复现评估的角度,本文提供了相对良好的复现条件。论文的代码和训练配置在GitHub上有开源实现,使用了标准的公开数据集(MATH的7,500个问题和AceReason-Math的9,400个问题)和公开可用的模型(DeepSeek-R1-Distill-Qwen-7B/14B、OpenMath-Nemotron-7B/14B、Nemotron-H-8B-Base-8K)。奖励函数是简单的规则奖励(答案正确性检查),易于实现。然而,复现需要显著的计算资源:7B模型需要2个节点(每节点8×NVIDIA A100 GPU),14B模型需要5个节点,总计约94.1 GPU小时完成一个完整训练。超参数设置在附录中有详细记录,包括学习率(1e-6)、批量大小(128)、温度(0.7)、完成长度(4096 tokens)等。评估框架使用NeMo-Skills,解码参数(温度0.6、top-p 0.95)也已明确。总体复现难度中等,主要挑战在于计算资源需求,但算法本身不涉及复杂的实现细节,熟悉GRPO/PPO的研究者应该能够顺利复现。
论文图表
该补充表格展示了OpenMath-Nemotron-14B在iGRPO增强后在大规模数据集OpenR1-Math-220k上的表现。iGRPO在所有基准上都有提升,AIME25从61.18%提升到66.04%(+4.86%),AIME24从73.28%提升到76.61%(+3.33%),平均从75.41%提升到78.24%。
验证iGRPO在更大规模训练数据上的有效性,展示其可扩展性
该补充表格列出了训练7B iGRPO模型的详细超参数配置,包括精度(bf16)、注意力实现(flash_attention_2)、vLLM设置、梯度累积步数(8)、学习率(1e-06)、调度器类型(cosine_with_min_lr)、预热比例(0.1)、训练轮数(1)、每设备批量大小(16)、最大完成长度(4096)、生成数量(4)、温度(0.7)和奖励函数(accuracy, format)。
提供完整的复现细节,帮助其他研究者重现论文结果
该补充表格对比了iGRPO和GRPO的计算资源使用。峰值内存:GRPO 54.9286 GB vs iGRPO 54.9349 GB(差异约0.0063 GB,可忽略)。吞吐量:GRPO 0.41 samples/sec vs iGRPO 0.34 samples/sec(下降约17%)。总GPU小时:GRPO 83.3 vs iGRPO 94.1(增加约13%)。
量化iGRPO的计算开销,证明其在资源受限环境中的可行性
该折线图展示了iOpenMath-Nemotron-14B在不同pass@N设置下的准确率变化。x轴是对数刻度的N值(1-256),y轴是准确率(%)。AIME24从pass@1的66.04%快速提升,在N=16时达到93.33%并饱和。AIME25从pass@1的66.04%持续提升,在N=256时达到96.67%,显示某些问题分布可能从更多采样尝试中受益。
展示iGRPO训练模型的多采样能力,证明即使top-1猜测有时不正确,模型也能在多次采样中生成正确解决方案
该双面板图展示了训练过程中的平均奖励和响应长度变化。左图显示iGRPO在整个训练过程中维持比GRPO更高的平均奖励。右图显示两者的响应长度几乎相同,GRPO平均略长。这表明iGRPO的改进来自响应质量的提升,而不是冗长性。
展示iGRPO的学习动态,证明其优势来自质量提升而非长度变化,排除了通过生成更长输出获得奖励的混淆因素