作者:李季 LI JI
AI 前沿技术跟进
全局:

推理与强化学习 Reasoning & RL

RLVR、策略优化、信用分配、探索、奖励设计,让模型学会更好地推理与行动。

条目(5 项)

日期条目 / 一句话通俗分类
2026-10-26Multiscale Policy Optimization for Long-Horizon LLM Agents (MPO)
智能体每一「轮」(一次搜索/一次工具调用)才是真正的决策单位:先按「轮」分功劳,再在轮内按「词」微调,再用「整轮」的概率变化限制更新幅度。
推理与强化学习 智能体
2026-10-24Eat Your Precision: Precision Objective Mitigates Reward Hacking in Rubric-based RL(Fork)
用「得分点清单」当奖励,模型就会学着把话越写越长、把所有点都堆上去;Fork 在奖励里加上「每句话是否必要且正确」的精确率,用调和平均(F-score)把两者绑在一起。
推理与强化学习 安全与对齐
2026-10-20ReAPO: Retry-Aware Policy Optimization for Efficient Exploration in LLM Reasoning
做错了就「告诉模型你错了,换个思路再来一次」,而且只奖励成功的那一次重试,不让前面的失败沾光;这样同样的算力学得更快。
推理与强化学习
2026-10-08DiffusionART: Efficient, Stable and Scalable Reinforcement of Diffusion Models
先把同一个提示词下所有样本的奖励汇总成一个「变好的方向」,再让生成过程中每一步都朝这个方向挪一点;再用两次小更新自动「拉住缰绳」,不用调那个很难调的 EMA。
推理与强化学习 多模态与生成
2026-10-06TRACE: Rollout-Guided Quantization-Aware Training for FP4 Reinforcement Learning of MoE Language Models
RL 里用 FP4 做采样(rollout)省时间,但训练端与采样端对同一个数「舍入」得不一样,会放大偏差;TRACE 让训练端的舍入方向去「照抄」采样端,并只传一点点必要信息。
训练与数据 推理与服务 推理与强化学习

条目详解

2026-10-26arXiv 论文第1期 · 2026-10-31

Multiscale Policy Optimization for Long-Horizon LLM Agents (MPO)

面向长程 LLM 智能体的多尺度策略优化
Tao Ren, Hui Yang, Jia Deng, Tingyang Chen, Qiyao Wang 等共 19 位作者(HF 页面机构标注:IQuest)
一句话通俗:智能体每一「轮」(一次搜索/一次工具调用)才是真正的决策单位:先按「轮」分功劳,再在轮内按「词」微调,再用「整轮」的概率变化限制更新幅度。
阅读模式:

报告的主要结果(均为原文自报,非本站复现)

指标(原文自报,百分比)Qwen3-4B-Thinking:最强非 MPO 基线 → MPOQwen3-30B-A3B-Instruct:最强非 MPO 基线 → MPO
BrowseComp-Plus pass@143.75(SAO)→ 48.0846.73(VAPO)→ 50.32
GAIA pass@135.62(SAO)→ 39.4739.62(SAO)→ 43.42
HLE pass@19.26(GSPO)→ 9.929.91(SAO)→ 10.03
Claw-Eval pass^350.78(PPO)→ 51.7357.23(SAO)→ 58.35
Automation Bench pass@112.06(SAO)→ 13.3314.20(SAO)→ 15.00

原文称 MPO 在两个尺度上的 9 项指标全部最高。4B 上 pass@32:BrowseComp-Plus 75.90、GAIA 69.90、HLE 34.00,分别比 GRPO 与 PPO 中较强者高 3.61 / 5.82 / 5.80 分。

通俗解释

想象一个客服机器人要解决一个复杂问题:它先查订单、再查物流、再回复客户,一共说了好几句话(每句话就是一「轮」),最后只有一个总分——问题解决了没有。

训练时最难的是「功劳怎么分」。老办法有两类:一类(GRPO 等)给整段对话里的每个词发同样的分数,太粗;另一类(PPO、VAPO 等)给每个词单独估分,又太碎,而且忽略了「一轮」才是真正和环境交互的单位。

MPO 的做法是「两级分账」:先看每一轮对最终结果的贡献,把总功劳按轮分下去;每一轮内部,再让个别关键的词多拿一点、不重要的少拿一点,但这一轮的平均功劳保持不变。更新模型时,也用「整轮话的概率变了多少」来限幅,而不是逐词限幅。论文用一次搜索智能体的训练数据观察到:评分员(critic)给出的价值估计主要在「轮与轮之间」变化,这就是它这样设计的依据。

专业解释

MPO 把一个完整的 assistant 输出 $a_t$(一轮:搜索 query、工具调用、最终回答)视作 macro action,在不改变策略参数化(逐 token 自回归)的前提下,在两个尺度上分别处理 credit assignment 与 importance correction。

动机来自 critic 的统计结构:在一次 search-agent 训练的 8 个 checkpoint batch 上,共 2,048 条轨迹、16,323 个 turn,轮间均值差异占 critic token 加权轨迹内价值方差的中位数 89.5%;93.8% 的轨迹轮间方差大于轮内方差。因此长程信用应在轮尺度上传播,轮内只做局部再分配。

方法由两部分组成:(1) MAE(Multiscale Advantage Estimation):在 macro 时间轴上做 turn-level GAE,再用「保持轮均值不变」的中心化局部修正在 token 间分配;该分配被推导为一个带均值约束的最小二乘投影的唯一解;(2) turn-level importance sampling:用整轮概率比 $\rho_t=\prod_i\rho_{t,i}$ 做前向加权与 clip,同时通过 stop-gradient 构造保留每个 token 自己的 MAE 优势。

理论部分:沿用 TRPO 的 surrogate + distribution-shift 分析,给出 turn 级单调改进下界;并证明用 token surrogate 替代 turn surrogate 会引入额外的「轮内分布漂移」惩罚项,在相同 KL 预算下 token 方案的惩罚多出 $(T+1)(L_{\max}-1)\Delta$ 量级。critic 采用 $\lambda_{\text{critic}}=1$(沿用 VAPO 的 decoupled GAE),直接回归终端回报 $R(\tau)$。

详细算法

1. 问题设定

交互轨迹 $\tau=(o_0,a_0,o_1,a_1,\ldots,o_T,a_T)$,其中 $o_0$ 为任务 prompt,$a_t$ 为第 $t$ 轮 assistant 输出(工具调用或最终回答),$o_{t+1}$ 为环境反馈。每个输出是 token 序列 $a_t=(a_{t,1},\ldots,a_{t,L_t})$。历史 $h_t=(o_0,a_0,\ldots,o_t)$,token 前缀 $h_{t,i}=(h_t,a_{t,<i})$。

策略自回归分解:$\pi_\theta(a_t\mid h_t)=\prod_{i=1}^{L_t}\pi_\theta(a_{t,i}\mid h_{t,i})$。只有终端奖励 $R(\tau)$,且仅最后一个 token 获得:$r_{t,i}=R(\tau)$ 当 $t=T,i=L_T$,其余为 0。目标 $\mathcal J(\theta)=\mathbb E_{\tau\sim p_\theta}[R(\tau)]$。环境 token 进入历史但不是策略动作。

2. 关键公式

价值边界与 TD 残差(critic $V_{t,i}=V(h_{t,i})$,rollout 时固定;$\gamma=1$):

$$V_{t,L_t+1}:=V_{t+1,1},\quad V_{T+1,1}:=0,\qquad \delta_{t,i}=r_{t,i}+V_{t,i+1}-V_{t,i}$$

轮级 GAE(轮内 TD 残差求和后,内部 critic 预测相互抵消):

$$\delta^{\text{turn}}_t=\sum_{i=1}^{L_t}\delta_{t,i}=\mathbf 1\{t=T\}R(\tau)+V_{t+1,1}-V_{t,1},\qquad A^{\text{turn}}_t=\delta^{\text{turn}}_t+\lambda A^{\text{turn}}_{t+1}$$

轮内局部 trace(只在本轮内截断):

$$A^{\text{intra}}_{t,i}=\sum_{j=i}^{L_t}\lambda_{\text{intra}}^{\,j-i}\,\delta_{t,j}$$

