← 返回 2026-10-07

理性引导策略优化:以自适应提示支架学习推理 Rationale-Guided Policy Optimization: Learning to Reason with Adaptive Rationale Scaffolding

Hoang Phan, Minh Pham, Chau Pham, Chinmay Hegde, Trung Le, Qi Lei 📅 2026-10-05 👍 6 2026-11-11 18:30
LLM推理 RLVR 多模态推理 强化学习 自适应提示

把标准答案当临时脚手架,自适应提示加奖励门控蒸馏,破解RLVR奖励稀疏

前置知识

RLVR(可验证奖励强化学习)

用可自动验证的奖励信号(如检查模型最终答案是否与标准答案一致)对语言模型做策略梯度训练的范式。相比人类偏好奖励模型,验证器噪声小、成本低,能激发长链条推理行为。DeepSeek-R1 等工作证明大规模 RLVR 可以在不完全依赖监督推理轨迹的情况下学出复杂推理能力,已成为前沿模型后训练的关键组件。

RGPO 的整个框架建立在标准 RLVR 训练循环之上,其要解决的核心问题正是 RLVR 中的奖励稀疏;读懂本文必须先知道 RLVR 的奖励如何产生、优化什么。

GRPO(组相对策略优化)

对同一提示采样一组 $G$ 个回答,用组内归一化计算优势 $\hat{A}_i=(r_i-\text{mean}(\{r_j\}))/\text{std}(\{r_j\})$,以此替代 PPO 的价值网络基线,再优化 PPO 式裁剪目标。REINFORCE++ 则改用全局 batch 归一化。当某提示的组内全部回答同错或同对时,奖励无方差,优势信号为零。

RGPO 以 GRPO 或 REINFORCE++ 为主 RL 目标;论文对奖励稀疏的分析(全失败事件下组中心归一化的 GRPO 优势为零)正是基于 GRPO 的机制。

奖励稀疏与全失败事件

当训练题难度超出模型当前推理能力时,一个提示的全部 $K$ 条采样轨迹都解错,验证器给出一致的零奖励,组内无方差、无可学习的梯度信号,训练恰好 stagnate 在最需要提升的难题上。对小模型尤其严重:大量算力花在采样失败轨迹上却拿不到任何正例。

这是论文的出发点。RGPO 的引导式第二轮采样正是为了在全失败事件下以概率 $(1-p)^K[1-(1-s)^{K_g}]$ 挖掘出可用的正例轨迹。

提示泄露与捷径学习

若训练时始终把完整参考答案放在上下文里,模型会学会直接从提示中抄答案:输出变得极短、训练奖励虚高,但推理能力并未提升;评测时去掉提示性能立刻崩塌(论文模拟中跌到 1% 的随机水平),且策略梯度损失与梯度范数出现剧烈尖峰,优化不稳定。

RGPO 的两大设计——奖励门控、以及把辅助 SFT 条件在原始提示而非提示增强提示上——正是为了规避这一陷阱;理解捷径风险才能理解方法各项设计的动机。

混合 SFT–RL 训练

将离线策略数据(专家轨迹、拒绝采样结果、更强模型生成的解)以监督似然损失的形式混入 RL 目标,代表工作有 LUFFY(对离线轨迹做策略塑形)、CHORD(动态加权 SFT 辅助项)等。核心难点是演示分布偏离当前策略,直接最大化其似然容易导致模仿式记忆、分布失配与泛化差。

RGPO 属于该家族但机制不同:它的辅助样本由当前策略在提示引导下在线生成、经验证器奖励门控筛选、并条件在原始提示上蒸馏,是与离线轨迹模仿的本质分界。

研究动机

纯在线策略 RLVR 的有效性受限于当前策略自身能力:当训练题难度超出模型推理能力时,一个提示的全部 $K$ 条采样轨迹可能全错,奖励全为零且组内无方差(GRPO 的组相对优势直接归零),优化在最需要突破的难题上完全停滞,对小模型尤其严重。论文的受控模拟量化了这一点:无引导的 GRPO 在简单题测试集上可达 99.1%,但在难题测试集上训练三个 epoch 后仍是 0.0%。现有补救办法是引入外部引导——离线策略演示、专家轨迹、或从更强模型拒绝采样——但它们通常要求辅助数据与 RL 任务的最终答案格式一致,且直接最大化参考解似然会强迫模型模仿远离自身分布的轨迹,造成分布失配、记忆化和泛化差:模拟中直接对含系统性错误的参考解做 SFT 只得到 12.8% 的难题测试准确率;小模型(2B)在 Mulberry 式复杂模板下甚至难以优化。

