← 返回 2026-10-19

基于信念更新与世界模型的长程智能体失败恢复 Failure Recovery via Belief Updates and World Modeling for Long-Horizon Agents

Xiyang Wu, Zhichao Liu, Shengxin zhang, Zongxia Li, Ming Li, Tianyi Zhou, Dinesh Manocha 📅 2026-10-13 👍 6 2026-11-11 18:30
世界模型 信念更新 失败恢复 知识蒸馏 终端智能体 长程智能体

用持久信念图与隐式世界模型跨回滚积累证据、指导失败恢复,并蒸馏到9B小模型降低教师成本

前置知识

长程智能体(Long-Horizon Agent)

在需要数十到上百步连续决策的任务(自主科研、计算机操作、终端运维)中工作的 LLM 智能体。环境部分可观测,上下文随交互不断增长,早期小错误会沿长轨迹传播放大,把智能体困在重复尝试或难以撤销的过早承诺里,因此失败恢复与记忆管理成为核心难题。

本文的问题设定、评测基准(Terminal-Bench 2.0/3.0、LHTB)与全部设计动机都围绕长程任务的失败恢复展开,不理解长程失败的传播特性就无法理解为什么要维护跨回滚的信念状态。

隐式世界模型(Implicit World Model, IWM)

世界模型是智能体对环境动态的内部预测器,用于行动前想象后果。隐式版本不学显式转移函数 $s_{t+1}=f(s_t,a_t)$,而是让 LLM 直接从上下文推理后果。本文 IWM 输出 $(\hat d,\hat p,\hat r)$:证据支持度、预期进展、前沿间关系。

IWM 是 BELIEFWEAVER 的核心引擎,负责假设生成、探针设计、前瞻评估与关系推断四个角色,也是后续被蒸馏进 9B 学生的对象,理解它才能理解方法主体。

信念状态与因子乘积更新

把『某假设是否仍值得追』建模为二值变量 $z_i$,用均匀先验 $\phi_i^0$ 与若干一元证据因子 $\psi_a$ 相乘再归一化得到后验 $q_i(z_i)$,思想接近贝叶斯滤波与马尔可夫随机场中的局部归一化:每个前沿的信念可独立计算,无需全局推断。

论文公式 (1)(3)(4) 的信念更新完全建立在这套因子语言上,附录中所有 $\lambda$ 超参($\lambda_{att}$、$\lambda_{rel}$ 等)都是这些证据因子的权重。

检查点与回滚(Checkpoint / Rollback)

长程执行中定期保存可恢复的环境状态,失败后回到先前检查点重新探索。各方法的差异在于回滚后『学到的经验』是否保留:丢弃记忆会重复踩坑,保留证据则每次失败都有信息价值。

BELIEFWEAVER 的关键主张是证据跨回滚持久化并按关系传播,这是它与普通树搜索回滚、AgentRewind 等方法的本质区别。

监督微调(SFT)与 LoRA

监督微调用教师模型的请求—响应对训练学生模仿其输出格式与行为;LoRA 冻结基座模型、只训练低秩适配矩阵(本文 rank 16、scaling 32、学习率 $2\times10^{-5}$、1 个 epoch),训练成本极低且适配器可插拔组合。

蒸馏第一阶段就是 LoRA SFT,让学生学会探针生成、证据判定、关系推断、条款评估四个 IWM 角色的结构化输出。

在策略蒸馏(On-Policy Distillation, OPD)

Agarwal 等(2024)提出的训练方式:学生自己执行任务产生上下文,影子教师对同样的请求给出标签但不替换学生的输出,用这些『学生实际会遇到的状态』上的教师监督继续训练。相比离线 SFT,它缓解了训练分布与推理分布的偏移。

OPD 是学生反超教师的关键(SFT+OPD 达 58.4% vs 匹配教师 53.7%),理解它才能读懂蒸馏实验并判断结果的合理性。

研究动机

