← 返回 2026-09-15

MInTRL:稀疏离策略干预可增强在策略强化学习 MInTRL: Off-policy Intervention can boost On-policy RL

Mingyu Chen, Yefan Tao, Gerald Friedland, Xuezhou Zhang, Chris Kong 📅 2026-09-11 👍 12 2026-11-02 18:30
LLM后训练 RLVR 优势回归 强化学习 探索与利用 知识蒸馏

在在策略rollout中做稀疏局部纠正,用优势回归学习混合来源轨迹,兼顾探索与可学习性。

前置知识

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

以自动可验证的结果奖励(数学答案对错、代码是否通过测试)训练大语言模型的强化学习范式:模型对每个问题采样多条推理轨迹,验证器给出 0/1 奖励,再用策略梯度类算法更新策略,无需人工标注过程。DeepSeek-R1、Qwen3 等推理模型的后训练核心手段。

本文全部理论与实验都建立在 RLVR 设定之上,理解其「结果奖励 + 自采样轨迹」机制是读懂动机与方法的起点。

在策略(on-policy)与离策略(off-policy)

在策略指训练数据由当前策略自身采样得到,训练与行为分布一致、优化稳定,但学习上限受限于模型自己能采到什么;离策略指使用其他分布(如更强教师模型)生成的数据,能引入模型之外的知识,但分布偏移大,直接套用策略梯度会不稳定。

本文的核心问题正是在两者之间折中:用极少量离策略干预扩展探索边界,同时让轨迹大体保持在策略,理论部分讨论的覆盖—可学习性权衡就源于此。

GRPO 与重要性采样

GRPO 是 RLVR 常用算法:同一问题采样一组轨迹,用组内平均奖励作基线计算优势后做策略梯度更新。当数据来自非当前策略的行为策略时,需逐 token 乘以 $\pi/\mu$ 的重要性比率修正,该比率沿轨迹连乘后极易爆炸或塌缩,导致训练崩溃。

MInTRL 的轨迹混入了干预策略写的 token,若直接用 GRPO 必须做重要性修正;论文改用回归目标正是为了从根上绕开这一步,这是理解方法设计的钥匙。

优势回归(advantage regression)目标

从 KL 正则化 RL 目标 $\max_\pi \mathbb{E}[r] - \beta D_{KL}(\pi\|\pi_t)$ 的闭式最优解出发,可得到逐轨迹最优性条件 $\beta \log \frac{\pi^*_\beta(y|x)}{\pi_t(y|x)} = A^*_\beta(x,y)$,即每条轨迹的对概率比应等于其优势。回归损失 $\mathcal{L}(\theta) = \mathbb{E}\left[(\beta \log \frac{\pi_\theta(y|x)}{\pi_t(y|x)} - A^*_\beta)^2\right]$ 直接拟合该条件,对任何 $\pi_t(y|x)>0$ 的轨迹都有定义,与生成它的行为策略无关。

这是 MInTRL 能直接消化「策略 + 干预混合来源轨迹」的数学基础,是方法一节的核心公式。

Pass@k 与策略熵

Pass@k 指对每题采样 $k$ 条答案、至少一条正确的比例,衡量给定采样预算下模型的探索上限;策略熵衡量输出分布的随机性,熵骤降意味着探索多样性丧失、训练过早收敛。RLVR 中常见「奖励升、熵崩」现象,是衡量训练健康度的重要指标。

论文用 Pass@k(直到 k=32)和训练熵论证 MInTRL 在提升奖励的同时保住了多样性,这是「扩展覆盖」主张的关键实证证据。

正确轨迹的 ε-支撑集

对二值验证奖励,策略 $\pi$ 的 ε-支撑集定义为 $supp_\epsilon(\pi;x)=\{y \in C(x): \pi(y|x)>\epsilon\}$,即概率大于 ε 的正确轨迹集合。有限采样预算下只有「推理支撑集」(阈值 $\epsilon_{inf}$)内的轨迹能被观察到,而训练仍能从概率更低的「训练支撑集」(阈值 $\epsilon_{train} < \epsilon_{inf}$)轨迹中学习。