本文的目标是本文目标是让训练在不要求离线数据匹配 RL 任务格式的前提下,充分利用训练集里现成的标准答案推理链(rationale),系统性地缓解奖励稀疏。具体包括三点:把参考解当作临时的『脚手架』而非直接的模仿目标——引导只用于帮助模型自己探索出正确轨迹;只把奖励更高、由模型自己生成的解传回无引导的原始任务分布;引导强度随模型当前能力逐题自适应调节——能独立解出的题减少或撤掉提示,反复失败的题逐步揭示更多参考解段落。最终希望在纯文本与视觉语言两类模型的推理后训练中,同时取得更快的训练收敛、更高的最终准确率,并保证无提示评测时的性能与健康的探索熵。

与已有工作不同的是,本文的独特切入有两条线。其一是认知科学动机:『辅助困境』指出帮助太少学习者卡住、太多则诱发浅层加工损害迁移;先独立尝试再受指导的『有效挣扎』能激活先验知识、帮助内化后续讲解;逐步淡出的解题范例优于永久支持。RGPO 把这套原则工程化为训练算法:先无引导采样、失败才加提示、成功即撤提示。其二是机制上的本质区别:LUFFY/CHORD 直接对离线参考轨迹做塑形或加权 SFT,GHPO 按解题能力调整引导量但仍依赖参考解格式匹配,QuestA 给难题拼接部分解,BREAD 从专家前缀分支采样,HINT 提供启发式提示并强调训练亲和性。RGPO 则把引导仅当作提议分布:引导下的新回答必须通过奖励门控(超过同题所有无引导采样的最高奖励加边际 $\delta$)才进入辅助缓冲区,且监督损失条件在原始提示 $x_i$ 而非提示增强的 $\tilde{x}_i$ 上,从而把引导收益迁移回部署时的无提示分布。

核心方法

直觉上 RGPO 模仿人类做题:先不看书自己解;解出则直接强化该回答;解不出才翻开参考解的一小段当提示,引导自己再独立写一遍完整解答,且只有当带提示的重写确实更好时才把功劳记在自己名下。技术路线是在标准 RLVR 循环上加两件事。其一,每 epoch 双轮采样:第一轮从原始提示 $x_i$ 无引导采样 $K$ 条用于主 RL 目标(GRPO 或 REINFORCE++,$\beta=0$);若该题有非空提示,再在引导提示 $\tilde{x}_i^{(t)}$(参考解前缀加验证器反馈)上采 $K_g$ 条。其二,奖励门控的辅助 SFT:引导回答仅当 $\tilde{r}_{i,k}>\max_j r_{i,j}+\delta$ 时进入缓冲区,损失为条件在原始提示上的 $\mathcal{L}_{\text{SFT}}=-\mathbb{E}[\log\pi_\theta(\tilde{y}|x_i)]$,总目标 $\mathcal{L}_{\text{total}}=\mathcal{L}_{\text{RL}}+\lambda_t\mathcal{L}_{\text{SFT}}$。

核心创新可概括为『引导即脚手架、门控定真伪、原题收功劳』三件套。与已有方法的本质区别在于:辅助训练信号不是外部参考解本身,而是当前策略在提示下生成、且被验证器确认更优的自己的轨迹,天然对齐策略分布。奖励门控确保错误参考解不污染训练——模拟中参考解对 8 类困难决策中的 3 类系统性出错,去门控消融立刻退化到与直接 SFT 相同的 12.8%,而直接 SFT 参考解也只有 12.8%。条件在原始提示 $x_i$ 而非 $\tilde{x}_i$ 上做蒸馏是防『提示依赖』的关键:若在 $\tilde{x}_i$ 上训练,模型学会的是依赖提示而非独立推理。自适应提示按题维护已揭示段数 $g_i^{(t)}$,且有负漂移保证 $\mathbb{E}[g_{t+1}-g_t|\mathcal{H}_t]=2(1-p_t)^K-1$,当成功率超过 $p^\star=1-2^{-1/K}$ 后提示量自然衰减回零。

