← 返回 2026-02-11

iGRPO:自我反馈驱动的LLM推理 iGRPO: Self-Feedback-Driven LLM Reasoning

Ali Hatamizadeh, Shrimai Prabhumoye, Igor Gitman, Ximing Lu, Seungju Han, Wei Ping, Yejin Choi, Jan Kautz 📅 2026-02-09 👍 19 2026-07-13 08:35
GRPO扩展 强化学习 推理优化 数学推理 策略优化 自我反馈

通过两阶段迭代自我反馈机制提升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通过延迟熵崩溃改变了学习动态,在训练中期维持更高的探索性,从而避免了过早的模式坍缩。

Iterative GRPO (iGRPO): During Exploratory Draft Generation, the model selects a high-scoring 'best draft' from initial samples and appends it to the prompt for Conditioned Refinement.
Figure 1: Iterative GRPO (iGRPO): During Exploratory Draft Generation, the model selects a high-scoring 'best draft' from initial samples and appends it to the prompt for Conditioned Refinement.
Entropy dynamics. Per-token policy entropy during training.
Figure 3: Entropy dynamics. Per-token policy entropy during training.

实验结果

本文的核心发现在多个层面上验证了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延迟了过早的熵崩溃(训练中期熵值更高,最终收敛到相似水平)。

Performance comparison of 7B, 8B, and 14B models across multiple mathematical reasoning benchmarks.
Table 1: Performance comparison of 7B, 8B, and 14B models across multiple mathematical reasoning benchmarks.
Beyond GRPO: adding the same self-feedback refinement layer to other methods consistently improves the average Pass@1.
Table 2: Beyond GRPO: adding the same self-feedback refinement layer to other methods consistently improves the average Pass@1.
Effect of replacing the rule-based reward with a GPT-5 judge in iGRPO.
Table 3: Effect of replacing the rule-based reward with a GPT-5 judge in iGRPO.
Pass@1 results for OpenReasoning-Nemotron-7B with and without iGRPO.
Figure 2: Pass@1 results for OpenReasoning-Nemotron-7B with and without 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的研究者应该能够顺利复现。