论文理论部分用这套语言刻画稀疏干预最多能扩展多少覆盖($N$ 次纠正扩展 $(1/\nu)^N$ 倍)、何时越过可学习边界,是读懂第 3 节的前提。

研究动机

RLVR 几乎全部建立在纯在策略学习之上:训练数据完全来自当前策略自己的采样。这带来一个根本瓶颈——如果策略在有限次采样内根本采不到正确答案,该问题就只能产生零优势梯度,训练信号直接浪费掉。近期研究(Chen et al. 2026b; Wu et al. 2025)证实,在策略 RLVR 往往无法发现基座模型在有限采样预算下本就不可达的推理轨迹。数字非常直观:Qwen3-1.7B 基座在 AIME25 上 Pass@1 只有 8.85%,HMMT25 只有 4.48%,大量题目在 rollout 中几乎见不到正确解;标准 GRPO 训练 500 步后数学三基准平均也只从 7.40 提到 17.74。另一条路是引入外部知识:SFT 或蒸馏能完整展示教师模型的成功路径,但整条轨迹都来自教师,与学生分布偏差巨大(SFT+GRPO 在 1.7B 数学上平均只有 25.21,不如部分干预式方法),且学生难以从「远超自身能力」的示范中可靠学习。

本文的目标是本文要回答一个基础问题:LLM 能否在 RL 训练过程中有效吸收外部知识?如果可以,应该引入多少离策略信息?具体目标是设计一种机制,在 rollout 阶段只注入极少量(实验中约 2–4% 的 token)离策略内容,让当前策略暴露于靠自己采样难以发现的正确推理路径,同时保持轨迹整体在策略性质以保住可学习性;配套地,需要一个不依赖重要性采样的训练目标来消化这些混合来源轨迹;最后用理论刻画「干预多少最合适」这一权衡。

与已有工作不同的是,已有工作分两类:一类整条替换轨迹(SFT+GRPO、ReGFT、ReLIFT),分布偏移大;另一类做干预但用策略梯度 + 重要性权重(MENTOR 在 token 级混合学生与专家分布,需要投机采样协调解码)。MInTRL 的独特切入点是「局部性 + 回归」:干预只在错误发生的步骤附近出现,是轨迹中孤立的短片段,其余部分完全由策略自主生成,纠完立刻交还控制权;训练端则用 KL 正则目标的闭式最优解做序列级优势回归,行为策略只决定「看到哪些轨迹」而不进入梯度修正,从根上免除重要性采样。此外,本文首次用支撑集理论把「稀疏纠正能扩展多少覆盖、何时越过可学习边界」变成可定量分析的定理,为倒 U 形干预强度曲线提供了解释框架。

核心方法

直觉上,RL 的问题是「看不见正确答案就学不到」,SFT 的问题是「看到太多自己学不动的答案」。MInTRL 让策略像平时一样自己写推理,一个裁判—干预模型(judge–intervention policy,$\pi_{JI}$)像 code review 一样分段检查:发现第一个致命错误时,砍掉错误后缀、由 $\pi_{JI}$ 补一小段纠正(不超过 64 个 token),然后立刻把笔交还给策略继续写。这样一条轨迹中约 96–98% 的 token 仍由当前策略生成,但关键岔路口被拉回正轨,策略得以从「够得着的正确前缀」继续自主探索。技术路线分两个阶段:第一阶段是半在策略 rollout 采样(Algorithm 1),策略按 chunk 生成、裁判逐段审查、KEEP 或 REVISE(截断 + 短纠正)循环直至完成;第二阶段是回归式 RL 训练,用 KL 正则 RL 目标的闭式最优性条件构造序列级平方回归损失,优势基线只用纯在策略控制 rollout 估计,干预 token 的参考概率用常数锚点 $\kappa$ 放宽。工程上还配套了步骤级错误定位(按 \n\n 或代码围栏切分)、双温度 $\beta_1 \gg \beta_2$ 设计,以及「前 200 步开干预、后 300 步纯在策略」的早停策略。