方法步骤详情

第一步,设定与分段:训练集 $\mathcal{D}=\{(x_i,s_i,a_i)\}$ 含提示、参考推理链、标准答案与可验证奖励 $r(x,y,a)$;参考解切成 $N$ 段,初始 $g_i^{(0)}=0$。第二步,无引导首轮:估计可解性 $C_i^{(0)}=\mathbb{I}[\max_k r_{i,k}^{(0)}\ge\tau_{\text{correct}}]$,全错则下 epoch 揭示 $1/N$ 段。第三步,提示升降级:无正确轨迹 $g^{(t+1)}=\min(g+1,N)$,有则 $g^{(t+1)}=\max(g-1,0)$,自动趋向『恰好可学』的最小提示量。第四步,双轮采样:无引导轮供主 RL 损失;引导轮在$\tilde{x}_i^{(t)}$ 上采 $K_g$ 条。第五步,奖励门控:仅保留 $\tilde{r}_{i,k}>\max_j r_{i,j}+\delta$ 的引导回答入池。第六步,联合优化:主 RL 用组相对优势裁剪目标,辅助项条件在 $x_i$ 上、权重 $\lambda_t$。理论上 Proposition 1 给出$\mathcal{P}_{\text{RGPO}}=1-(1-p)^K(1-s)^{K_g}$,只要 $s>p$ 即优于纯采样;Theorem 2 证明到达阈值 $\tau$ 的期望时间为 $O(\frac{1}{p_0}\log\frac{1}{p_0})$,优于纯 GRPO 的 $\Omega(\frac{1}{p_0})$。

技术新颖性

技术新颖性体现在四个层面。第一,数据机制:辅助样本不是现成的离线数据,而是『引导提议—奖励筛选』在线产生的自生成轨迹,天然对齐当前策略分布,既规避了离线专家轨迹的分布失配,也不要求辅助数据匹配 RL 答案格式,更无需从更强模型拒绝采样。第二,训练信号:辅助 SFT 条件在原始提示上,把引导收益迁移回部署分布——与 Hint-conditioned RL(始终在有提示的上下文里做 RL,评测去掉提示即崩到 1%)和 full hint(学会抄答案捷径、输出极短)形成鲜明对照。第三,调度机制:逐题成败驱动的提示升降级而非固定课程或全局调度,并有 Proposition 1 的发现概率刻画与 Theorem 2 的样本复杂度界支撑,提示量在 $p_t>p^\star=1-2^{-1/K}$ 后具有可证的负漂移。第四,稳健性:因为从不训练参考解文本本身,RGPO 对推理模板不敏感——DeepSeek-R1 简洁模板下奖励平台期高于 GRPO 且熵轨迹相当,而 Mulberry 复杂模板下混合离线 SFT 的 AHPO 出现尖锐的熵坍缩,RGPO 则不受影响,这解释了其格式的无关性优势。

Overview of the proposed method: Inspired by how humans approach a problem, the model first tries to solve each problem without access to hints. When it succeeds, the correct response is directly reinforced. When it fails, progressively stronger rationales are introduced as temporary scaffolds to guide the model toward a correct solution.
Figure 1: Overview of the proposed method: Inspired by how humans approach a problem, the model first tries to solve each problem without access to hints. When it succeeds, the correct response is directly reinforced. When it fails, progressively stronger rationales are introduced as temporary scaffolds to guide the model toward a correct solution.

实验结果

三组实验。受控模拟:无引导 GRPO 简单题 99.1%、难题 0.0%;RGPO 难题测试 99.5% 且不损简单题;直接 SFT 含系统性错误的参考解仅 12.8%,去门控消融同样退化;全提示设定评测去提示后跌至 1%;全程平均仅 7.4 条引导回答被接受即可自举学习。语言模型(Qwen2.5-3B,3 epoch 对比 HINT 的 10 epoch):分布内平均 26.8,超 GRPO(19.7)7.1 分、超 HINT(21.8)5.0 分;MATH-500 55.2(+6.6),Minerva 22.1(+8.7),OlympiadBench 24.9,AIME24 avg@32 5.1;分布外平均 32.2 对 29.9(+2.3),ARC 51.2、GPQA-D 14.7、MMLU-Pro 30.8 均最高。视觉语言模型(Qwen2-VL-2B/7B):2B 上比 GRPO 少 2.95 倍步数追平基线且终值更高,比 REINFORCE++ 快 3.18 倍、终值 +5.3;7B 快 2.34 倍、终值高 14.1 分。MMStar-Math 70.1(基座 46.4,+23.7);MathVista-Math 73.65(基座 41.11,超 MM-Eureka 的 72.59),GEO/ALG/GPS 三子集最佳,TQA 70.35 提升有限。Figure 4 显示全提示训练奖励虚高、输出极短、梯度尖峰频发,RGPO 平滑且无提示验证更强。

