APPO:面向智能体强化学习的过程级策略优化 APPO: Agentic Procedural Policy Optimization
在思维过程细粒度决策点上做分支与信用分配的智能体RL算法
前置知识
智能体强化学习 (Agentic RL)
用强化学习训练 LLM 的多轮工具调用能力。模型作为策略 $\pi_\theta$ 与工具集 $T$ 交互,仅在轨迹末端获得稀疏奖励 $r_\phi$,目标为最大化期望奖励同时约束 KL。
APPO 工作的基本设定;理解轨迹稀疏奖励如何导致信用分配困难是阅读本文的关键
信用分配 (Credit Assignment)
在仅获得末端奖励的多步决策中,判断轨迹中每一步(或每个 token)应当为最终成败承担多少正向或负向责任。困难在于成功轨迹中可能包含大量无效步骤,失败轨迹中也可能有正确决策,导致学习信号噪声大、方差高。
APPO 的核心动机就是改进 agentic RL 中的信用分配;从粗粒度单元迁移到细粒度决策点
重要性采样比与 PPO 裁剪
重要性采样比 $\rho_i(\theta)=\pi_\theta/\pi_{\mathrm{old}}$ 衡量新旧策略差异,PPO 用 clip 限制在 $[1-\epsilon,1+\epsilon]$。APPO 用其构造未来价值 $\Omega_{n,i}$。
理解 $\Omega$ 如何表征当前 token 对下游策略分布的影响是把握 Branching Score 的基础
Token 熵与高熵分支
Token 熵 $H_t=-\sum_j p_{t,j}\log p_{t,j}$ 反映词表上的局部不确定性。ARPO、GIGPO 等方法把 top-k 高熵 token 作为分支位置。APPO 的先导实验表明:单纯高熵并不等同于决策重要性,很多高熵来自词汇长尾而非推理转折。
APPO 正是在这一点上改进,把熵与未来价值相乘得到 Branching Score
研究动机
现有面向智能体的强化学习方法面临严重的信用分配问题:在 RLVR 范式下,奖励只在完整轨迹末端给出,导致学习信号粗糙且纠缠。当前的改进路线是在中间位置扩展分支——比如 Tree-GRPO 在思考-动作步骤随机分支、ARPO 在每次工具调用后的高熵 token 处重采样——但这些方法都把分支单元限定在工具调用边界或固定工作流。作者在 ToolStar 54K 数据集上的先导分析(图 1a)发现,高熵 token 在整个思维过程中广泛分布而非集中在工具调用附近;进一步统计(图 1b.1)显示,把每个 token 当作分支点并按熵分桶后,分支轨迹的平均准确率并没有随熵单调上升,说明高熵 token 中相当一部分是词汇层面的不确定性而非任务相关的关键决策。这使得基于工具调用或纯熵的分支会浪费大量预算在低价值位置,限制了对真正影响成败的关键决策点的探索效率。
本文的目标是本文旨在把智能体强化学习中的分支与信用分配单元从粗粒度的工具调用/工作流粒度下移到生成序列中的细粒度决策点。具体而言,作者希望 (1) 找到一种比 token 熵更可靠的「高影响力决策点」识别指标;(2) 让模型围绕这些决策点做有针对性的分支,从而在固定 rollout 预算 $M$ 下获取更密集、更有信息量的监督信号;(3) 在策略优化阶段为高影响力决策赋予更大的优势权重,使得真正影响下游成败的关键步骤获得更多学习信号。
与已有工作不同的是,APPO 的独特切入角度是把「分支点选择」与「信用分配」统一到同一个未来价值信号 $\Omega$ 上。$\Omega_{n,i}=\exp\sum_{i'\ge i}\gamma^{i'-i}\log\rho_{i'}(\theta)$ 通过新旧策略在当前位置之后的累积对数似然比,量化当前 token 触发的下游策略分布偏移;再与 token 熵在 rollout 内做 z-score 归一化相乘得到 Branching Score(式 5),同时在优势函数里以 $1+b\cdot\hat{A}^{\mathrm{fut}}$ 形式放大高 $\Omega$ 决策的权重。这样既解决了「在哪里分支」,又解决了「分支的功劳分给谁」,并在 Theorem 3.1 与 3.2 中分别给出方差缩减与策略改进下界的理论支撑。
核心方法
APPO 的整体思路是「先识别高价值决策点,再围绕它们分支,最后用未来感知优势放大其贡献」。直觉上,作者相信一条长轨迹中真正决定成败的只有少数几个关键决策 token(如「先搜索再验证」),但它们藏在长思维过程中;与其在工具调用边界或随机位置浪费分支预算,不如直接定位这些「潜变量决策点」。技术路线分四步:(1) 用当前策略 $\pi_\theta$ 与初始策略 $\pi_{\mathrm{old}}$ 计算每个 token 的未来价值 $\Omega$;(2) 将 $\Omega$ 与熵 $H$ 在 rollout 内做 z-score 归一化后相乘得到 Branching Score(BS);(3) 每个初始轨迹选 top-B 个 BS 最高的 token 作为分支点,从当前 mini-batch 策略 $\pi_\theta$ 重采样后续内容并入树;(4) 训练阶段为初始回放与分支分别计算 group-relative 优势,再用 $\hat{A}^{\mathrm{fut}}$ 对优势做过程级放大。
APPO 的核心创新是 Branching Score(式 5)$\mathrm{BS}_{n,i}=Z(\mathrm{clip}(\Omega_{n,i}, 1-\epsilon', 1+\epsilon'); H_n)\cdot Z(H_{n,i}; H_n)$,它把局部不确定性(熵)与未来策略分布偏移($\Omega$)在同一轨迹内归一化后相乘,乘积高意味着「当前位置既不确定、又对下游分布有强牵引力」。这与已有方法有本质区别:Tree-GRPO 随机选分支、ARPO 只在工具调用后挑高熵 token,都没有度量当前位置对未来分布的影响;APPO 用累积对数 IS 比 $\Omega$ 把「序列先验」与「序列后验」通过策略桥接起来,从而筛掉词汇层面虚高但下游影响小的熵峰。第二个创新是优势层面的 $\hat{A}^{\mathrm{fut}}$(式 7),让分支不仅作为对比信号,还显式放大高影响力决策的梯度贡献。
方法步骤详情
APPO 执行分四阶段。(1) 初始化:用当前策略采样 $N$ 条 rollout 作为树的根,预算 $M$。(2) 评分:对每 token 算 $\Omega$(累积对数 IS 比 + 折扣 $\gamma$)与熵 $H_t$,rollout 内 z-score 归一化相乘得 Branching Score。(3) 树扩展:选 top-$B$ 个 BS token 重采样得分支,$L=1$ 时 $(B+1)N=M$。(4) 优化:$T^{\mathrm{init}}$ 与 $T^{\mathrm{branch}}$ 分别做组内归一化得 $\hat{A}^{\mathrm{base}}$,叠加 $\hat{A}^{\mathrm{fut}}$ 后用 PPO-clip 目标 $J(\theta)=\mathbb{E}[\min\rho\hat{A},\mathrm{clip}(\rho)\hat{A}]-\beta\mathrm{KL}$ 优化,分支不直接回传梯度。
技术新颖性
新颖性体现在三方面。第一是 BS 形式:$\Omega$ clip 到 $[1-\epsilon',1+\epsilon']$ 再 z-score 归一化,避免单条 rollout 极端 token 主导分支,与 PPO 风格一致;折扣 $\gamma=2^{-1/\tau}$($\tau=32$)抑制远端方差。第二是 dual-group 优势:把 $T^{\mathrm{init}}$($\pi_{\mathrm{old}}$ 生成)和 $T^{\mathrm{branch}}$(mini-batch $\pi_\theta$ 生成)拆开做组内归一化,避免不同策略分布样本混合产生偏差,这是 ARPO 等方法的隐含缺陷。第三是理论:Theorem 3.1 在「条件方差随 BS 单调」假设下证明 APPO 梯度方差上界不超过随机分支;Theorem 3.2 在 KL 受限与 $\hat{A}^{\mathrm{fut}}$ 有界条件下给出策略改进下界。
实验结果
论文在 13 个基准验证 APPO。数学与知识推理(Table 1):Llama3.1-8B 平均 57.4 较 ARPO 55.3 提升 2.1 点;Qwen2.5-7B 平均 62.2 较 ARPO 58.3 提升 3.9 点,AIME24 从 30.0→36.7、2Wiki 从 76.1→81.5。深度搜索(Table 2):Qwen3-8B 在 GAIA 取得 42.7(vs ARPO 38.8),Qwen3-14B 取得 46.6 超过 DeepSeek-R1-671B 的 25.2,HLE 从 10.0→12.3。Pass@K(图 3)显示优势随 $k$ 扩大;消融(Table 4)证实 BS、$\hat{A}^{\mathrm{fut}}$、dual-group 三组件互补。Table 3 显示 $M=16$ 时 $(N=4,B=3)$ 平均 58.1 优于 $(8,1)$ 的 57.9 与 $(2,7)$ 的 56.1,验证多样根 + 适量分支最优折中。
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| AIME24 (Llama3.1-8B) | Pass@1 | 30.0 | ARPO 23.3 / GIGPO 20.0 | +6.7 vs ARPO, +10.0 vs GIGPO |
| MATH500 (Llama3.1-8B) | Pass@1 | 69.4 | ARPO 64.6 | +4.8 |
| 2WikiMultihopQA (Qwen2.5-7B) | F1 | 81.5 | ARPO 76.1 | +5.4 |
| Bamboogle (Qwen2.5-7B) | F1 | 77.6 | ARPO 71.5 | +6.1 |
| GAIA 全部难度 (Qwen3-14B) | Pass@1 | 46.6 | ARPO 43.7 / DeepSeek-R1-671B 25.2 | +2.9 vs ARPO, +21.4 vs DeepSeek-R1-671B |
| WebWalkerQA Avg (Qwen3-14B) | Pass@1 | 43.4 | ARPO 40.5 | +2.9 |
| Humanity's Last Exam (Qwen3-14B) | Pass@1 | 12.3 | ARPO 10.0 | +2.3 |
| 10 数据集平均 (Llama3.1-8B) | Avg | 57.4 | ARPO 55.3 / GIGPO 53.2 | +2.1 / +4.2 |
| 10 数据集平均 (Qwen2.5-7B) | Avg | 62.2 | ARPO 58.3 / GIGPO 57.1 | +3.9 / +5.1 |
局限与改进
作者承认三点局限:(1) BS 在每条 rollout 内做 z-score,当 rollout 短或分支点集中时归一化稳定性下降;(2) $\Omega$ 计算需 mini-batch 内多次前向推理,计算开销高于普通 PPO;(3) 短链任务上 APPO 相对 ARPO 提升收窄到 1.4 点(GSM8K 88.6→90.0),暗示细粒度分支优势在短轨迹场景下被稀释。我额外观察到:(a) 128 batch + 16 mini-batch 对算力门槛高,小团队难复现;(b) $\beta=0$ 关闭 KL 正则的设置缺乏与 KL 调优对比,是否过度依赖 reference-free 训练尚未充分论证;(c) 双组优势分离只在 $L=1$ 下分析,$L>1$ 多轮扩展稳定性未给出;(d) Theorem 3.1 依赖「条件方差随 BS 单调」假设,实际是否成立还需在更细粒度回放分布上经验验证。
独立分析的弱点
独立分析下 APPO 至少有四个可改进弱点。第一,$\Omega$ 累加即便有折扣,远端 token 仍贡献可观噪声,特别在 $\pi_\theta$ 与 $\pi_{\mathrm{old}}$ 差距小时退化为零信号,可考虑用指数移动平均或截断窗口替代全局累加。第二,BS 中 clip 上界 $1+\epsilon'$ 选择敏感:过紧会丢失真正激进的下游策略转移,过松会放大个别极端 token。第三,双组优势分离避免分布混合但忽略跨组语义互补,分支有时可能比根轨迹更优,简单组内归一化会丢弃这种跨组信号,可考虑加入 importance-weighted 跨组对比项。第四,$\hat{A}^{\mathrm{fut}}$ 中 $b$ 取值缺乏消融,未来需研究 $b$ 与训练进度、奖励稀疏度的关系。
未来方向
作者在结论中提出两个延伸方向:(1) 将「过程」概念泛化到多模态 agent(视觉、音频)与多智能体协作场景,让 BS 跨模态度量决策影响力;(2) 把 BS 与过程奖励模型(PRM)结合,进一步细化到子步骤级别的奖励标注。基于结果可延伸的方向还包括:(a) 在 reasoning 模型(如 QwQ、DeepSeek-R1)上验证 APPO 是否仍带来 4 点提升,因为这些模型本身就具备长思维链,可能已经隐式学会了细粒度决策;(b) 探索 BS 与 MCTS 的结合——目前 BS 只选 top-B 个分支点,没有回溯到整棵树打分,可以借鉴 MCTS-TS 的 UCB 思想让 BS 自适应;(c) 把 APPO 应用到 web agent(WebArena、Mind2Web)等开放环境,验证其在非数学场景的可迁移性;(d) 研究 BS 对模型解释性的潜在价值——既然 BS 标定了「关键决策 token」,可以反推用来自动标注训练数据中的关键推理步骤。
复现评估
复现方面,论文给出项目页(Github,未明确 release 状态),主体基于 VeRL 框架;超参(batch 128、mini-batch 16、$\gamma=2^{-1/\tau}$($\tau=32$)、$\beta=0$、数学 2 epoch / 搜索 5 epoch)均明示。计算资源方面需在 Llama3.1-8B、Qwen2.5-7B、Qwen3-8B、Qwen3-14B 四种 backbone 各跑一次完整训练,按 128 batch + 14B 模型估计至少 32 张 A100 级别 GPU,搜索任务训练 5 epoch 成本更高;评测侧需 Qwen2.5-72B-Instruct + vLLM 作为 LLM-judge,额外 72B 推理资源。复现难度评估为中高:算法本身可实现,但双组优势分离、$\Omega$ 计算、VeRL 工程细节会增加调试成本;主要风险在于 $\pi_{\mathrm{old}}$ 与 mini-batch $\pi_\theta$ 的对齐、clip $\epsilon'$ 取值,以及 Bing top-10 检索结果的可重复性。
论文图表
(a) 展示 ToolStar 54K 数据集上工具调用步与过程步的 token 熵分布:横轴是熵的 bin,纵轴是 token 数,可见高熵 token 广泛分布于思维过程而非集中在工具调用附近。(b.1) 按纯熵分桶后,各桶分支轨迹的平均准确率并未随熵单调上升,说明高熵≠高决策价值。(b.2) 同样分桶但用 APPO 的 Branching Score,平均准确率随 BS 上升而上升,验证 BS 更能反映决策重要性。(c) APPO、ARPO 等基线在 10 个数据集上的 pass@1 对比,APPO 取得明显领先。
这是论文核心动机的视觉证据——读者需要看到「高熵 token 在思维过程中广泛分布」以及「纯熵无法识别真正决策点」这两个观察,才能理解 APPO 为何提出 BS
左侧 ARPO 的分支点在二维嵌入空间分布弥散、簇间区分度低;右侧 APPO 的分支点形成更紧凑、更分离的聚类,说明 APPO 探索发生在结构化推理路径而非词汇层面。
为「过程级分支 vs. token 级分支」提供定性证据,补充定量指标的不足
上方词云显示纯熵选中的 token 包含大量低频名词(march/november/cent 等);下方词云显示 BS 选中的 token 集中在「verify/break/sum/solve/should」等真正改变推理方向的动词上。
直观解释为什么纯熵分支会浪费预算,并展示 BS 的语义针对性
在 WebWalker/HotpotQA/2Wiki/Musique/Bamboogle 五个数据集上,比较 $M=4,8,16$ 下不同 $(N,B)$ 组合;$M=16$ 时 $(N=4,B=3)$ 平均 58.1 优于 $(N=8,B=1)$ 的 57.9 与 $(N=2,B=7)$ 的 56.1;同时给出 ARPO $M=16$ 的 52.7 作为参考。
为实践者提供 rollout 预算分配指导,证明「多样根 + 适量分支」是最优折中