← 返回 2026-05-05

T²PO:不确定性引导的探索控制,用于稳定多轮智能体强化学习 T^2PO: Uncertainty-Guided Exploration Control for Stable Multi-Turn Agentic Reinforcement Learning

Haixin Wang, Hejie Cui, Chenwei Zhang, Xin Liu, Shuowei Jin, Shijie Geng, Xinyang Zhang, Nasser Zalmout, Zhenyu Shi, Yizhou Sun 📅 2026-05-04 👍 10 2026-07-13 08:36
LLM智能体 不确定性估计 多轮交互 强化学习 探索-利用平衡

通过token级思考截断与turn级动态重采样,让多轮智能体RL训练摆脱过思考困境。

前置知识

多轮智能体强化学习

智能体在外部环境中以多轮(multi-turn)方式交互,每轮产生状态-动作-奖励三元组 $(s_k, a_k, r_k)$,最终形成完整轨迹 $\tau=\{(s_1,a_1,r_1),\dots,(s_K,a_K,r_K)\}$。与传统单轮RL相比,多轮设置存在奖励稀疏(终端才有反馈)和信用分配困难(数千个token需逐一归因)两大挑战。本文涉及的关键操作包括:动作空间来自LLM词表 $\mathcal{V}$、策略 $\pi_\theta(a_k|s_k,q)$ 同时输出 thinking tokens $a_k^c$ 和 action tokens $a_k^o$(用特殊标签包裹)。

本文核心问题正是多轮交互中广泛出现的训练崩溃现象,只有理解了稀疏奖励和长视野信用分配的困难,才能理解为什么作者必须放弃'修改奖励函数'的传统思路,转而直接干预 rollout 阶段的探索行为。

策略熵与置信度

在LLM生成第 $t$ 个 token 时,策略输出词表上的概率分布 $p_t=\pi_\theta(\cdot|R_{<t},x;\mathcal{T})$。两种常用不确定性度量包括:(1) Shannon 熵 $H_t=-\sum_i p_t^{(i)}\log p_t^{(i)}$;(2) 置信度 $C_t=-\frac{1}{j}\sum_{i=1}^j \log p_t^{(i)}$,即 top-$j$ token 平均负对数概率。两者的取值范围和盲区不同:熵在近均匀分布处判别力差,置信度忽略尾部概率分布。

本文整个方法的基础是构建一个'自校准'的不确定性信号 $M_t$,融合熵与置信度二者,因此必须先理解它们各自的缺陷才能理解为什么要做融合。

GRPO与组内优势估计

Group Relative Policy Optimization 是一种无评论家(critic-free)的策略优化方法,对同一 prompt 采样 $G$ 个 rollout,在组内通过回报归一化得到相对优势 $A=\frac{R-\mu}{\sigma}$。GiGPO(Group-in-Group Policy Optimization)将其推广到多轮设置:先计算全轨迹优势 $A(\tau_k^i)$,再按相同状态 $s$ 出现的 turn 组成子组计算 turn 级优势 $A_{\text{turn}}$,最终融合 $A'=\text{clip}(\rho_\theta A', 1\pm\epsilon)A' -\beta D_{KL}$。

T2PO 是在 GiGPO 框架之上的 plug-and-play 增强,本文报告的所有对比方法和融合式 advantage 计算(组内组内优势)都依赖于这一基础。

探索-利用权衡

强化学习的核心张力:在尝试新动作(探索)和利用已知最优策略之间分配算力。Mehlhorn 等人 2015 年总结了人类与动物决策研究中的这一权衡。本文将违反这一权衡的失败模式定义为 'hesitation'(犹豫):token 级表现为过思考、entropy 信息增益饱和;turn 级表现为重复无生产的相同推理。

本文将多轮 RL 训练崩溃的根因定位为系统性的探索-利用失衡,而非简单归咎于奖励稀疏或策略梯度方差,这是论文最核心的洞察。

研究动机