Overall Performance Comparison of RGPO vs. Baselines. RGPO obtains the highest in-distribution average, improving over HINT by 5.0 points (26.8 vs. 21.8), and improves the out-of-distribution average by 2.3 points (32.2 vs. 29.9).
Table 1: Overall Performance Comparison of RGPO vs. Baselines. RGPO obtains the highest in-distribution average, improving over HINT by 5.0 points (26.8 vs. 21.8), and improves the out-of-distribution average by 2.3 points (32.2 vs. 29.9).
Performance comparison on MMStar-Math and the mathematical subset of MathVista
Table 2: Performance comparison on MMStar-Math and the mathematical subset of MathVista
RGPO accelerates and improves RLVR training. Across Qwen2-VL-2B and Qwen2-VL-7B models, RGPO reaches the final performance of RLVR baselines substantially faster and achieves higher final accuracy.
Figure 3: RGPO accelerates and improves RLVR training. Across Qwen2-VL-2B and Qwen2-VL-7B models, RGPO reaches the final performance of RLVR baselines substantially faster and achieves higher final accuracy.
Effect of failure-conditioned hinting versus always-on full hints: Training (solid lines) and validation (dashed lines) rewards. Full hints rapidly inflate training reward but generalize poorly, while RGPO learns more gradually and achieves stronger no-hint validation performance.
Figure 4: Effect of failure-conditioned hinting versus always-on full hints: Training (solid lines) and validation (dashed lines) rewards. Full hints rapidly inflate training reward but generalize poorly, while RGPO learns more gradually and achieves stronger no-hint validation performance.
Training dynamics under different reasoning prompt formats. Left: using the concise DeepSeek-R1-style format, RGPO achieves higher reward than GRPO while maintaining a comparable entropy trajectory throughout training. Right: under the original Mulberry-style template, AHPO eventually improves reward after learning the required output structure, but this improvement is accompanied by a sharp entropy collapse.
Figure 5: Training dynamics under different reasoning prompt formats. Left: using the concise DeepSeek-R1-style format, RGPO achieves higher reward than GRPO while maintaining a comparable entropy trajectory throughout training. Right: under the original Mulberry-style template, AHPO eventually improves reward after learning the required output structure, but this improvement is accompanied by a sharp entropy collapse.
查看结构化数据
任务指标本文基线提升
数学推理·语言模型分布内平均(AIME24/MATH-500/OlympiadBench/Minerva) 四基准平均分 26.8 GRPO 19.7;最强基线 HINT 21.8 +7.1 vs GRPO,+5.0 vs HINT
MATH-500(Qwen2.5-3B) pass@1 55.2 HINT 48.6 / GRPO 44.0 +6.6 vs HINT
Minerva Math(Qwen2.5-3B) pass@1 22.1 HINT 13.4 / GRPO 12.2 +8.7 vs HINT
分布外推理平均(ARC-C/GPQA-Diamond/MMLU-Pro) 三基准平均分 32.2 HINT 29.9 / GRPO 28.3 +2.3 vs HINT
MMStar-Math(Qwen2-VL-7B) 准确率 70.1 基座 Qwen2-VL-7B 46.4 / R1-VL-7B 68.4 +23.7 vs 基座
MathVista-Math 全集(Qwen2-VL-7B) 准确率 73.65 基座 41.11 / MM-Eureka 72.59 +32.54 vs 基座,+1.06 vs MM-Eureka
训练样本效率(Qwen2-VL-2B vs GRPO / REINFORCE++) 追平基线终值所需训练步数 分别减少 2.95 倍 / 3.18 倍步数 标准 GRPO / REINFORCE++ 同预算终值更高(REINFORCE++ 下 +5.3)

局限与改进