长程智能体(自主科研、计算机使用、终端任务等)常因早期错误在长轨迹中传播而陷入重复尝试或难以撤销的过早承诺,而定位错误所需的证据往往要到许多步之后才浮现。现有反思方法(Reflexion、Self-Refine)与树状回滚只能做局部修复:当多条修复路线共享前提或跨步交互时,一次尝试获得的证据本应改变其他相关路线的优先级,但树结构的分支之间不共享假设、回滚后记忆常常丢失,导致智能体反复验证同样的错误方向。数据上这个问题非常直观:在强调长程执行的 LHTB 基准上,原生 Terminus-2 延续的最终得分只有 $12.3\pm8.9\%$,在 720 秒执行预算内几乎无法从失败中恢复;TB3 上也只有 $36.5\pm13.5\%$。

本文的目标是本文要把失败恢复从『触发另一次重试』升级为『累积式学习』:失败尝试产出的验证器锚定证据应被跨回滚保存,并用来重塑后续探索方向。具体目标有三:(1) 维护一个持久的关系型探索状态,把前沿(frontier,即可检验的修复假设)、尝试、观察与检查点组织成图 $G_t=(V_t,R_t)$,使证据跨回滚存活;(2) 用隐式世界模型(IWM)评估修复假设、预估有界修复尝试的期望进展、推断假设间的同机制/支持等关系,指导假设的生成、细化与重开;(3) 将这些被反复调用的推理角色蒸馏进 Qwen3.5-9B 学生模型,在保持甚至提升性能的同时大幅降低教师 API 成本。

与已有工作不同的是,独特切入在于:不去修改底层执行器或重构搜索树,而是在其上方维护一个『探索信念状态』。每次尝试的新证据 $e_{t+1}$ 通过 $B_{t+1}=\mathrm{Update}(B_t,e_{t+1})$ 变成一元证据因子,去更新所有相关前沿的优先级 $b_i^t$——包括最新一轮并未执行的候选;同时把『是否有资格执行』(诊断证据门控)与『按什么顺序执行』(信念分数排序)解耦成两个独立机制。与 FLARE、Qwen-AgentWorld 等学习型世界模型不同,IWM 不学显式环境转移模型,而是直接从上下文做前瞻推理;与 HTR、Argus 等持久记忆/证据图方法不同,这里把验证器证据显式转化为可解释的优先级更新,而非只作为参考信息堆在上下文里。

核心方法

直觉上,BELIEFWEAVER 像一位严谨的工程师调试系统:维护一块『假设—证据板』,每个疑似故障机制是一条可检验假设,先做低成本探针实验收集证据,证据既决定哪条假设有资格动手修,也同步调整所有相关假设的可信度。技术上,作者把问题求解建模为对时变前沿集 $F_t=\{f_1,\dots,f_{N_t}\}$ 的探索:维护探索图 $G_t=(V_t,R_t)$ 与信念状态 $B_t=(G_t,E^t,b^t)$,其中每个前沿的信念 $b_i^t=q_i^t(z_i{=}1)$ 表示『该假设仍值得追』的后验概率;隐式世界模型 $\psi_\theta(B^t,f_i)=(\hat d_i^t,\hat p_i^t,\hat r_i^t)$ 负责生成假设、评估证据支持度、预估进展并推断关系;信念按因子乘积 $q_i^t(z_i)=\frac{1}{Z_i^t}\phi_i^0(z_i)\prod_{a\in A_i^t}\psi_a(z_i;E^t,G^t)$ 更新;执行排序用 $S_t(f_i)=b_i^t-\lambda_{fail}\ell_i^t$($\ell_i^t$ 标记最近一次尝试失败)。