核心创新有三点,每一处都对应已有方法的明确短板。第一,「最小干预」的半在策略数据生成:与 MENTOR 的 token 级混合分布不同,MInTRL 的离策略信息只以孤立的短 span 出现在错误点,学生解码在两次审查之间完全连续;与 SFT 的整轨迹替换相比,轨迹的状态分布基本留在当前策略附近,这正是「可学习性」得以保住的原因——干预被当作引导探索的路标而非监督信号,策略优化始终由可验证的结果奖励驱动,纠正内容本身错了也不会直接污染梯度。第二,用优势回归替代策略梯度:回归目标 $\mathcal{L}(\theta)=\mathbb{E}[(\beta\log\frac{\pi_\theta(y|x)}{\pi_t(y|x)}-A^*_\beta)^2]$ 对任意 $\pi_t(y|x)>0$ 的轨迹都有定义,与行为策略无关,因此混合来源轨迹可以直接当回归样本用,完全不需要重要性加权。第三,支撑集理论:在最优 token 可达性假设下,$N$ 次最优纠正最多把可见正确轨迹的概率下界扩展 $(1/\nu)^N$ 倍,但只有当 $\epsilon_{inf}\nu^N \geq \epsilon_{train}$ 时扩展的轨迹仍在训练支撑集内——第一次把「该干预多少」表述成覆盖与可学习性之间的定量权衡。

方法步骤详情

完整流程如下。输入:问题 $x$、当前策略 $\pi_t$、裁判—干预策略 $\pi_{JI}$(主实验用 Qwen3-4B-Instruct-2507)。第一步,半在策略采样(Algorithm 1):初始化接受前缀 $p=\emptyset$;$\pi_t$ 以 chunk 为单位生成(数学 512 token、代码 128 token);用双换行 \n\n 或代码围栏把新 chunk 切成步骤;$\pi_{JI}$ 在 $(x, p)$ 上下文中审查,输出 KEEP 或 REVISE 加最早错误步骤索引(Prompt 模板见附录 C);KEEP 则整个 chunk 追加进前缀,REVISE 则保留错误步骤之前的内容,由 $\pi_{JI}$ 以保留前缀为条件生成不超过 64 token 的短纠正 $h$,拼接后控制权交还 $\pi_t$;每条轨迹最多审查 4 次,循环直至响应完成(上限 16384 token)。第二步,训练数据构造:每个 prompt 采 8 条半在策略轨迹(允许干预)加 8 条纯在策略控制轨迹。第三步,优势估计:基线只用 8 条控制轨迹的平均奖励 $\hat{V}(x)=\frac{1}{N_c}\sum_j r(x,y_j^{ctrl})$,优势 $\hat{A}=r-\hat{V}$,控制与半在策略轨迹都作为回归样本。第四步,回归更新:最小化 $(\beta_2 \log\frac{\pi_\theta}{\hat{\pi}_t} - \hat{A})^2$,其中策略 authored token 的锚点取 $\log\pi_t$(MInTRL-Proxy)或常数 $\kappa$(MInTRL-Const,数学 −0.2、代码 −0.1),$\beta_1=\infty$(均值基线)、$\beta_2=10^{-3}$。第五步,调度:AdamW、学习率 $10^{-6}$、共 500 步,前 200 步允许干预,之后关闭 $\pi_{JI}$ 转纯 GRPO 式在策略训练。

技术新颖性