多轮智能体强化学习训练普遍存在崩溃问题(图1显示 SOTA 基线在不同随机种子下成功率骤降、KL 散度和梯度范数爆涨)。现有缓解方案包括细粒度信用分配(如 GiGPO 在组内再做 turn 级归一化)、内部/过程奖励建模(如基于熵的奖励 shaping)、轨迹级失败过滤(如 DAPO 的动态采样滤除空 prompt 组),但这些方法要么在粗粒度 trajectory 层操控、要么通过隐式奖励 shaping,对超参数和 rollout 分布高度敏感,最终仍频繁引发训练崩溃。具体而言:token 层面,LLM 智能体常出现 over-thinking,生成的信息量饱和却仍在采样噪声中累积的 token;turn 层面,智能体在早期偏离成功动作空间后,不断执行重复无生产的轮次,受限于固定预算几乎无法恢复。本文用 'hesitation'(犹豫)一词命名这一系统性违反探索-利用权衡的失败模式。

本文的目标是本文旨在同时提升训练稳定性与探索效率,使效果与效率不再是零和博弈。文章定义了双层不确定性控制的总体目标:在 token 层,监控不确定性动态,一旦边际不确定性变化低于阈值即触发'思考干预'终止推理;在 turn 层,识别探索进展可忽略的交互并动态重采样该 turn,避免无效 rollout 浪费算力。最终在 WebShop、ALFWorld、Search QA 三大基准上验证该方法在训练稳定性和性能上的显著提升。

与已有工作不同的是,本文的独特切入角度是根因诊断——不是把训练崩溃归因于奖励稀疏或梯度方差,而是诊断为 'hesitation' 这一未被命名过的探索低效行为模式。基于此,提出了'自校准不确定性信号'(融合 Shannon 熵与 top-$j$ 置信度,缓解二者各自的判别盲区),并在 token 与 turn 两个细粒度层级同时施加显式控制,而非依赖粗粒度 trajectory 过滤或单尺度启发式奖励 shaping。此外,T2PO 是 plug-and-play 的,可无缝替换到 GiGPO、GSPO 等不同策略更新方案上,无需修改核心优化算法。

核心方法

T2PO 的整体思路是'用不确定性说话':与其在奖励函数上做手脚,不如直接观察模型在 rollout 阶段对自身预测稳定性的变化,自动判断'什么时候该停思考、什么时候该重采样'。技术上分为两条独立但可叠加的流水线——Token-Level Thinking Intervention(TTI)和 Turn-Level Dynamical Sampling(TDS)。TTI 在解码时构造一个融合熵与置信度的标量信号 $M_t$,当滑动窗口内的边际变化 $\Delta_t^k=|M_t^k-M_{t-1}^k|$ 的均值低于 $\epsilon$ 时,强制覆盖 logits 输出 推理终止符,让模型转去生成 动作;TDS 则把所有 token 信号聚合到 turn 级,得到 $\Phi^k=\prod_{t=1}^{T_k} M_t^k$(几何平均),当相邻 turn 间变化 $\Gamma^k=|\Phi^k-\Phi^{k-1}|$ 过小时,整段 turn 丢弃并重新采样。最后 policy update 用 GiGPO 的 hierarchical advantage 完成。

核心创新在于'自校准不确定性信号 $M_t=\alpha\tilde H_t+(1-\alpha)(1-\tilde C_t)$'的构造以及对'犹豫'这一失败模式的显式控制。已有方法的本质区别在于:(1) SEED-GRPO 等把熵直接塞进优势估计,DeepConf 仅用置信度做硬过滤——它们都依赖单尺度启发式,而 $M_t$ 通过在轨迹内归一化($\tilde H_t=(H_t-H_{\min})/(H_{\max}-H_{\min})$,$\tilde C_t$ 类同)并加权融合,同时保留 top-1 主导层的判别力和熵对尾部分布的敏感性,contour 几何非退化;(2) DAPO 在 prompt 组粒度过滤、SimpleTIR 过滤 void turn、长度奖励施加全局启发——它们都在粗粒度层做事,而 TTI+TDS 首次在 token 级和 turn 级同时做精细干预,且不用修改奖励函数。

方法步骤详情

