让大模型在长任务中分离信念与行动,提升决策准确性。
Agent-BRACE: Decoupling Beliefs from Actions in Long-Horizon Tasks via Verbalized State Uncertainty

- 将信念状态抽象为自然语言陈述加口语化可信度标签
- 在长程任务中实现平均14.5%的性能提升,上下文长度恒定
- 适合需要持续推理与不确定处理的复杂智能体系统
大语言模型在部分可观测环境中的长程任务中面临双重挑战:需对未观测世界属性保持不确定性,且长期交互导致上下文无限膨胀,稀释关键信息。理想解法是信念状态——基于历史观测与动作的环境状态后验分布,可压缩历史用于决策,不随任务时长增长。然而,文本的开放性使如何表示该分布不明确。为此,我们提出Agent-BRACE:通过强化学习联合优化信念模型与策略模型。信念模型生成结构化近似信念分布:一组关于环境的原子性自然语言陈述,每条附带从‘确定’到‘未知’的有序口语化可信度标签。策略模型基于此紧凑、结构化的信念进行决策,显式处理不确定性。在多个长程、部分可观测的具身语言环境中,Agent-BRACE实现平均绝对提升+14.5%(Qwen2.5-3B-Instruct)和+5.3%(Qwen3-4B-Instruct),优于强基线,同时保持近恒定上下文窗口。进一步分析显示,随着证据积累,学习到的信念在任务进程中逐渐校准。
原文摘要 · Abstract (English)
Large language models (LLMs) are increasingly deployed on long-horizon tasks in partially observable environments, where they must act while inferring and tracking a complex environment state over many steps. This leads to two challenges: partial observability requires maintaining uncertainty over unobserved world attributes, and long interaction history causes context to grow without bound, diluting task-relevant information. A principled solution to both challenges is a belief state: a posterior distribution over environment states given past observations and actions, which compactly encodes history for decision making regardless of episode length. In LLM agents, however, the open-ended nature of text makes it unclear how to represent such a distribution. Therefore, we introduce Agent-BRACE: Agent Belief state Representation via Abstraction and Confidence Estimation, a method that decouples an LLM agent into a belief state model and a policy model, jointly optimized via reinforcement learning. The belief state model produces a structured approximation of the belief distribution: a set of atomic natural language claims about the environment, each annotated with an ordinal verbalized certainty label ranging from certain to unknown. The policy model conditions on this compact, structured approximate belief rather than the full history, learning to select actions under explicit uncertainty. Across long-horizon, partially observable embodied language environments, Agent-BRACE achieves an average absolute improvement of +14.5% (Qwen2.5-3B-Instruct) and +5.3% (Qwen3-4B-Instruct), outperforming strong RL baselines while maintaining a near-constant context window independent of episode length. Further analysis shows that the learned belief becomes increasingly calibrated over the course of an episode as evidence accumulates.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。