对比各条已有路线可以看清新颖性所在。对比纯在策略方法(GRPO/DAPO):首次把「稀疏局部干预」系统地作为扩展探索的原语,并给出支撑集层面的理论刻画。对比 SFT+RL 类方法(ReGFT、ReLIFT、BREAD):不整条替换轨迹,干预 span 只占约 2–4% token,且优化信号始终来自结果奖励而非模仿教师,从机制上隔离了教师错误的危害(图 9 的失败案例仍被结果奖励正确惩罚)。对比最接近的 MENTOR:MENTOR 在 token 级混合学生与专家分布,需要重要性权重修正和投机采样协调;MInTRL 分段审查、学生解码连续,实验上训练时间反超 2.50–3.45 倍。对比 on-policy distillation(OPD/GKD):OPD 逐 token 匹配教师分布,MInTRL 只在失败处请教师介入、目标仍是结果奖励,因此不会把教师风格强加给学生。对比 A*PO 等优势回归工作:本文是首批把回归目标用于「策略 + 干预混合来源」轨迹的场景,并提出用常数锚点 $\kappa$ 处理低概率干预 token 的实用技巧——消融显示这个看似粗糙的技巧反而普遍优于精确的策略锚点。

Overview and main results of MInTRL.
Figure 1: Overview and main results of MInTRL.
Empirical support expansion.
Figure 2: Empirical support expansion.

实验结果

主实验(Table 1)覆盖 Qwen3-1.7B/4B 两个规模、数学与代码两个领域,每题采 32 样本估计 Pass@1。Qwen3-1.7B 上 MInTRL-Const 数学平均 35.45(AIME25 40.73、AIME26 41.56、HMMT25 24.06),比 GRPO 的 17.74 和 OPD 的 21.84 高出 +13.61pp,且六个子基准全部领先;比最强竞争者 MENTOR 的数学平均 26.01 高 9.44pp(摘要所称的最大领先幅度即来自这里)。代码平均 61.95(LCB 37.85、HumanEval+ 80.81、MBPP+ 67.20),相对 GRPO/OPD 较强者 +14.12pp。Qwen3-4B 上规模增大后收益收窄但仍稳定:数学 55.73(+3.02pp vs GRPO 52.71),AIME25 达 65.31;代码 72.63(+6.80pp vs GRPO 65.83)。一个反直觉发现是常数锚点的 MInTRL-Const 普遍优于用精确策略对数概率锚定的 MInTRL-Proxy,作者归因于弱策略下干预 token 概率极低、被策略锚点过度约束。训练动态(Figure 3、4):MInTRL 的训练奖励与熵双高,Pass@k 直到 k=32 仍优于 GRPO;干预 token 上的经验 forward KL 随训练下降,半在策略与纯在策略 rollout 的奖励差距收敛到零,说明策略逐步内化了纠正知识。行为分析(Figure 5):代码上 wait/actually/maybe 等反思词频率相对 GRPO 上升 55.0–84.2pp,compute/use/we can/we will 等执行词下降 25.5–33.8pp。消融(Table 2)显示自我干预(基座模型 + 特权上下文充当 $\pi_{JI}$)依然有效:Proxy 52.29 vs GRPO 47.26。Figure 6 给出核心消融:干预强度呈倒 U 形,离策略 token 占比约 2–4% 最优,过量干预甚至不如不干预的基线。效率上(Table 5),达到相近性能时 MInTRL-Const 比 MENTOR 快 2.50–3.45 倍(如 1.7B 数学 6.44h vs 22.21h),wall-clock 缩减 60.0–71.0%。