T2PO 的执行流程可分解为五步:第一步,rejective fine-tuning 作为冷启动,用基础 Qwen3 在目标环境跑 rollout 收集 $\mathcal{D}_{RFT}=\{(h_k,a_k)\}$,仅保留终端分数超阈值的轨迹,从中抽取 state-action 对做单 epoch SFT,得到 $\mathcal{L}_{RFT}=-\mathbb{E}_{(h_k,a_k)\sim\mathcal{D}_{RFT}}[\log\pi_\theta(a_k|h_k)]$。第二步,构造自校准信号:在每个解码步 $t$ 计算 $H_t=-\sum_i p_t^{(i)}\log p_t^{(i)}$ 与 $C_t=-\frac{1}{j}\sum_{i=1}^j\log p_t^{(i)}$($j$ 为 top-token 数),按轨迹内极差归一化得 $\tilde H_t,\tilde C_t$,再融合 $M_t=\alpha\tilde H_t+(1-\alpha)(1-\tilde C_t)$,实验中 $\alpha=0.4$ 最优。第三步,TTI 干预:起算忽略前缀 $L_{\min}$ 内的 token,监控滑动窗口 $\frac{1}{N+1}\sum_{i=0}^N\Delta_{t-i}^k<\epsilon$(实验中 $\epsilon=10^{-4}$)是否成立,一旦触发即把 softmax 前 logits 中 token id 151668(对应 标记)的值置 $+\infty$,其余 $-\infty$,确定性发射推理终止符;其后注入固定队列 $Q=[\text{ 标记},\text{ 换行},\text{}]$;flag $I_{\text{stop}}$ 一次性激活避免重复触发,并以 $L_{\max}$ 作为兜底上限。第四步,TDS 动态采样:把 turn $k$ 内所有 token 信号聚合 $\Phi^k=\prod_{t=1}^{T_k} M_t^k$,相邻 turn 变化 $\Gamma^k=|\Phi^k-\Phi^{k-1}|$;若 $\Gamma^k<\eta$(实验 $\eta=10^{-3}$),则在同一状态 $s_k$ 下重新 rollout 新推理轨迹直至 $\Gamma^k\geq\eta$ 或预算 $B_{\max}$ 用尽。第五步,policy update:用 $R(\tau^k)=\sum_{j=k}^K\beta^{j-k}r_j$ 做 turn-level 折扣回报,融合 GiGPO 的 hierarchical advantage $A'(a_k^i)=A(\tau_k^i)+\omega\cdot A_{\text{turn}}(a_k^i)$,做标准 PPO 风格裁剪目标 $J(\theta)=\mathbb{E}_h[\min(\rho_\theta A',\text{clip}(\rho_\theta,1\pm\epsilon)A')]-\beta D_{KL}(\pi_\theta\|\pi_{\text{ref}})$。

技术新颖性

技术新颖性体现在三点:(a) 提出 'hesitation' 这一诊断概念并将训练崩溃归因于探索低效而非稀疏奖励或梯度方差,改变了问题建模视角;(b) 设计 $\tilde H_t\oplus(1-\tilde C_t)$ 这一非退化的不确定性融合公式,用数学方式显式补偿单一指标的盲区,论文图 3 通过 contour 可视化展示了融合后区分度更高;(c) 在 token 和 turn 两层做 plug-and-play 控制而不引入额外模型参数或任务特定启发式,与已有的 trajectory 级过滤(SimpleTIR)、静态奖励 shaping(lengthy reward)、固定预算(hard thinking budget)形成本质区别。此外算法具有通用性,文中附录 D.1 给出在 GSPO 上同样将成功率从 85.18 提升到 91.79(+7.76%)的证据。

Overview of the proposed Uncertainty-Guided Exploration Control at both token and turn levels
Figure 2: Overview of the proposed Uncertainty-Guided Exploration Control at both token and turn levels
Contour comparison of $H_t$, $C_t$ and $M_t$
Figure 3: Contour comparison of $H_t$, $C_t$ and $M_t$
(a) Uncertainty dynamics of self-calibrated signal $M_t$ on response length; (b) Word Cloud of Uncertain Tokens; (c) Truncated Tokens by TTI Rule
Figure 4: (a) Uncertainty dynamics of self-calibrated signal $M_t$ on response length; (b) Word Cloud of Uncertain Tokens; (c) Truncated Tokens by TTI Rule

