让大模型智能体更精准地评估每步操作的价值,提升长任务决策能力。
Gated-BEPO: Confidence-Gated Bellman Credit Assignment for Large Language Model Agents

- 基于经验回放图构建节点价值估计,实现更真实的逐步奖励分配。
- 在多个任务上显著提升模型表现,尤其在复杂长程任务中优势明显。
- 动态融合多级奖励信号,适合需要精细动作评估的智能体研究者。
在长时序环境中训练大语言模型智能体,需将稀疏的最终结果奖励合理分配给各步骤动作。现有无评论者方法将轨迹级奖励均匀传播至各步,而近期方法通过匹配重复状态构造步骤级分组,并在组内比较动作。前者无法区分失败轨迹中有效动作与成功轨迹中无效动作;后者依赖直接从单条轨迹结果获取的步骤级奖励,且以固定权重融合段落级信用。本文提出Gated-BEPO,从经验回放图中推导步骤级信用。对每个回放组,构建经验图并利用均值回传贝尔曼不动点估计节点值,反映当前策略的经验动作分布。随后通过广义优势估计沿采样轨迹累积这些时序差分残差,获得包含即时与后续影响的步骤级贝尔曼优势。为自适应融合段落级与步骤级信用,引入置信度门控机制:仅在具有多个可观测后继状态的节点使用贝尔曼信用,否则采用段落级信用。在WebShop、ALFWorld及视觉Sokoban任务上的实验表明,该方法在语言与视觉-语言模型上均取得一致改进;诊断性消融分析支持贝尔曼不动点价值估计的有效性,并表明应选择性而非统一地将步骤级信用融入最终优势量。
原文摘要 · Abstract (English)
Training large language model agents in long-horizon environments requires assigning credit from sparse terminal outcomes to individual actions. Existing critic-free methods propagate trajectory-level rewards uniformly across steps, while recent approaches construct step-level groups by matching repeated states and compare actions within each group. The former cannot distinguish useful actions in failed trajectories from ineffective actions in successful ones. The latter rely on step credit derived directly from individual trajectory outcomes and fixed-weight fusion with episode-level credit. We propose Gated-BEPO, which derives step-level credit from empirical rollout graphs. For each rollout group, Gated-BEPO constructs an empirical graph and estimates node values through a mean-backup Bellman fixed point that reflects the empirical action distribution of the current policy. We then accumulate these temporal-difference residuals along each sampled trajectory using generalized advantage estimation, yielding step-level Bellman advantages that capture both immediate and downstream effects. To adaptively fuse episode- and step-level credit, a confidence gate incorporates Bellman credit only at states with multiple observed successors and otherwise uses episode-level credit. Experiments on WebShop, ALFWorld, and visual Sokoban show consistent improvements across language and vision-language models, while diagnostic ablations support the effectiveness of Bellman fixed-point value estimation and show that step-level credit should be incorporated selectively rather than uniformly into the final advantage.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。