Performance with self judge-intervention.
Table 2: Performance with self judge-intervention.
Performance on math and code at final checkpoint.
Table 3: Performance on math and code at final checkpoint.
Wall-clock efficiency relative to MENTOR.
Table 5: Wall-clock efficiency relative to MENTOR.
Core hyperparameters used in the main experiments.
Table 6: Core hyperparameters used in the main experiments.
Training dynamics. MInTRL improves reward while maintaining longer responses and higher empirical token entropy throughout training.
Figure 3: Training dynamics. MInTRL improves reward while maintaining longer responses and higher empirical token entropy throughout training.
Keyword shifts relative to GRPO. Reconsideration and branching markers (wait, actually, maybe, and alternatively) increase, whereas early execution and commitment markers (compute, use, we can, and we will) decrease.
Figure 5: Keyword shifts relative to GRPO. Reconsideration and branching markers (wait, actually, maybe, and alternatively) increase, whereas early execution and commitment markers (compute, use, we can, and we will) decrease.
Performance peaks at moderate off-policy intensity.
Figure 6: Performance peaks at moderate off-policy intensity.
Qualitative mathematics rollout with two local interventions.
Figure 7: Qualitative mathematics rollout with two local interventions.
Wall-clock efficiency on Qwen3-1.7B mathematics training.
Figure 10: Wall-clock efficiency on Qwen3-1.7B mathematics training.
查看结构化数据
任务指标本文基线提升
数学推理(Qwen3-1.7B,AIME25/AIME26/HMMT25 平均) Pass@1(每题32采样均值) MInTRL-Const 35.45 GRPO 17.74;OPD 21.84;MENTOR 26.01;SFT+GRPO 25.21 较最强竞争者 MENTOR +9.44pp,较 GRPO +17.71pp
代码生成(Qwen3-1.7B,LCB/HumanEval+/MBPP+ 平均) Pass@1(每题32采样均值) MInTRL-Const 61.95 GRPO 47.01;OPD 47.83;MENTOR 50.76;SFT+GRPO 59.88 较 GRPO/OPD 较强者 +14.12pp
数学推理(Qwen3-4B) Pass@1(每题32采样均值) MInTRL-Const 55.73(AIME25 65.31) GRPO 52.71;MENTOR 55.35 +3.02pp(vs GRPO)
代码生成(Qwen3-4B) Pass@1(每题32采样均值) MInTRL-Const 72.63 GRPO 65.83;MENTOR 71.80;SFT+GRPO 70.81 +6.80pp(vs GRPO)
训练效率(对比 MENTOR,同为干预期) wall-clock 时间 1.7B 数学 6.44h(3.45×);4B 代码 9.59h(3.28×) MENTOR 1.7B 数学 22.21h;4B 代码 31.49h 2.50–3.45× 加速,时间缩减 60.0–71.0%
干预强度扫描(消融) 相对无干预基线的 Pass@1 增益 离策略 token 占比约 2–4% 时增益最大 0% 干预(纯在策略对照);40% 过量干预 倒 U 形:过量干预增益转负、低于基线

局限与改进

作者在附录 I 承认三点局限:其一,实验只覆盖 Qwen3 家族 1.7B/4B 与数学、代码两类单轮任务,未验证长程 agentic 场景;其二,judge 审查与干预生成本身带来额外推理开销(尽管 wall-clock 分析显示被训练效率收益抵消);其三,judge 可能拒绝正确的推理或给出错误纠正——Figure 9 的失败案例里学生已自行纠正符号错误得到 $f'(1)=-1$,judge 却误判并注入错误符号的纠正,最终答案错成 1。我的补充观察:第一,主表报告的是「三基准平均最高的 checkpoint」而非最终 checkpoint,而附录 E 显示 4B 上 MInTRL-Proxy 最终掉到 47.88、低于 GRPO 的 52.43,说明 Proxy 变体在较大模型上存在后期退化,方法的稳健性对锚点选择敏感;第二,理论假设「最优下一个 token 概率 $\geq \nu$」且纠正即最优 token,与实现中的步骤级、启发式裁判之间有明显落差,定理更多是概念性解释而非性能保证;第三,常数锚点 $\kappa$ 取 −0.2/−0.1 仅靠 pilot rollout 粗估,未做敏感性消融;第四,每题仅 32 个采样评估,AIME 这类小基准上方差不可忽略;第五,方法隐式要求一个比学生强、且与学生分布不太远的裁判模型,这个「既要强又要近」的匹配条件在现实中并不总容易满足(Table 4 中更强的 DeepSeek-V4-Flash 判官反而没带来一致提升)。