核心创新有三点。其一,证据跨回滚持久化并沿关系边传播:矛盾性探针证据不仅压低当前假设的信念(矛盾因子 $\lambda_{ctr}=0.2$),还通过 same_mechanism、supports 等关系边(关系因子 $\lambda_{rel}=3.0$)降低共享机制的兄弟假设优先级,使一次尝试的教训惠及未被执行的相关候选,并能重开之前被搁置的方向。其二,执行资格与排序解耦:只有拿到 consistent 证据判定且目标检查尚未通过的前沿才有执行资格,信念分数只决定先后次序,避免『高信念但证据不足』的假设浪费宝贵的执行预算。其三,四类证据——重复尝试($\lambda_{att}=0.34$,由开发集重试统计标定)、细化关系、历史目标反馈($\lambda_{tgt}=0.5$)、探针反驳——全部设计为可跨回滚累积,使恢复真正变成累积式探索而非原地重试。

方法步骤详情

完整循环(Algorithm 1):(1) 初始化检查点 $c$ 与持久状态 $B=(G,E,b)$;(2) 假设生成:LLM 读入任务条款、行级源码、当前诊断与历史尝试,输出含缺陷机制、预测结果、待收集证据、目标检查与可选父假设的结构化前沿,校验后入候选目录;(3) 诊断探针:为每个候选生成诊断命令,在检查点临时副本中执行(超时 30 秒、输出截断 1500 字符),证据判定器对照假设声称的行为给出 consistent/insufficient/contradicted 三值判定;(4) 前瞻评估:local_code 专家追踪控制流数据流估计机制有效性,contract_judge 检查是否违反任务条款,按 $p_{prog}=m\alpha+(1-m)\beta$ 组合出严格进展概率,关系查询输出 same_mechanism/supports/requires/conflicts 及强度;(5) 用四类证据因子做局部归一化更新各前沿信念;(6) 选择:合格前沿按 $S_t$ 排序交给执行器做聚焦修复,无合格前沿、阻塞失败或停滞(2 次无改进)则退回普通延续;(7) 验证器评分:改进则保留检查点,否则回滚到 $c$,但证据与尝试历史始终写入图。最后把 4 个高频角色经 SFT(859 例)+ 3 轮 OPD 蒸馏进 Qwen3.5-9B,控制器与更新规则不变。

技术新颖性

技术新颖性体现在三点。第一,信念建模刻意保持轻量:所有证据都是一元 odds 因子、按前沿独立局部归一化,避免两两势带来的组合复杂度,使信念更新可精确计算且可解释——图 2 案例中能直接读出 $q=0.167$ 这样的信念值;代价是无法表达互斥、依赖等全局约束。第二,蒸馏的是『决策角色』而非整个智能体:控制器逻辑、信念更新规则、前瞻评估全部保持教师版不变,只把探针生成、证据判定、关系推断、条款评估四个结构化角色换成学生,用 LoRA SFT 加 3 轮在策略蒸馏(影子教师在学生自己诱导的状态上打标),使 SFT+OPD 学生反超匹配教师 4.7 个点,说明 OPD 不只是模仿教师输出,而是把 IWM 角色适配到了学生自身的探索分布。第三,把验证器反馈当作信念证据源而非仅仅当作奖励信号,让每一次失败都改变后续探索的优先级排序。

Overview of BELIEFWEAVER: 结合过去尝试的证据与对候选修复的前瞻推理。
Figure 1: Overview of BELIEFWEAVER: 结合过去尝试的证据与对候选修复的前瞻推理。

实验结果