MAE:带均值约束的最小二乘投影 $\min_{\mathbf w}\tfrac12\|\mathbf w-\beta\mathbf A^{\text{intra}}_t\|_2^2$,约束 $\tfrac1{L_t}\sum_i w_i=A^{\text{turn}}_t$,其唯一解:

$$A^{\text{MAE}}_{t,i}=A^{\text{turn}}_t+\beta\big(A^{\text{intra}}_{t,i}-\overline{A}^{\text{intra}}_t\big),\qquad \overline A^{\text{intra}}_t=\tfrac1{L_t}\textstyle\sum_i A^{\text{intra}}_{t,i}$$

整轮重要性比与 stop-gradient 构造:

$$\rho_{t,i}=\frac{\pi_\theta(a_{t,i}\mid h_{t,i})}{\pi(a_{t,i}\mid h_{t,i})},\quad \rho_t=\prod_{i}\rho_{t,i},\quad \tilde\rho_{t,i}=\rho_{t,i}\cdot\mathrm{sg}\Big[\prod_{j\ne i}\rho_{t,j}\Big]$$

前向值 $\tilde\rho_{t,i}=\rho_t$,梯度 $\nabla\tilde\rho_{t,i}=\rho_t\nabla\log\pi_\theta(a_{t,i}\mid h_{t,i})$。

MPO 目标($\widehat A$ 为在 batch 内对所有有效 assistant token 归一化后的 MAE 优势,更新期间固定):

$$\mathcal L_{\text{MPO}}(\theta)=\widehat{\mathbb E}_{t,i}\Big[\min\big(\tilde\rho_{t,i}\widehat A^{\text{MAE}}_{t,i},\ \mathrm{clip}(\tilde\rho_{t,i},1-\epsilon,1+\epsilon)\widehat A^{\text{MAE}}_{t,i}\big)\Big]$$

理论:turn 级改进下界($A_{\max}$ 为优势上界,$\alpha_M$ 为各历史处整轮输出分布的最大 TV 距离):