独立分析的弱点

第一,对裁判质量的依赖是最大软肋:判错轻则浪费算力、重则毁掉本来正确的轨迹(Figure 9)。改进方向是给裁判加置信度门槛、多次采样投票,或训练专门的过程奖励模型做错误定位,只在「高置信错误」时才触发干预。第二,超参维度多且相互耦合:chunk 大小、审查次数、纠正长度、锚点 $\kappa$、干预停止时机共约五个,目前靠粗调选定;论文自己给出的「半在策略与在策略奖励差距收敛」信号其实可以直接改造成自适应干预开关——差距大就多干预、趋零就停,替代固定 200 步的硬调度。第三,4B 规模上 MInTRL-Proxy 后期明显退化(最终 checkpoint 数学平均 47.88 vs GRPO 52.43),说明策略锚点在大模型 + 长训练下不稳定,可尝试教师概率与策略概率的插值锚点或动态退火。第四,理论与实现脱节:定理需要最优 token 纠正和均匀可达性假设,真实裁判给的是近似最优的步骤级补丁,把分析推广到「近似最优、步骤级」情形会让理论更有预测力。第五,评估面窄:只在 Qwen3 上验证,且缺少对同一计算预算下与「把 judge 推理算力拿去做更多 GRPO 采样」这一朴素替代方案的公平对比,目前无法断言稀疏干预是这批推理算力的最优用法。

未来方向

作者明确提出的方向包括:扩展到更多学生模型家族与长程 agentic 任务;设计更高效的 rollout 调度与自适应干预预算;提升错误检测与局部纠正的可靠性以减少无益干预。基于本文成果还可延伸:其一,用奖励差距信号做干预的自动启停,把「干预多少」变成在线决策而非超参;其二,让干预策略本身可学习——学习何时干预、在哪里干预(与 InT 等自提议干预工作结合),甚至用 RL 优化裁判的触发策略;其三,支撑集框架理论上可以推出最优干预预算 $N^*$ 的闭式(令 $\epsilon_{inf}\nu^N=\epsilon_{train}$ 处),值得形式化并实验验证;其四,纠正来源不必是更强教师——检索结果、工具执行反馈、验证器反例都可充当局部干预,构成「多源最小干预」;其五,把稀疏干预与异步/陈旧数据 RL 结合,用同一套覆盖—可学习性语言统一刻画两类离策略信息;其六,在多轮对话、工具调用等长轨迹场景检验「局部性」假设是否仍然是收益的主要来源。

复现评估

复现要素相当齐全,但未见明确的开源代码链接。有利条件:训练数据全部公开(AceReason-Nemotron 过滤子集、DeepCoder-Preview);基座与裁判全是开源权重模型(Qwen3-1.7B/4B、Qwen3-4B-Instruct-2507、DeepSeek-V4-Flash);数学与代码两套裁判/干预 prompt 全文给出(附录 C);超参完整列出(附录 H Table 6:batch 32 prompt × 16 轨迹、8 条在策略 + 8 条半在策略、温度 1.0、top-p 1.0、响应上限 16384、AdamW 学习率 1e-6、梯度裁剪 1.0、$\beta_2=10^{-3}$、$\kappa$ 取值、chunk 配置、200 步干预期);训练基于开源框架 veRL。算力需求中等偏重:NVIDIA H100 80GB,1.7B 数学干预期约 6.44 小时 wall-clock、全程 500 步,4B 规模干预期约 10–11 小时,需要多卡。主要工程难点在于实现分段「生成—审查—截断—纠正—续写」的 rollout 循环,涉及双模型服务与学生 KV cache 的中断恢复。难度评估:有 veRL 工程经验的团队约需数周复现主结果;无代码情况下属于「可复现但费工程」,且 Table 1 报告的是最优 checkpoint,严格复现数值还需逐一评估每 50 步保存的全部 checkpoint。

相似论文推荐