以下逐一分析各项实验。主结果(表 1):在 113 个任务(TB2 65、TB3 23、LHTB 25)上,教师版 BELIEFWEAVER 的 Overall Final 为 $53.7\pm6.6\%$,比原生 Terminus-2 延续的 $48.6\pm6.7\%$ 高 5.1 个点,表 2 显示该配对增量 $+5.1\pm2.7$ 在统计上显著,Step-AUC 提高 2.8、Time-AUC 提高 1.6,说明系统不仅拿到更好的最终检查点,还在延续过程中更早取得有效进展。分基准看,增益集中在 TB3($46.6$ vs $36.5$,$+10.1$)与 TB2($71.2$ vs $66.9$,$+4.3$),LHTB 仅 $+2.5$。消融实验:结构化延续(检查点+回滚)单独贡献 $+2.9$;世界模型接地单独叠加反而比结构化延续低 1.7 个点($49.8$ vs $51.5$);再加上持久关系信念与证据门控才提升 $+3.9$,说明收益主要来自信念控制器而非探针本身。资源(表 3):执行步数 $-13.2\%$、执行时间 $-17.8\%$、执行器成本 $-16.6\%$,代价是控制器预留从 $\le9.1$ 涨到 33.4 次/任务。蒸馏(表 4):SFT+OPD 达 $58.4\%$ Overall Final,比原生基线高 9.8 个点、比匹配教师高 4.7 个点,LHTB 上 $23.4\%$ 远超教师 $14.8\%$;SFT+OPD+GEN 把教师请求从 32.9 降到 8.4 次/任务($-74.5\%$)、估计成本从 $0.550 降到 $0.100/任务($-81.9\%$),token 用量减少 81.1%。案例研究:图 2 中 TB3 cargo-flight-dispatch 从 70.37%(19/27 检查)修复到 88.89%(24/27);图 3 的 nginx 任务则 7 轮 702.4 秒无提升,展示方法的失败模式。

TB2、TB3 与 LHTB 上的任务性能与探索效率。
Table 1: TB2、TB3 与 LHTB 上的任务性能与探索效率。
相对基线的增益显著性分析。
Table 2: 相对基线的增益显著性分析。
教师延续的性能与资源占用。
Table 3: 教师延续的性能与资源占用。
学生蒸馏性能与教师 API 效率。
Table 4: 学生蒸馏性能与教师 API 效率。
蒸馏训练与数据采集设置。
Table 6: 蒸馏训练与数据采集设置。
查看结构化数据
任务指标本文基线提升
TB2 终端任务延续(65 任务) Final (%) 71.2±9.6(教师版) 66.9±9.9(Terminus-2 原生延续) +4.3 个百分点
TB3 终端任务延续(23 任务) Final (%) 46.6±13.7(教师版) 36.5±13.5(Terminus-2) +10.1 个百分点
LHTB 长程终端任务(25 任务) Final (%) 14.8±9.3(教师版)/ 23.4±10.5(SFT+OPD 学生) 12.3±8.9(Terminus-2) +2.5 个百分点(学生 +11.1)
Overall 汇总(113 任务) Final (%) / Step-AUC / Time-AUC 53.7±6.6 / 29.5±7.4 / 48.4±6.9(教师);学生 SFT+OPD 58.4±6.6 48.6±6.7 / 26.7±7.3 / 46.8±7.0(Terminus-2) 教师 +5.1pp(显著);学生 +9.8pp
执行器资源效率(Overall) 步数 / 时间 / 成本 19.1 步 / 397.9s / $3.389 每任务 22.0 步 / 483.8s / $4.062 每任务(Terminus-2) −13.2% / −17.8% / −16.6%
教师 API 效率(蒸馏) 估计成本($/任务) $0.100(SFT+OPD+GEN,8.4 次请求/任务) $0.550(匹配教师,32.9 次请求/任务) 成本 −81.9%,请求 −74.5%

局限与改进

作者承认的局限:IWM 预测或推断的关系出错时会错误排序前沿、浪费探索预算;信念状态只建模少量关系类型且全部是一元因子,多个故障原因交互时表达能力不足;实验局限于有验证器反馈和可靠检查点回滚的终端任务,在反馈稀疏或回滚带噪/不可用的场景是否成立尚不清楚。我的补充观察:(1) 世界模型接地单独使用时 Overall Final 反而比结构化延续低 1.7 个点,说明探针与目标检查本身带噪声与开销,收益依赖后续的信念控制器,且这两项组件是捆绑引入的,单项贡献无法归因(作者也承认);(2) 控制器开销巨大(33.4 次预留/任务 vs 其他配置 ≤9.1),且成本核算不含控制器与学生推理时间,交互式场景的延迟代价被低估;(3) 图 3 暴露局部执行成功与检查点性能之间的鸿沟——nginx 本地启动成功但探针再次连不上,信念机制无法弥合会话级/文件系统持久化这类跨探针不一致的故障;(4) 学生在 LHTB 上的反常优势(23.4 vs 教师 14.8)的 95% 置信区间(±10.5/±8.9)大量重叠,可能只是小样本方差,作者未做深入分析;(5) 案例研究只有两个,且作者自己注明『部分修复不能孤立归因于信念变化』。