作者承认的成本与前提:引导轮需要额外 $K_g$ 次采样,虽然样本效率更高,但单步计算开销接近翻倍;方法假设训练集配有标准答案推理链和可自动验证的奖励,对无参考解或奖励不可验证的任务不直接适用;验证主要集中于数学推理类基准,AIME24 上 5.1 对 HINT 4.9 的提升其实很小。我自己的补充观察:模拟中参考解的错误是系统性的且比例已知(8 类困难决策中 3 类出错),真实数据里参考解噪声的分布未知,门控在更隐蔽错误下能否完全过滤尚待检验;分段粒度 $N$、门控边际 $\delta$、辅助权重 $\lambda_t$ 三个超参没有系统消融;SFT 项对输出多样性的长期影响只通过熵曲线间接观察;2B 模型在 Mulberry 模板下的失败说明方法仍偏好简洁模板生态;多模态训练集与 MathVista 存在域重叠,TQA 子集仅 70.35 的有限提升也暗示了泛化边界。

独立分析的弱点

第一,每步近双倍的采样开销:无引导 $K$ 条加引导 $K_g$ 条,实现中每个提示都要跑两轮推理,即使多数题并不需要引导。改进方向:仅对上一 epoch 失败的题触发引导轮、引导轮早停或用投机解码降本。第二,依赖高质量参考解:若参考解整体错误率高,接受池虽受门控保护,但『提示无用』时仍白花 $K_g$ 次采样。改进:用引导成功率在线估计提示质量,动态关闭低质量提示或改用教师模型生成候选推理链。第三,超参鲁棒性未知:$N$、$\delta$、$\lambda_t$ 均无敏感性分析,跨域使用需重新调参。改进:报告敏感性曲线,或把 $\lambda_t$ 设计成随接受池大小自适应的函数。第四,辅助 SFT 可能压缩策略多样性:训练中熵接近 GRPO,但更长训练或更大 $\lambda$ 下未验证。改进:对引导前策略加 KL 约束或定期回放无引导数据。第五,评测集中于数学与二元验证奖励:agentic、代码、长程任务未测。改进:扩展到过程奖励或可执行测试用例等连续奖励场景。

未来方向

作者提出的方向是把 RGPO 作为降低奖励稀疏、稳定 RL 训练、提升文本与多模态推理后训练的通用方案继续推进。基于其成果可延伸的研究:(1)无标准答案场景——用教师模型生成候选推理链再由验证器筛选,或把模型自己历史上的正确解作为脚手架,摆脱对人工标注推理链的依赖;(2)把提示调度推广到过程奖励或部分得分场景,让 $g$ 的升降依据连续奖励而非二元成败;(3)与课程学习、数据难度估计结合,把有限的引导预算优先分配给高价值难题;(4)理论延伸:在非二元奖励、非平稳成功率 $p_t$ 与随机分段质量下收紧 Theorem 2 的界,并分析 $\lambda_t$ 的收敛作用;(5)迁移到 agentic RL,把工具调用轨迹当作推理链,用环境反馈作验证器;(6)研究提示调度状态 $g_i$ 跨模型规模或训练运行的迁移,减少冷启动开销;(7)多模态上探索图像区域提示(bounding box 提示已被证明优于纯文本 CoT SFT 的 65.56 对 40.37)与 RGPO 的结合。

复现评估

复现条件较好。代码开源:https://github.com/VietHoang1512/rgpo,基于社区广泛使用的 verl 框架实现,兼容 GRPO 与 REINFORCE++。数据公开:语言实验沿用 HINT 的 DAPO-Math-27K 划分(源自 DAPO-Math-170K),多模态沿用 MINT-CoT 协议;评测基准 AIME24、MATH-500、OlympiadBench、Minerva、ARC-C、GPQA-Diamond、MMLU-Pro、MathVista、MMStar 均公开。超参完整:AdamW 学习率 $10^{-6}$,batch 256(语言)/512(多模态),最大提示长度 1024/768,最大回答长度 4096/1024,每题 8 条采样(4 带 4 不带提示),训练 3 epoch,bfloat16 + FSDP2 + 梯度检查点、不开启 offload。未详述 GPU 数量与时长,但 3B/7B 规模用数张 A100 级显卡可复现;受控模拟为合成环境、计算量小。总体复现难度中等偏低,主要工作量在 verl 分布式环境搭建。

相似论文推荐