实验结果

在 WebShop 基准上(基于 Qwen3-4B-RFT)T2PO 取得 Task Score 93.84±0.22 和 Success Rate 81.64±0.39,明显反超 GiGPO+DAPO 的 86.54/74.02;基于 Qwen3-8B-RFT 时进一步做到 91.65/82.42,相比 GiGPO 的 88.76/77.92 提升约 3 个百分点。ALFWorld 上 T2PO 把 ALFWorld 整体成功率推到 90.23(4B)和 92.41(8B),相对 GiGPO+DAPO 提升约 8-12 个点;在 Search QA 上,T2PO 在 4 个单跳/多跳集(NQ、TriviaQA、PopQA、HotpotQA、2Wiki、MuSiQue、Bamboogle)平均 EM 拿到 54.93,较 GiGPO 的 52.97 稳定提升;其中 MuSiQue 上比此前最佳成绩翻倍以上(16.64 vs 13.40)。消融实验(表3)剥离 RFT、TTI、TDS 后在 WebShop 分别退化到 Task Score 79.28/81.28/72.40 和成功率 61.32/73.27/63.67,说明三个模块缺一不可。表4与其他思考控制策略对比中,lengthy reward 仅得到 77.96/65.87,short-CoT cold-start 82.39/71.29,hard thinking budget 84.96/79.21,void turn filtering 85.17/76.20,T2PO(93.84/81.64)全面领先。融合系数 $\alpha$ 敏感性实验($\alpha=0.2,0.4,0.6,0.8$)显示 $\alpha=0.4$ 最优,$\alpha=0.2$ 时成功率仅 79.36,说明平衡熵置信度权重对控制质量至关重要。效率侧(图5/6)显示 T2PO 在 WebShop 成功轨迹上平均节省约 20% 的 token、~25% 的交互 turn;在 ALFWorld 上则达到 ~16% 的 turn 削减。奖励分解(表8)进一步显示 T2PO 在 title score(65.61)和 reward option(57.49)等精细组件上的优势最大,表明去掉冗余探索后模型能更聚焦于精准信息获取。

Comparison with different policy optimization methods on WebShop and ALFWorld
Table 1: Comparison with different policy optimization methods on WebShop and ALFWorld
Performance on search-augmented QA tasks
Table 2: Performance on search-augmented QA tasks
Ablation study of key modules on WebShop
Table 3: Ablation study of key modules on WebShop
Ablation of alternative thinking-control methods on WebShop with Qwen3-4B-RFT
Table 4: Ablation of alternative thinking-control methods on WebShop with Qwen3-4B-RFT
Key training hyperparameters for Agentic RL experiments
Table 5: Key training hyperparameters for Agentic RL experiments
Task Performance and Exploration Efficiency comparison
Figure 5: Task Performance and Exploration Efficiency comparison
Effect of different maximum response length on GiGPO vs T2PO
Figure 6: Effect of different maximum response length on GiGPO vs T2PO
查看结构化数据
任务指标本文基线提升
WebShop(4B) Task Score 93.84±0.22 GiGPO+DAPO: 86.54±9.81;GiGPO: 86.03±4.18 +7.81
WebShop(4B) Success Rate 81.64±0.39 GiGPO+DAPO: 74.02±8.18;GiGPO: 73.83±3.04 +7.62
ALFWorld(4B) Overall Success Rate 90.23±1.38 GiGPO+DAPO: 80.86±1.37;GiGPO: 80.47±2.43 +9.37
WebShop(8B) Task Score 91.65±0.84 GiGPO: 88.76±5.63 +2.89
ALFWorld(8B) Overall Success Rate 92.41±1.42 GiGPO+DAPO: 89.06±4.76 +3.35
Search QA(4B,平均7个数据集) Exact Match 54.93 GiGPO: 52.97 +1.96
Search QA - MuSiQue EM 16.64 GiGPO: 13.40 +3.24

局限与改进