独立分析的弱点

独立分析的弱点与改进方向:(1) 一元因子近似丢失全局一致性——requires/conflicts 关系只以一元方式注入,无法表达『A 与 B 互斥』这类约束,改进方向是引入小型两两势或用 MAP 推理/束搜索做联合前沿排序;(2) 证据判定依赖单次 LLM judge——探针因环境噪声(端口未就绪、超时、返回码非零)被判 contradicted 就直接以 $\lambda_{ctr}=0.2$ 压低信念,可能误杀正确假设,可引入重复探针、按返回码加权或校准置信度后再更新;(3) 因子参数手工固定($\lambda_{att}=0.34$ 来自开发集重试统计,其余在开发数据上调定),跨域迁移性差,可用贝叶斯在线估计或从轨迹数据学习更新规则,这也呼应作者提出的不确定性感知更新;(4) 强依赖可靠回滚——对有外部副作用的任务(真实服务器配置、数据库写入)不适用,可探索影子环境模拟或『软回滚』补偿机制;(5) 恢复能力受验证器质量上界约束——LHTB 教师只有 $14.8\%$,说明仅有分数反馈的超长任务依旧接近失败,可结合自动生成的中间子目标检查或密集中间奖励来增强证据来源。

未来方向

作者明确提出的方向包括:学习式与不确定性感知的信念更新(替代手工 $\lambda$ 参数)、在控制器各组件间做更细粒度的消融归因、以及在稀疏反馈或不完美回滚下的恢复。基于本文成果还可以延伸:(1) 把关系推断与信念更新升级为可学习模块(如图神经网络或小型因子图评分器),用最终任务奖励端到端训练,摆脱手工标定;(2) 将 OPD 扩展到前瞻评估与控制器策略本身——目前前瞻评估仍是教师职责,GEN 实验已证明把更多角色移给学生能带来最大成本节省;(3) 让探索图跨任务持久化,形成可复用的故障机制记忆库,与技能库方法(SkillRL、COS-PLAY)互补;(4) 将框架移植到 GUI 操作、科研实验等非终端域,检验不可逆动作下的安全恢复机制;(5) 从理论上分析一元因子近似的排序质量界限,给出局部归一化何时足以近似联合后验的条件;(6) 学生反超教师的现象值得专门研究——OPD 是否在纠正教师的系统性偏差、学生更小的假设空间是否反而起正则化作用。

复现评估

复现信息相当充分:正文与附录给出全部控制器超参(表 5:$\lambda_{att}=0.34$、$\lambda_{rel}=3.0$、$\lambda_{tgt}=0.5$、$\lambda_{ctr}=0.2$、$\lambda_{fail}=0.5$、初始信念 0.5、探针超时 30s、停滞触发 2 次无改进等)与蒸馏配置(表 6:LoRA rank 16、SFT lr $2\times10^{-5}$ 1 epoch、859 个 SFT 样本按任务划分 723/136、3 轮 OPD 每轮 28 任务(12 TB2/10 TB3/6 LHTB)、GEN 适配器 609/150 样本、最大序列 65,536 token),评估协议(预算 TB2 ≤1800s、TB3 1200s、LHTB 720s;token 单价 $1/$4 每百万;200,000 次 bootstrap)与前瞻评估公式 (10)–(15) 也完整公开。但正文未提及开源代码、提示词全文或轨迹数据的发布链接;教师为 DeepSeek V4 Pro API、学生为 Qwen3.5-9B,复现需支付教师 API 费用(约 $0.55/任务全量)并实现探针/回滚/图维护的编排工程。核心难点在控制器与提示词工程而非模型训练本身,总体复现难度中等偏高。

相似论文推荐