$$\mathcal J(\pi')-\mathcal J(\pi)\ \ge\ \mathcal S_M-2A_{\max}T(T+1)\alpha_M^2$$

3. 逐步伪代码
输入: 策略 π_θ, critic V_φ, 任务集 D, 超参 λ, λ_intra, β, ε
      (论文消融中最佳: λ=0.80, λ_intra=0.95, β=1.0; critic 的 λ_critic=1)

repeat 每个训练迭代:
    # 1. Rollout
    对每个任务采样轨迹 τ=(o0,a0,...,oT,aT); 记录每个 assistant token 的 log π_old 与终端奖励 R(τ)

    # 2. 取旧策略下的 critic 值 (估计优势期间固定)
    V[t,i] = V_φ(h_{t,i});   V[t,L_t+1] := V[t+1,1];   V[T+1,1] := 0

    # 3. token 级 TD 残差
    δ[t,i] = r[t,i] + V[t,i+1] - V[t,i]          # r 仅在轨迹最后一个 token 为 R(τ)

    # 4. 轮级 GAE (跨轮传播信用)
    A_turn[T+1] = 0
    for t = T downto 0:
        δ_turn[t] = Σ_i δ[t,i]
        A_turn[t] = δ_turn[t] + λ * A_turn[t+1]

    # 5. 轮内再分配 (均值保持)
    for 每一轮 t:
        A_intra[t, L_t+1] = 0
        for i = L_t downto 1:
            A_intra[t,i] = δ[t,i] + λ_intra * A_intra[t,i+1]
        Ā = mean_i A_intra[t,i]
        A_MAE[t,i] = A_turn[t] + β * (A_intra[t,i] - Ā)

    # 6. batch 内对所有有效 assistant token 归一化 -> Â, 更新期间固定

    # 7. 策略更新 (整轮重要性比 + stop-gradient)
    ρ[t,i]  = π_θ(a_{t,i}|h_{t,i}) / π_old(a_{t,i}|h_{t,i})
    ρ̃[t,i] = ρ[t,i] * stopgrad( Π_{j≠i} ρ[t,j] )      # 前向值 = 整轮比 ρ_t
    L = mean_{t,i} min( ρ̃·Â , clip(ρ̃, 1-ε, 1+ε)·Â )
    θ <- θ + lr * ∇_θ L

    # 8. critic 更新: 以 λ_critic=1 直接回归终端回报 R(τ)
4. 训练 / 推理流程

训练:在三类智能体环境中分别训练——Search、Claw-Eval、Automation Bench;步数分别为 300 / 300 / 200。基座为 Qwen3-4B-Thinking 与 Qwen3-30B-A3B-Instruct。对比 Base、GRPO、GSPO、PPO、VAPO、SAO,另设「MPO w/ token IS」(保留 MAE,但把整轮比换回 token 比)以隔离更新目标的作用。

评测协议:每次训练取最佳 checkpoint,离线评测 5 次取均值。Search 以 BrowseComp-Plus、GAIA、HLE 报告 pass@1 / pass@3(训练后模型用每题 32 条独立 rollout 估计);Claw-Eval 报告 pass@3 与 pass^3;Automation Bench 报告 pass@1。

推理:推理时就是普通的智能体推理,MPO 只改变训练,不引入额外推理开销。

与既有方法对比

  • GRPO / GSPO / DAPO(critic-free):每条序列一个共享的组内相对优势,不区分轮与 token 的贡献。MPO 引入 critic,换来跨轮与轮内的细粒度信用。
  • PPO / VAPO / SAO(token 级 critic):把每个 token 当作动作,不显式区分「轮内生成」与「轮间环境交互」。MPO 的 $\lambda$(每轮衰减一次)与 $\lambda_{\text{intra}}$(每 token 衰减一次)分开设置,因为两者的物理时间尺度差很多:$1/(1-\lambda)$ 在 $\lambda=0.8$ 时约 5 轮,而 $\lambda_{\text{intra}}=0.95$ 时约 20 个 token。
  • 消融证据:把整轮比换成 token 比,所有报告分数都下降(4B 上 Search pass@1 下降 2.36 / 3.12 / 1.03 分);即便不用整轮比,MAE 本身在 6 项 Search 指标上都优于 PPO。$\beta=0$(只有轮级 GAE)仍优于 PPO,但比 $\beta=1$ 低 6.28 / 5.27 / 2.04 分。

优势

  • 把「轮」作为一等公民:信用传播、重要性比、clip 都在同一个尺度上对齐,理论上有 turn 级改进下界,并能定量说明 token 代理的额外惩罚来源。
  • 只用一个 critic 就完成两级信用分配(约束最小二乘投影,有闭式解),实现改动集中在优势估计与比值构造两处。
  • 消融设计干净:分别拆开了轮级 GAE、轮内修正、整轮 IS,并扫描 $\lambda,\lambda_{\rm intra},\beta$,看得到每个部件的边际贡献。
  • 增益在 4B 稠密和 30B-A3B MoE 两个规模、三类任务上方向一致;pass@k 随采样预算增大仍保持领先。

劣势 / 局限

  • 需要训练 critic,显存与算力开销高于 GRPO 类 critic-free 方法;论文未报告相对 GRPO 的单位算力收益(待确认)。
  • 评测只覆盖 Search / Claw-Eval / Automation Bench 三类任务与 Qwen3 两个规模;论文没有单独的局限节。
  • HLE 上增益仅 0.12 ~ 0.66 分,且评测取「最佳 checkpoint」后 5 次评测均值,论文未见置信区间或多训练种子方差(待确认)。
  • 整轮重要性比是 token 比的乘积,轮很长时数值方差可能很大——这是编者推断,论文未讨论。
  • 最优超参($\lambda=0.80,\lambda_{\rm intra}=0.95,\beta=1.0$)来自消融扫描,迁移到新任务时需要重新调。

与工业 AI 的关系(作者判断,非论文结论)

设备运维 / 产线诊断智能体本质上是多轮工具调用:查历史工单、读 PLC 点位、调用诊断模型、给出处置建议,且奖励稀疏(最终是否排除故障)。MPO 的「按轮分功劳」可直接借鉴来训练这类长任务智能体;但需要可反复交互的仿真或数字孪生环境,现场实操不能拿来试错。短期更现实的用法是离线仿真内训练,并配合只读、受控上线的边界。

待确认事项

  • 待确认:MPO 是否有开源代码与权重 依据:HF Papers API 返回中无 githubRepo 字段,论文 HTML 摘要未给仓库链接(2026-10-31)
  • 待确认:各基线是否在同等 rollout 预算与相同 critic 规模下比较 依据:已读 6.1 节设置,仅见步数与 rollout 数,未见算力对齐表述(2026-10-31)
核验:arXiv 摘要页标题与提交日期(2026-10-26)核对一致;已通读 HTML 全文的方法(§3–§5)、实验(§6)与结论;表中数字取自论文 Table 1 与 §6 文字。(2026-10-31(北京时间))
2026-10-24arXiv 论文第1期 · 2026-10-31

Eat Your Precision: Precision Objective Mitigates Reward Hacking in Rubric-based RL(Fork)

「吃掉你的精确率」:用精确率目标缓解 rubric 强化学习中的奖励投机(Fork)
Hyun Ryu, Jong-Heon Kim, Yiqing Xie, Seungone Kim, Carolyn Rose, Eunho Yang(Carnegie Mellon University 等)
一句话通俗:用「得分点清单」当奖励,模型就会学着把话越写越长、把所有点都堆上去;Fork 在奖励里加上「每句话是否必要且正确」的精确率,用调和平均(F-score)把两者绑在一起。
阅读模式:

报告的主要结果(均为原文自报,非本站复现)

RubricHub-Medical 测试集与 HealthBench(Qwen2.5-7B-Instruct,原文 Table 1,参考 grader 为 GPT-5.4):

方法RecallPrecision$F_1$对基座 win rate长度(tokens)HealthBench win rate
基座22.646.626.3—2,432—
Standard35.725.526.43.19,1957.1
OnlineRubrics38.726.828.810.68,73622.1
Holistic Preference27.547.531.365.43,58471.6
Fork(β=1)32.553.237.568.03,53374.1
  • Science(β=1.25):对基座 win rate 61.5%(Universal Rubrics 42.5%),ResearchQA 上 67.1%(对比 26.6%)。
  • 更换 judge prompt(Rezaei 等)后 Fork 仍大多领先(Table 4)。
  • 注意:摘要中的「10.6% → 68.0%」是 OnlineRubrics 与 Fork 对基座策略的 win rate,不是 Fork 自身的前后变化。

通俗解释

很多没有标准答案的任务(医疗问答、科学问答)用「评分细则」(rubric)来训练:细则里列了回答里应该出现的要点,覆盖越多分越高。

问题是这相当于只奖励「召回」——没说到的要点扣分,但废话、重复、甚至错误内容却没人扣分。模型很快学到:多写、什么都写,覆盖率就高。结果回答更长、更啰嗦、还夹带错误,人来比反而不如原始模型。

Fork 的做法:把回答拆成一句一句的「单元」,检查每一句是不是(1)确实对应某个评分要点且不是重复的(必要)、(2)事实上正确。「又必要又正确」的句子占比就是精确率。最后的奖励是「覆盖率」和「精确率」的调和平均(F 值),想靠堆砌来刷分就不再划算。

专业解释

动机:现有 rubric 奖励 $\mathrm{Rec}(x,y)=\sum_j w_js_j/\sum_j w_j$ 本质是 recall objective——约 90.2% 的 RubricHub 规则是 presence-based,几乎不惩罚冗余、不必要或错误的内容。RL 训练使 recall 快速饱和,而 precision 与对基座模型的 win rate 持续下降,响应变长(reward hacking)。

Fork reward:(1) Response unitization:LLM 把响应切成近似单句的 *units*,只做最小改写使其自包含,丢弃问候、标题、过渡句;(2) Credit-based necessity:对每个 rubric 分配总额 1 的「单元额度」,均分给匹配该 rubric 的单元,单元的累计额度 $c_i=\sum_j a_{ji}$ 越高越必要;冗余单元因 $1/K_j$ 被稀释;LLM grader 给出阈值 $\tau$ 得到二值 $N_i$;(3) Factual correctness $C_i$:严格的单元级二值判断;(4) Precision:$\mathrm{Prec}=\frac1n\sum_iN_iC_i$;最终奖励为 Prec 与 Rec 的 $F_\beta$(加权调和平均),用 GRPO 优化。

训练 grader 的选择:5 个候选中 Qwen3.6-35B-A3B 与 3 个前沿模型的判断相关性最高($r=0.714$);评测用 GPT-5.4 作参考评判。

详细算法

1. 问题设定

非可验证任务。数据 $\mathcal D=\{(x,\mathcal R(x))\}$,$\mathcal R(x)=\{(r_j,w_j)\}_{j=1}^M$ 为由 LLM 合成的评分细则及重要性权重;策略 $\pi_\theta$ 采样开放式响应 $y$。LLM grader 对每个 rubric 给二值覆盖标签 $s_j=\mathbb I[y\text{ satisfies }r_j]$。基线奖励是加权覆盖率(recall)。目标:设计既不依赖更多人工细则、又能抑制「堆砌」的奖励。

2. 关键公式

Rubric recall(式 1):

$$\mathrm{Rec}(x,y)=\frac{\sum_{j=1}^{M}w_j\,s_j(x,y)}{\sum_{j=1}^{M}w_j}$$

必要性额度(式 2–3;$m_{ji}=\mathbb I[u_i\text{ 对 }r_j\text{ 有贡献}]$,$K_j=\sum_i m_{ji}$):

$$a_{ji}=\begin{cases}m_{ji}/K_j,&K_j>0\\0,&K_j=0\end{cases},\qquad c_i=\sum_{j=1}^Ma_{ji},\qquad N_i=\mathbb I[c_i\ge\tau]$$

事实正确性(式 4):$C_i=\mathbb I[u_i\text{ 事实正确}]$。精确率(式 5):

$$\mathrm{Prec}(x,y)=\frac1n\sum_{i=1}^{n}N_iC_i$$

Fork 奖励(式 6,$\beta>1$ 偏重 recall,$\beta<1$ 偏重 precision):

$$r_{\rm FORK}(x,y)=\frac{(1+\beta^2)\,\mathrm{Prec}\cdot\mathrm{Rec}}{\beta^2\,\mathrm{Prec}+\mathrm{Rec}}\quad(\mathrm{Prec}+\mathrm{Rec}>0;\ \text{否则 }0)$$

目标(式 7):

$$\max_\theta\ \mathbb E_{x\sim\mathcal D,\ y\sim\pi_\theta(\cdot|x)}\big[r_{\rm FORK}(x,y)\big]$$

3. 逐步伪代码
# 每个训练步 (GRPO, 16 rollouts/prompt)
for 每个 prompt x (含 rubric 集 R(x) = {(r_j, w_j)}):
    采样 16 条响应 y_1..y_16 ~ π_θ(·|x)
    for 每条响应 y:
        units = LLM_unitize(y)                          # 约一句一个 unit; 丢弃问候/标题/过渡
        s_j   = LLM_cover(y, r_j)       for j in 1..M   # 覆盖标签
        Rec   = Σ w_j s_j / Σ w_j
        m_ji  = LLM_match(r_j, u_i)                     # 每个 (rubric, unit) 是否贡献
        K_j   = Σ_i m_ji;   a_ji = m_ji/K_j (K_j>0 else 0)
        c_i   = Σ_j a_ji
        τ     = LLM_threshold(c_1..c_n)                 # grader 给出必要性阈值
        N_i   = 1[c_i >= τ];   C_i = LLM_factcheck(u_i, x)
        Prec  = (1/n) Σ N_i * C_i
        r     = F_β(Prec, Rec)                          # 奖励
    A_k = (r_k - mean(r)) / std(r)                      # GRPO 组内优势
    用 GRPO 目标更新 θ

# 超参: batch 128, lr 4.2e-6 (常数 + 5% warmup), 3 epochs, 8×H100, 上下文 2,000 tokens, T=1.0
# 训练 grader: Qwen3.6-35B-A3B;  评测参考 grader: GPT-5.4
4. 训练 / 推理流程

基座:Qwen2.5-7B-Instruct。数据:RubricHub 的 Medical 子集(训练 12,519 / 测试 1,391)与 Science 子集(19,806 / 2,201,去除 ResearchQA 来源以便作外部评测)。外部评测:HealthBench(医疗)、ResearchQA(科学)。评测方式:(1) rubric-based:macro recall / precision / $F_1$;(2) rubric-free:对基座的 win rate 与平均长度。基线:Standard、OnlineRubrics、Universal Rubrics、Reweighted Absence-Based Rubrics(覆盖类);Holistic Preference(偏好类);以及作者自己的精确率变体 Generic / Offline / Online Criteria。

与既有方法对比

  • Standard 覆盖奖励(Gunjal 等 2026):只有 recall,结果响应长度 9,195 token(基座 2,432),精确率 25.5%(基座 46.6%),对基座 win rate 仅 3.1%。
  • OnlineRubrics / Universal Rubrics / 重加权 absence 规则:仍属覆盖类奖励,补充规则或重加权,但精确率只有 24.2% ~ 27.6%,win rate 4.3% ~ 10.6%。
  • Holistic Preference(偏好奖励):精确率高(47.5%)、win rate 高(65.4%),但召回显著下降(27.5% vs Fork 32.5%)。
  • Fork 的精确率变体:Generic / Offline / Online Criteria(让 grader 自行给出精确率标准)都不如基于 rubric 的「必要性额度」;作者以此说明 rubric-grounded necessity 的必要性。

优势

  • 把「覆盖率」补上「精确率」,用一个 $F_\beta$ 即可把召回—精确率的帕累托权衡显式化,$\beta$ 可调。
  • 必要性额度「每条 rubric 总额 1、均分给匹配单元」这个设计既避免重复被奖励,也不要求新增细则,几乎是零标注成本。
  • 训练前后行为可解释:召回基本不降,长度降到 Standard 的约一半,且 absence-based 规则满足率提升 5.7 pp(β=1.25)。
  • 代码开源;评测同时做了 rubric 评测、rubric-free win rate、换评判 prompt 稳健性。

劣势 / 局限

  • 奖励依赖一个训练用 LLM grader 同时完成切分、匹配、阈值与事实核验,grader 的偏差和成本直接影响训练;论文用 Qwen3.6-35B-A3B(与 3 个前沿模型相关 0.714)。
  • 「事实正确性」由 LLM 判定,本身可能出错;对没有外部知识核验的专业领域(医疗)是否会引入新的错误,需要人工评估(本站未核实,待确认)。
  • 实验只有 Qwen2.5-7B-Instruct 一个基座与医疗 / 科学两个领域;win rate 依赖 GPT-5.4 作评判,与人评的一致性文中仅给出部分验证。
  • $\beta$ 需调:Medical 用 1,Science 用 1.25 等;跨任务的鲁棒选择未给出规则。
  • 切分粒度是「近似单句」,换成按换行切分的结果在 §4.3 另述;对列表、表格类输出的适用性本站未核实。

与工业 AI 的关系(作者判断,非论文结论)

工业场景里大量「非可验证」的任务(故障分析报告、检修建议、工艺复盘)也会用评分细则做奖励,同样会遇到「写长」「堆砌」。Fork 的「必要性 + 正确性」精确率思路可以直接借鉴来做报告类 agent 的 RL / 偏好评估:一方面给要点覆盖打分,另一方面严格扣掉冗余与不实内容。关键前提是要有可靠的事实核验来源(设备手册、历史工单),否则「正确性」评判会变成模型自说自话。

待确认事项

  • 待确认:Fork 训练出的响应在人评(专家)下是否同样更优 依据:论文主要用 GPT-5.4 作参考评判;人评部分未在本站核实(2026-10-31)
  • 待确认:Limitations 小节内容 依据:文本抽取未定位到该小节(2026-10-31)
核验:arXiv 摘要页标题与提交日期(2026-10-24)核对一致;GitHub 仓库 HTTP 200;已读 §1、§3 方法与 §4.1–§4.3,表中数字取自 Table 1、2、3、4。(2026-10-31(北京时间))
2026-10-20arXiv 论文第1期 · 2026-10-31

ReAPO: Retry-Aware Policy Optimization for Efficient Exploration in LLM Reasoning

ReAPO:面向 LLM 推理高效探索的「重试感知」策略优化
Nghi Dao(Columbia), Calder Morton-Ferguson(Yale), Noah Golowich(UT Austin), Adam Block(Columbia)
一句话通俗:做错了就「告诉模型你错了,换个思路再来一次」,而且只奖励成功的那一次重试,不让前面的失败沾光;这样同样的算力学得更快。
阅读模式:

报告的主要结果(均为原文自报,非本站复现)

  • Qwen3-0.6B-Base(GSM8K-Platinum):达到「GRPO 双倍组大小」的峰值精度,ReAPO 少生成 39% 的 token;达到「GRPO 双倍步数」的峰值,少 31%(原文自报)。
  • 在 Qwen3 与 OLMo-2 的 0.6B ~ 4B 与 8 个基准上,首次尝试性能优于 GRPO 及近期基线;pass@32 保持或提升(因 pass@32 在该基准上接近饱和,增幅较小)。
  • 答案多样性:第二次尝试复述失败答案的比例迅速接近 0。

通俗解释

像考试:GRPO 相当于让学生对同一道题独立做很多遍,其中很多遍会重复同样的错误思路,浪费时间。

ReAPO 改成:第一遍做错了,就告诉他「你的答案错了,反思一下,换个思路再来」,然后让他在看得到上一次错误的情况下重做。对「重做」这件事怎么打分也有讲究:论文把它看成「到做对为止,平均要试几次」的最短路径问题,目标是让这个次数越少越好;做对的那次重试拿高分,前面的错误尝试仍然拿低分,不会因为后面做对了就一起被奖励。

结果是:在小模型上,用更少的生成 token 就能达到 GRPO 需要两倍样本数或两倍步数才能达到的水平。

专业解释

RLVR 中独立随机 rollout 容易反复采样相似的失败策略。ReAPO 提出 verifier-conditioned exploration:初始 $G$ 条 rollout 与 GRPO 相同;若验证器判错,把错误回答连同 retry prompt("Your answer is wrong. Reflect on your previous answer and try something else.")追加到上下文,让模型在条件于失败历史的情况下重新作答,直至答对、达到最大尝试数 $K$ 或 token 预算耗尽。训练组大小因此按题目难度自适应扩张。

关键的理论贡献是 attempt-level credit assignment:把「首次成功的尝试序号 $\tau$」的期望失败次数 $\mathbb E[\tau-1]$ 作为最小化目标,这是经典的 Stochastic Shortest Paths (SSP) 问题(动作=一次尝试,状态=题目+此前尝试,常数代价)。在两个简化假设下(失败尝试只通过「失败」影响后续成功概率;期望剩余尝试数 $c_m$ 与 $m$ 无关),策略梯度恰好等价于「每个尝试按自己的(未归一化)奖励加权其自身 log-prob 梯度」。实践中再加 GRPO 式中心化 / 标准差归一化,并对尝试数设上限 $K$。

优势仅分配给对应尝试自身的 token,而不是摊到整条多尝试轨迹——这与 SCoRe、Chung 等把一个优势散布到整条轨迹的做法不同。

详细算法

1. 问题设定

语言模型为策略 $\pi_\theta(y\mid x)$,自回归生成。验证器 $r(x,y)\in\{0,1\}$。目标是 $\mathbb E_{x\sim\mathcal D}\mathbb E_{y\sim\pi_\theta}[r(x,y)]$。GRPO 对每个 prompt 采样 $G$ 条回答 $\{y_i\}$,用组内奖励的均值/标准差产生响应级优势。

ReAPO 的多尝试设定:第 $g$ 条 rollout 的第 $k$ 次尝试为 $y_{g,k}$,奖励 $r_{g,k}$;$\tau_g=\inf\{k\ge1: r_{g,k}=1\}$ 为首次成功的尝试序号;上下文 $h_{g,k-1}=(x,y_{g,1},\ldots,y_{g,k-1})$。训练时每题最多 $K$ 次尝试(主实验 $K=2$)。

2. 关键公式

GRPO 优势与目标(背景):

$$\hat A_i=\frac{r_i-\mu_r}{\sigma_r+\varepsilon},\quad \mu_r=\tfrac1G\sum_i r_i,\ \ \sigma_r=\sqrt{\tfrac1G\sum_i(r_i-\mu_r)^2}$$

SSP 视角:令 $q_m(\theta)=\mathbb P(r_m=1\mid r_j=0,\,j<m)$,$\alpha_m=\mathbb P(\tau\ge m)$,$c_m=\sum_{k\ge m}\prod_{j=m+1}^{k}(1-q_j)$(失败 $m$ 次后期望还需的额外尝试数)。则

$$-\nabla_\theta\,\mathbb E_{\pi_\theta}[\tau-1]=\sum_{m=1}^{\infty}\alpha_m(\pi_\theta)\,c_m(\pi_\theta)\,\nabla q_m(\theta)$$

命题 2:若 $c_m\equiv C(\pi_\theta)$ 与 $m$ 无关,则

$$-\nabla_\theta\,\mathbb E[\tau-1]=C(\pi_\theta)\cdot\mathbb E\Big[\tfrac1G\sum_{g=1}^{G}\sum_{k=1}^{\tau_g}r_{g,k}\,\nabla_\theta\log\pi_\theta(y_{g,k}\mid h_{g,k-1})\Big]$$

ReAPO 更新方向($\hat A_{g,k}$ 的均值与标准差在池化的全部尝试奖励 $\{r_{g,k}\}$ 上计算;$T=\sum_g\sum_{k\le\tau_g}|y_{g,k}|$):

$$\nabla_\theta\mathcal J_{\text{ReAPO}}=\mathbb E\Big[\frac1T\sum_{g=1}^{G}\sum_{k=1}^{\tau_g}\hat A_{g,k}\,\nabla_\theta\log\pi_\theta(y_{g,k}\mid x,y_{g,<k})\Big]$$

若所有尝试奖励相同,则归一化优势为 0,该题不产生梯度(与 GRPO 一致)。实验中 on-policy($\pi_{\theta_{\rm old}}=\pi_\theta$)且去掉 KL 项。

3. 逐步伪代码
# 据 §3 文字描述整理(并非逐行抄录论文 Algorithm 1)
输入: 策略 π_θ, 验证器 r, 数据集 D, 组大小 G, 最大尝试数 K, retry 提示语
for 每个训练 step:
    采样一批题目 {x}
    for 每个题目 x:
        for g = 1..G:                                  # 并行 G 条
            y[g,1] ~ π_θ(·|x);   r[g,1] = r(x, y[g,1]);   k = 1
            while r[g,k] == 0 and k < K and 未超 token 预算:
                ctx = x + y[g,1] + ... + y[g,k] + "Your answer is wrong. Reflect on your previous answer and try something else."
                y[g,k+1] ~ π_θ(·|ctx);   r[g,k+1] = r(x, y[g,k+1]);   k += 1
            τ_g = k
        R_pool = 池化所有 r[g,k]
        μ, σ = mean(R_pool), std(R_pool)
        Â[g,k] = (r[g,k] - μ) / (σ + ε)               # 所有尝试一起归一化
    # 优势只分配给对应尝试自己的 token
    loss = - (1/T) * Σ_g Σ_{k<=τ_g} Â[g,k] * Σ_{token∈y[g,k]} log π_θ(token | 其前缀与上下文)
    θ <- θ - lr * ∇loss
4. 训练 / 推理流程

训练:模型为 Qwen3-0.6B-Base(GSM8K-Platinum,8 个种子)、Qwen3-1.7B-Base 与 Qwen3-4B-Base,另用 OLMo-2-0425-1B-SFT 做跨模型族验证;覆盖 8 个数学与科学推理基准,主实验 $K=2$。

推理:评测只看「首次尝试」,推理时不需要验证器与重试——重试仅在训练中用来改进探索与信用分配。

对照:GRPO(同步数)、GRPO 双倍步数、GRPO 双倍组大小($G=32$,对应 $G=16,K=2$ 的最大尝试数)、SCoRe Stage-II-only、MaxRL,以及「无条件重试」「整条轨迹单一优势」两个消融。

与既有方法对比

  • GRPO:独立 rollout,容易重复相同的错误策略;ReAPO 在失败后串行重试并条件于失败历史,答案多样性更高(第二次答案与失败的第一次答案完全相同的比例很快降到接近 0)。
  • SCoRe(Kumar 等 2024):多轮 RL 联合优化两次尝试,不使用验证器条件,依赖较特设的奖励塑形;论文的 SCoRe Stage-II-only 对照在 token / 算力效率上明显不如 ReAPO,mean@32 也未追上,但 pass@32 略有提升。
  • 把单一优势散布到整条多尝试轨迹(Chung 等 2025 等):ReAPO 的消融显示,把优势只给当次尝试的 token 更好。
  • MaxRL:专为 pass@k 设计;ReAPO 的 在 pass@32 上与之持平(论文:matches MaxRL)。

优势

  • 改动极小:只是 GRPO 加「失败后重试」与「按尝试分配优势」,易于接入现有 RLVR 流水线。
  • 把信用分配落在经典 SSP 理论上,推导出的更新形式简单(按自身奖励加权),而不是堆叠启发式奖励塑形。
  • 对失败题目自动多采样、对容易题少采样,近似自适应的难度分配。
  • 推理时不需要验证器与重试,部署成本与 GRPO 训练的模型相同。

劣势 / 局限

  • 理论依赖两个「过于保守」的简化假设(作者原话),实践中失败的尝试显然会改变后续成功概率;$c_m$ 不随 $m$ 变化也只是近似。
  • 硬上限 $K$ 会截断无穷和,使策略梯度有偏;中心化引入阶为 $\nabla\mathbb E[1/\tau]$ 的小偏差(作者在附录讨论)。
  • 串行重试的计算成本随尝试数二次增长、并行度降低,作者以步数 / token / 墙钟三种方式对比,但组合场景下的优势仍需更大模型验证。
  • 实验只在数学 / 科学推理、最大 4B 模型;代码与智能体场景因资源限制未测(作者自述)。
  • 训练需要可靠的二元验证器(RLVR 的通病);不适用于难以验证的开放任务。

与工业 AI 的关系(作者判断,非论文结论)

在可验证的工业子任务上(如从工艺机理方程算出设定值是否落在容差内、PLC 逻辑是否通过仿真校验)可以借鉴「失败后带反馈重试」的训练方式;它提示我们:「把验证器反馈放进上下文」本身就是一种低成本的探索增强。但小模型数学推理上的结论,不能直接外推到工艺参数优化,需要先有可自动判对错的验证环境。

待确认事项

  • 待确认:ReAPO 在 4B 以上模型、代码与智能体任务上是否同样有效 依据:论文自述未测(资源限制)(2026-10-31)
  • 待确认:Algorithm 1 逐行伪代码 依据:HTML 文本抽取中 Algorithm 1 仅剩标题,上方伪代码是依据 §3 文字整理的(2026-10-31)
核验:arXiv 摘要页标题与提交日期(2026-10-20)核对一致;GitHub 仓库链接 HTTP 200;已读 §2 预备知识、§3 方法、§4 实验与局限节。(2026-10-31(北京时间))
2026-10-08arXiv 论文第1期 · 2026-10-31

DiffusionART: Efficient, Stable and Scalable Reinforcement of Diffusion Models

DiffusionART:高效、稳定、可扩展的扩散模型强化学习
Zhenglin Wan, Jiayu Chen, Zhengfeiyang Zhang, Pengfei Zhou, Wangbo Zhao, Fangyuan Kong, Pengfei Wan, Kun Gai, Yang You(KlingAI Research 等)
一句话通俗:先把同一个提示词下所有样本的奖励汇总成一个「变好的方向」,再让生成过程中每一步都朝这个方向挪一点;再用两次小更新自动「拉住缰绳」,不用调那个很难调的 EMA。
阅读模式:

报告的主要结果(均为原文自报,非本站复现)

(原文自报。) - HunyuanVideo-13B(VQ:MQ 等权):达到 DiffusionNFT 与 Flow-GRPO-Fast 后期平均奖励水平,所需 GPU 小时少 12.02× 与 11.72×,同时平均训练奖励高 28.8% 与 53.0%(摘要)。 - T2V(TA:VQ:MQ=7:2:1):达到 VQ 5.5 需 24 步 / 63.4 GPU 小时,DiffusionNFT(EMA) 需 273 步 / 1,401.4 GPU 小时(11.4× 少步数,22.1× 少 GPU 小时);评测 VQ / MQ 为 8.970 / 4.224,较最强基线高 31.4% / 57.4%。 - Pairwise、$K=2$ 与 $K=8$:400 步最终 VQ 12.235 vs 12.092,GPU 小时 260.8 vs 883.3(3.4× 更少)。

通俗解释

用奖励(美观度、动作流畅度等)来「调教」图像或视频生成模型时,有两派方法:一派(Flow-GRPO)每一步去赌「这一步是不是导致高分」,信号很稀,学得慢;另一派(DiffusionNFT)很快,但对一个叫 EMA 的稳定装置极其挑剔,参数换个任务就容易训崩。

这篇论文先搞清楚了为什么 NFT 不稳:它的训练目标里藏着一股「前向与反向不一致」的力,正负样本一起训能抵消一部分,却抵消不干净。

DiffusionART 的办法:对每个提示词,把一组样本和它们的奖励放在一起算一个「奖励提升方向」(协方差),然后让生成过程中经过的每一个中间状态都朝这个方向挪一步。这样那股不一致的力从构造上就没有了。再把每批数据拆成两次更新,第二次更新自带「离出发点越远拉得越狠」的回拉力,替代以前要手工调的 EMA。

专业解释

论文聚焦 rectified-flow 的在线 diffusion RL,分析 forward-process RL(DiffusionNFT) 的不稳定性:

(1) 隐藏的 forward–reverse consistency force:NFT 的目标由前向加噪端点构造,隐含假设模型的采样速度场 $v_{\rm rvs}^{\rm old}$ 等于其自身端点诱导的理想前向速度 $v_{\rm fwd}^{\rm old}$;二者之差 $e$ 使 NFT 在 on-policy 点的输出空间力为 $\mathcal F_{\rm NFT}=2g+(2q-1)e$,其中 $g$ 是奖励倾斜方向,$(2q-1)e$ 是不一致力。正负分支各自带有未抵消的 $e$;联合训练只能在 $q$ 平均为 $1/2$ 时部分抵消。

(2) EMA 靠「恢复力」而非 off-policy 起效:NFT 损失可拆为线性项(引导 + 不一致力)与二次项 $\beta^2\|d\|^2$(把 $v_\theta$ 拉回 $v_{\rm old}$)。纯 on-policy 时 $d=0$,没有恢复力。实验:保留 EMA 但去掉旧策略锚点仍会崩;改成每个 rollout 做两次更新(无 EMA)就不崩但慢 15.2 倍。

DiffusionART 的两个设计:Aggregation-first guidance——对每个 prompt 用 $K$ 个端点的协方差 $\overline{\rm Cov}(R,x_0\mid c)$ 做 RMS 归一化得到共享方向 $G(c)$,对 *reverse rollout 实际经过的每个状态* 回归目标 $v_{\rm tgt}=v^{(0)}_\theta+G(c)$(目标中不含前向速度,一致力被构造性消除);两次更新的漂移自适应恢复——把 rollout batch 拆成两个不相交的 mini-batch 各做一次 MSE 更新,第二次的输出梯度为 $\tfrac{2w}{D}(G-d)$,恢复项 $-d$ 随漂移增长。外加 trust-region 掩码(用 ODE→SDE 的参考高斯转移定义 KL 代理,屏蔽「超过阈值且继续向外」的更新)。还提供 Pairwise 变体(成对偏好代替标量奖励)。

详细算法

1. 问题设定

rectified-flow 参数化:$t=1$ 为噪声,$t=0$ 为干净样本;前向状态 $x_t=\alpha_tx_0+\sigma_t\epsilon$,样本速度目标 $u_t=x_0-\epsilon$。每次迭代对每个 prompt $c$ 从冻结的 rollout 策略 $p_{\rm old}$ 采 $K$ 个端点 $x_0^{(i)}$,得到黑盒奖励 $R_i=R(x_0^{(i)},c)$,不要求奖励可微。目标:用这些端点级反馈高效而稳定地更新速度网络 $v_\theta(x_t,t,c)$。

2. 关键公式

NFT 的目标(背景),$r_i=\tfrac12[1+\mathrm{clip}((R_i-\bar R_c)/Z_c,-1,1)]$,$v_{\theta,\pm}=v_{\rm old}\pm\beta(v_\theta-v_{\rm old})$:

$$\mathcal L_{\rm NFT}=\mathbb E\big[r\|v_{\theta,+}-u_t\|^2+(1-r)\|v_{\theta,-}-u_t\|^2\big]=\beta^2\mathbb E\|v_\theta-v_{\rm tgt}\|^2+\text{const}$$

命题 1(NFT 力分解),$e=v_{\rm fwd}^{\rm old}-v_{\rm rvs}^{\rm old}$:

$$\mathcal F_{\rm NFT}(s_t)=2g+(2q-1)e,\qquad \mathcal F_+=g+qe,\quad \mathcal F_-=g-(1-q)e$$

指数倾斜与协方差引理:$p_\tau(x_0\mid c)\propto p_{\rm old}(x_0\mid c)\,e^{\tau R(x_0,c)}$,则

$$\left.\frac{\partial}{\partial\tau}\mathbb E_{p_\tau}[x_0\mid c]\right|_{\tau=0}=\operatorname{Cov}_{p_{\rm old}}(R,x_0\mid c)$$

聚合优先的引导方向(式 5):

$$\overline{\rm Cov}=\frac1K\sum_{i=1}^K(R_i-\bar R_c)(x_0^{(i)}-\bar x_{0,c}),\qquad G(c)=\frac{\overline{\rm Cov}}{\mathrm{RMS}(\overline{\rm Cov})}$$

冻结目标与逐状态损失($w$ 含时间步权重与掩码,$D$ 为输出维数):

$$v^{\rm ART}_{\rm tgt}(s_t)=v^{(0)}_\theta(s_t)+G(c),\qquad \mathcal L_\theta(s_t)=w(s_t)\cdot\tfrac1D\|v_\theta(s_t)-v^{\rm ART}_{\rm tgt}(s_t)\|_2^2$$

两次更新的输出梯度($d=v_\theta-v^{(0)}_\theta$ 为漂移):

$$-\nabla_{v_\theta}\mathcal L\big|_{v^{(0)}_\theta}=\tfrac{2w}{D}G(c),\qquad -\nabla_{v_\theta}\mathcal L\big|_{v^{(1)}_\theta}=\tfrac{2w}{D}\big(G(c)-d(s_t)\big)$$

Trust-region 掩码(KL 代理 $m$;$\delta=10^{-5}$):

$$m(s_t)=\frac{\|\mu(s_t,v_\theta)-\mu(s_t,v^{(0)}_\theta)\|_2^2}{2D\,\sigma_{\rm ref}^2(t)},\qquad w(s_t)=t\Big[1-\mathbf 1\{m\ge\delta\}\,\mathbf 1\{\langle d,\,G-d\rangle>0\}\Big]$$

3. 逐步伪代码
# 据 §4 文字整理 (论文 Algorithm 1 的正文在文本抽取中缺失, 故非逐行抄录)
输入: 速度网络 v_θ, 黑盒奖励 R, prompt 集, 每组样本数 K
for 每个训练迭代:
    # Rollout (ODE 采样, 不需要 SDE 似然)
    for 每个 prompt c:
        用当前网络 v_θ(0) 采 K 个端点 x0^(i); 缓存沿途每个状态 s_t 的预测 v_θ(0)(s_t)
        R_i = R(x0^(i), c)                      # 标量, 或成对偏好标签 (Pairwise 变体)
        # 聚合优先: 每个 prompt 一个共享方向
        Cov = (1/K) Σ_i (R_i - R̄)(x0^(i) - x̄0)
        G(c) = Cov / RMS(Cov)
        对该 prompt rollout 经过的每个状态 s_t:  v_tgt = v_θ(0)(s_t) + G(c)
    # 把本批数据拆成两个不相交的 mini-batch B1, B2
    for B in (B1, B2):                          # 两次更新
        对 B 中每个状态:
            d = v_θ(s_t) - v_θ(0)(s_t)
            m = KL 代理 (式 7);  w = t * (1 - 1[m>=δ] * 1[<d, G-d> > 0])
            loss += w * MSE(v_θ(s_t), v_tgt)
        θ <- AdamW 更新
    # 第 2 次更新自带恢复力 -d; 无需 EMA 调度
4. 训练 / 推理流程

T2I:SD3.5-M(LoRA r=32, α=64, 512×512, lr $3\times10^{-4}$, 10 步 rollout),奖励 GenEval / OCR / PickScore(及联合 CLIPScore、HPSv2.1);FLUX.1-dev (12B):6 步 rollout、guidance 嵌入 3.5。T2V:全参数微调 CFG 蒸馏的 HunyuanVideo-13B,lr $10^{-5}$,每批 32 个 prompt × 8 个视频,53 帧、480×480、8 FPS、16 步采样;VideoReward(VQ 视觉质量 / MQ 运动质量 / TA 文本对齐)按 7:2:1 加权。另有 I2V 与 Pairwise 实验。

基线:Flow-GRPO、Flow-GRPO-Fast (CPS)、DanceGRPO、Flow-DPPO、DiffusionNFT(有 / 无 EMA)。

与既有方法对比

  • Flow-GRPO 类(reverse-process):更新每个被选转移沿其自身的采样噪声方向,由轨迹级优势加权,而该转移只部分决定奖励 → 信用分配差;ODE→SDE 才能得到可处理的似然。ART 沿共享的「奖励提升方向」更新,且 rollout 保持 ODE。
  • DiffusionNFT(forward-process):高效,但依赖联合正负训练 + 逐任务调 EMA;论文示例:把 SD3.5-M GenEval 调好的 EMA/β 迁到 FLUX 就会出现明显震荡。
  • ReFL / Diffusion-DPO:前者需要可微奖励,后者是离线偏好;ART 是在线、黑盒奖励。
  • 结果:T2I 上到达 Flow-GRPO 第 500 步奖励所需的步数分别少 10.0×(GenEval)、5.2×(OCR)、9.6×(PickScore)。

优势

  • 给出了 NFT 不稳定的机理解释(隐藏的一致力 + EMA 是恢复力),并用针对性消融(去掉锚点、两次更新)支持,而不只是工程技巧。
  • 无需可微奖励,不需要 ODE→SDE 似然;rollout 保持 ODE,兼容蒸馏模型。
  • 不需要逐任务调 EMA;同一套超参在图像(SD3.5-M、FLUX)与视频(Hunyuan 13B)上使用。
  • 对每组样本数不敏感,$K=2$ 的成对反馈即可,显著降低算力。代码开源。

劣势 / 局限

  • 关键理论结论依赖若干正则条件与近似(如 $g$ 与协方差只在 $t=1$ 处精确对应;推导在附录),在较低噪声水平的状态上是近似引导。
  • 论文承认对「为什么恢复力主导参数梯度」只给出现象,详细分析留给未来(§3.2)。
  • trust-region 掩码阈值 $\delta=10^{-5}$ 全程固定,且需要 ODE→SDE 的参考噪声 $\sigma_{\rm ref}$ 来构造 KL 代理;这是额外的设计成分。
  • 对比主要在 GenEval / OCR / PickScore / VideoReward 这类自动奖励上,奖励黑客与人工评测的对齐情况本站未核实(待确认)。
  • 全文中未见单独的 Limitations 小节(待确认:可能在附录)。

与工业 AI 的关系(作者判断,非论文结论)

工业场景里的生成式模型相关度不高,但这篇的方法论可迁移:当「奖励是黑盒、延迟、只在整条输出后给出」(例如整条工艺曲线或仿真结果的评分)时,把一组候选与奖励聚合为一个共享改进方向、并用「两次更新」控制漂移,是一种比逐步 RL 更省样本的思路。可作为数字孪生中「生成式工艺曲线 / 异常样本合成模型」的后训练参考;目前没有工业数据验证。

待确认事项

  • 待确认:论文 Algorithm 1 逐行内容 依据:文本抽取中该算法框仅剩标题,伪代码是据 §4.1–§4.3 文字整理(2026-10-31)
  • 待确认:是否有人工评测验证奖励提升与真实观感一致 依据:正文展示的均为自动奖励模型分数,未见人工评测描述(2026-10-31)
核验:arXiv 摘要页标题与提交日期(2026-10-08)核对一致;GitHub 仓库 HTTP 200;已读 §1–§5.2(含式 1–8 与 Table 1)。(2026-10-31(北京时间))
2026-10-06arXiv 论文第1期 · 2026-10-31

TRACE: Rollout-Guided Quantization-Aware Training for FP4 Reinforcement Learning of MoE Language Models

TRACE:面向 MoE 模型 FP4 强化学习的「rollout 引导」量化感知训练
Xin Wang, Hao Yu, Zhengyang Zhuge, Bochao Mao, Zheng Li, Junda Feng 等共 12 位(HF 页面机构标注:Qwen;致谢:Alibaba Research Intern Program)
一句话通俗:RL 里用 FP4 做采样(rollout)省时间,但训练端与采样端对同一个数「舍入」得不一样,会放大偏差;TRACE 让训练端的舍入方向去「照抄」采样端,并只传一点点必要信息。
阅读模式:

报告的主要结果(均为原文自报,非本站复现)

(原文自报,Table 1:Qwen3.5-35B-A3B,联合 NVFP4 W/A + NVFP4 KV rollout。)

方法LiveCodeBenchAIME24AIME25HMMT25平均
BF16 rollout(参考)67.183.881.367.574.9
QAT55.170.763.449.259.6
QaRL54.173.357.949.258.6
QUADS60.480.575.459.068.8
TRACE66.486.378.570.075.3

Table 2(更大模型,BF16 / QAT / QUADS / TRACE):Qwen3.5-122B-A10B(DeepSWE)33.4 / 28.8 / 29.1 / 33.0;Qwen3.8-Flash-Next(Terminal-Bench)68.8 / 60.4 / 66.4 / 70.6;Qwen3.8-2.4T-A95B(GDPval)90.3 / 88.7 / 85.9 / 90.2。

效率:解码吞吐最高达 BF16 rollout 的 5.4×(128K 输出长度);端到端 RL 步时间从 664 增至 713(+7.4%);rollout 引导开销:权重侧 4%、KV 侧 2%,KV 反量化 8%,前向 86%。

通俗解释

强化学习后训练要先让模型大量「生成答案」(rollout),这一步很耗时间和显存,所以大家想把模型压成 4 位浮点数(FP4)来加速。但训练时用的是另一套高精度计算,两边的数值总是对不上;FP4 的「格子」又很粗,两个原本只差一点点的数,可能被舍入到相邻两格,差距一下被放大。MoE 模型还会因路由选择不同进一步放大。

TRACE 的做法:采样时把「这个数被舍入到了哪一格」记下来,训练做量化感知(QAT)时,遇到要在两个相邻格子里二选一,就选采样端选的那一个,而不是按「四舍五入」。这样两端的偏差不会被舍入放大。

但是全量记录太大:例子里一步 RL 要多存约 51 TB。论文发现 99% 的不一致只差相邻一格,所以只需要传「尾数的 1 位 + 缩放系数」,并且只传后半部分层。

专业解释

问题:FP4(W4A4)rollout 能显著加速与省显存,但训练与 rollout 两条量化执行路径的数值差异会造成 policy mismatch,MoE 下还与专家路由相互作用而放大。已有方法(QUADS、Rollout-ResQ)通过提高每条路径各自的量化精度(降低 $\|X-Q(X)\|_F$)来间接缓解,但各自变准并不等于两端更一致——论文用具体数值例子说明(QUADS 反而把差距从 0 变为 0.10;另一例保持 0.99 对 1.0)。

TRACE(Train-Rollout Quantization Alignment via Compact GuidancE):

(1) Rollout-guided QAT:关键洞察是不一致的 FP4 舍入会放大原本微小的 BF16 差异(论文举例:60.24 vs 59.76 仅差 0.48,归一化后 2.51 与 2.49 位于舍入边界两侧,被舍入到 3 与 2,差距放大到 24)。训练侧在两个相邻 E2M1 码字 $\{q_-,q_+\}$ 中选择最接近 rollout 侧实际码字的那个,取代 RTN;在候选集相同的前提下,对每个被记录位点局部差异不会大于 RTN(式 3);注意该局部保证不推出整网或策略层面的单调下降。

(2) Mantissa-only 通信:>99% 的失配值只相差一个相邻码字,因此只需传尾数的少量比特与 scale;并且向更低码字的修正多发生在深层,所以只缓存后半部分层。以 Qwen3.5-35B-A3B 为例,保留 1 位尾数与摊销后的 scale 约为每 token 7.5 KB,较缓存完整量化激活约小 6×。

作用范围:rollout 时仅量化 MoE 路由专家与 softmax-attention 层的 KV cache,其余模块保持 BF16;训练用 GRPO + R3(回放路由)+ token 级重要性比(裁剪上界 5)。

详细算法

1. 问题设定

RL 后训练的两条执行路径:rollout(推理引擎,用 FP4 加速;这里用 SGLang)与训练(Megatron,BF16 前向/反向,QAT 时前向做 FP4 fake quantization)。对同一轨迹、token 位置与量化位点(site),记训练侧激活 $X_{\rm train}$ 与 rollout 侧激活 $X_{\rm rollout}$。目标:降低两条路径量化后的差异,同时不引入难以承受的存储 / 通信开销。

2. 关键公式

局部量化差异(式 1):

$$\mathcal D_{\rm act}=\big\|Q^{\rm train}_{\rm FP4}(X_{\rm train})-Q^{\rm rollout}_{\rm FP4}(X_{\rm rollout})\big\|_F$$

既有方法的目标(只降低各自量化误差):$\mathcal L_{\rm error}(X)=\|X-Q(X)\|_F$。

Rollout 引导舍入(式 2;$q_\pm$ 为 rollout 侧 scale 下训练激活相邻的两个归一化 E2M1 码字,$q_{\rm rollout}$ 为对应 rollout 激活实际得到的码字):

$$q_{\rm TRACE}=\arg\min_{q\in\{q_-,q_+\}}\big|q-q_{\rm rollout}\big|$$

局部不劣于 RTN(式 3):

$$|q_{\rm TRACE}-q_{\rm rollout}|\ \le\ |q_{\rm RTN}-q_{\rm rollout}|$$

相对额外差异(诊断量):

$$\frac{D_{\rm FP4}-D_{\rm BF16}}{D_{\rm BF16}}$$

($D$ 为训练—rollout 激活 MSE。)压缩通信下的参考重建:选择低 $b$ 位码与记录 $c$ 一致、且幅值最接近 $|z|$ 的 E2M1 幅值,符号取训练激活的符号;仅作局部引导目标,而非精确还原。

3. 逐步伪代码
# 据 §3 与 A.1 的文字描述整理 (论文 Algorithm 1 的框内行文本在抽取中缺失)
Rollout 阶段 (SGLang, FP4 路由专家 + FP4 KV):
    for 每个生成的 token, 每个量化位点 (MoE 专家输入 / 后半层 KV):
        记录 rollout 侧的 (尾数低 b 位 [b=1], scale)         # 后半层才记录
        激活侧: 先写入临时 GPU buffer -> 异步卸载到 CPU; KV 侧: 请求结束后从 FP4 KV cache 收集
    序列化并按 (trajectory, token, site) 索引保存

QAT 训练阶段 (Megatron, BF16 反向):
    for 每个微批:
        载入对应记录并按 (trajectory, token, site) 配对
        for 每个被记录位点的训练激活 z:
            (q-, q+) = BracketFP4(z)                           # 相邻 E2M1 舍入候选
            ref = Reconstruct(记录 c, scale s, z)              # 低 b 位码匹配的最近 E2M1 幅值, 符号取自 z
            q = 距 ref 更近的候选                              # 代替 RTN
            z_q = STE(z, q * s)                                # 前向用 q, 梯度穿透到 z
        GRPO + R3 路由回放 + token 级重要性比 (clip 上界 5, max_version_diff=3)
4. 训练 / 推理流程

模型:Qwen3.5-35B-A3B(推理 RL,400 步)、Qwen3.5-122B-A10B(代码 RL,50 步,DeepSWE v1.1)、Qwen3.8-Flash-Next(125B 总 / 6B 激活,代码 RL,400 步,Terminal-Bench 2.1)、Qwen3.8-2.4T-A95B(长程 RL,50 步,GDPval)。基线:vanilla NVFP4、QAT、QaRL、QUADS、Score Centering;MXFP4 变体;以及 BF16 训练后再做 post-hoc FP4(4over6、H-Scale)。设置:VeRL 异步 RL + Megatron + SGLang,disaggregated 部署;GRPO + R3 + token 级重要性比(上界 5)。每步采样 $256{\times}16$ / $128{\times}16$ / $64{\times}16$ / $64{\times}8$ 条轨迹。

与既有方法对比

  • QUADS / Rollout-ResQ(提高各自量化精度):目标是 $\|X-Q(X)\|_F$,不直接缩小两端之间的差异;论文以数值例子说明可能反而扩大。
  • Vanilla NVFP4 + QAT / QaRL:在 Qwen3.5-35B-A3B 联合 NVFP4 W/A + KV 下,平均 59.6 / 58.6,远低于 BF16 的 74.9。
  • Post-hoc FP4(先 BF16 训练再量化):论文称 TRACE 的最终 FP4 性能更强(§4.5,本站未逐项核对数值)。
  • 全量引导(保存完整 rollout 量化信息):效果有但吞吐显著下降;mantissa-only 把缓存降到约 1/6,吞吐接近 vanilla FP4 rollout。

优势

  • 问题诊断清晰:从「舍入边界放大」这个具体机制出发,指出间接降低量化误差并不等于缩小两端差异。
  • 方法极简:只改变候选码字选择,局部上不劣于 RTN;配合 mantissa-only 压缩,把通信 / 存储降低约 6×。
  • 覆盖 4 个 MoE 模型(35B ~ 2.4T 总参)与推理、代码、长程三类 RL 任务;训练动态(log-prob 差、熵、响应长度)有诊断图。
  • 额外训练开销小(+7.4% 步时间),且 FP4 rollout 的吞吐收益基本保留。

劣势 / 局限

  • 论文明确:局部保证不等于整网差异或策略层面发散的单调下降;也不能消除由异步训练的策略陈旧(staleness)引起的激活差异。
  • 依赖 rollout 与训练的精确配对(trajectory/token/site)与持久化 / 通信管线,对 RL 系统工程侵入较深;需要同一量化 scale 才能对齐。
  • 仅量化路由专家与 softmax-attention 的 KV,其余保持 BF16;对全模型 W4A4 的适用性没有展示。
  • Qwen3.8-2.4T-A95B 只训练 50 步;TRACE 90.2 略低于 BF16 90.3,长程任务上的长期稳定性证据较短。
  • 无独立 Limitations 小节(本站文本抽取未发现);作者来自 Qwen / 阿里实习项目,模型与基础设施高度相关,外部复现难度大(待确认)。

与工业 AI 的关系(作者判断,非论文结论)

对我们的相关性主要在「用更少 GPU 做 RL 后训练」:如果要在企业内部对运维 / 工艺大模型做 RL 微调,rollout 占大头,FP4 采样 + 训练对齐是潜在的降本手段。但门槛高:需要 Blackwell 级 FP4 硬件与 SGLang/Megatron 级别的 RL 基础设施,当前阶段更适合关注而非落地。

待确认事项

  • 待确认:论文 Algorithm 1 的逐行伪代码 依据:文本抽取只保留了说明段,伪代码框缺失;上方伪代码为按 §3 / A.1 文字整理(2026-10-31)
  • 待确认:是否有代码开源 依据:HF 页面无 githubRepo;全文抽取中未见链接(2026-10-31)
  • 待确认:§4.5 与 post-hoc FP4 的具体数值对比 依据:本站未摘录该节表格(2026-10-31)