作者在文中坦承的限制主要有两点:(1) off-policy staleness,turn-level 轨迹拆分意味着 learner 更新参数 $\theta_\mu$ 时,rollout worker 仍可能用 stale $\theta_{\mu-\delta}$ 采样,作者用 staleness ratio $\rho_{\text{stale}}=\delta/(1+\delta)$ 量化,但表6结果显示即便增大 $B_{\text{rollout}}$ 或 prompt 组,性能几乎不变(92.85→93.84),说明实际影响有限。(2) 几个超参数($\epsilon,\eta,N,\alpha$)依赖经验设定,论文通过敏感性实验给出推荐值但未做自动调优。我观察到的一个独立局限是:WebShop 和 ALFWorld 的实验都在 Qwen3-RFT 基础上做,并未展示直接对 Llama-3、DeepSeek 等其他模型系列的可迁移性,且所有对比都基于相同的 RFT 初始化和格式惩罚,差异主要来自算法本身——若换成不同 RL pipeline(如 AReaL 的异步架构),还需验证两层控制信号是否依然稳定。

独立分析的弱点

独立分析三个具体弱点:(1) Token-level 一次性激活约束 $I_{\text{stop}}$ 过于保守:当前实现每个 turn 只允许截断一次,但这对超长 CoT 不友好——遇到多段反复犹豫的复杂任务(如 ALFWorld 'pick two & place')后续的低信息量段无法再次压缩,suggested 改进是改为 budget-based 多次干预,每次触发后允许在新产生的 移动后的滑动窗口内重新监测。(2) 融合系数 $\alpha=0.4$ 在 WebShop 上最优,但 Search QA 上若 token 分布差异大,固定 $\alpha$ 可能不是全局最优,改进方向是引入 per-trajectory adaptive $\alpha$(例如用任务早期若干步的不确定性方差动态调整)。(3) TDS 的重采样使用同策略同状态刷新,最多 $B_{\max}$ 次,但当 $\Gamma^k$ 持续小于 $\eta$ 时容易陷入'卡死'等待 budget 耗尽,浪费计算;可以考虑在重采样前注入一次 temperature scale $\tau\leftarrow\tau\cdot\gamma$ 来增加多样性,或在 $\Gamma$ 增大前提前 fail-fast。

未来方向

作者明确提出的方向包括:扩展到异步 RL 架构(AReaL-style)和更长视野的检索-推理混合管线(如 Agentic RAG)。基于本文结果的延伸方向还有:(a) 把不确定性控制泛化到 multimodal agent(视觉动作空间下的不稳定性度量),因为 $M_t$ 的构造天然支持任意概率分布;(b) 探索 self-calibrated 信号在多智能体协作中的 federated uncertainty 表达,每个智能体维护本地 $\Phi^k$ 并通过对比抑制冗余通信;(c) 与 process reward model 结合,使用 TTI 截断点作为过程标签自动生成 PRM 训练数据,进一步降低对人工标注的依赖;(d) 把它和 test-time scaling 联动,让 $M_t$ 兼做 uncertainty-aware self-consistency 投票。

复现评估

复现友好度较高:论文承诺开源全部代码(GitHub 仓库 https://github.com/WillDreamer/T2PO ),框架基于 verl 与 vLLM 这两个主流 RL 推理栈;模型采用公开 Qwen3-4B/8B,并给出完整超参表(表5):WebShop/ALFWorld/Search QA 学习率统一 $1\times10^{-6}$,minibatch 128 或 256,KL 系数 0.01,格式惩罚 $\lambda_{\text{fmt}}=0.1$,监控窗口 $N=20$ 或 $15$,$\epsilon=10^{-4}$,$\eta=10^{-3}$,$\alpha=0.4$,Thinking budget 450 tokens。最大交互步数 WebShop 15、ALFWorld 50、Search QA 4;硬件需求 8×H100 GPU 单节点,单次完整 run 表1在 WebShop 上预计数十小时,ALFWorld 类似。RFT 冷启动只需 1 epoch;总体而言复现工程量集中在改造 verl 的 decoding/rollout pipeline 以支持 token 级 logit overwrite,难度中等。