arXiv:2605.07042cs.AIcs.LG2026-05被引 3

用强化学习框架优化大模型搜索,减少重复和过早停止。

The Context Gathering Decision Process: A POMDP Framework for Agentic Search

  • 将搜索过程建模为部分可观测决策过程,动态更新信念状态。
  • 提升多跳推理准确率最高达11.4%,节省39%搜索令牌。
  • 适合构建稳定、高效的大模型搜索代理,尤其在复杂环境里。

大型语言模型(LLM)代理在复杂环境(如大规模代码库、企业数据库和对话历史)中运行时,相关状态远超其上下文窗口。为导航这些空间,代理需迭代探索以获取相关信息。然而,缺乏显式机制会导致工作记忆退化为搜索状态的有损表示,引发重复循环和过早终止。本文提出上下文收集决策过程(CGDP),一种专门化的部分可观测马尔可夫决策过程,目标是自适应地精炼信念状态以定位任务所需信息。我们将LLM行为建模为该CGDP中的近似Thompson采样,并引入基于谓词的方法,将LLM隐式的搜索分解为显式且模块化的操作。进一步提出两种即插即用的干预策略:持久的谓词驱动信念状态,在控制上下文的同时保持多跳推理能力;以及程序化的耗尽检测门,可在不提前终止的前提下阻止无效搜索。在四种方法和三个问答领域中,实验验证:用CGDP启发的信念状态替代隐式状态,可使多跳推理性能提升最高达11.4%;而模块化的耗尽检测可节省高达39%的令牌,且不影响代理表现。最终,我们主张将LLM代理循环视为CGDP,能指导模块化、非干扰性改进,提升智能体搜索效能。

原文摘要 · Abstract (English)

Large Language Model (LLM) agents are deployed in complex environments -- such as massive codebases, enterprise databases, and conversational histories -- where the relevant state far exceeds their context windows. To navigate these spaces, an agent must iteratively explore the environment to find relevant information. However, without explicit infrastructure, an agent's working memory can degrade into lossy representations of the search state, resulting in redundant work (e.g. repetitive looping) and premature stopping. In this work, we formalize this challenge as the Context Gathering Decision Process (CGDP), a specialized Partially Observable Markov Decision Process, where an agent's objective is to adaptively refine its belief state to isolate the necessary information for a task. We model an LLM's behavior as approximate Thompson Sampling within this CGDP, and introduce a predicate-based method that decomposes an LLM's implicit search into explicit and modular operations. We then derive two plug-and-play interventions for iterative LLM agents: a persistent, predicate-based belief state that bounds context while preserving multi-hop reasoning, and a programmatic exhaustion gate that halts unproductive search without premature stopping. Across four methods and three question-answering domains, we empirically validate that replacing an LLM's implicit state with our CGDP-motivated belief state improves multi-hop reasoning by up to $11.4\%$; while the modular programmatic exhaustion detection saves up to $39\%$ of tokens without any degradation in agent performance. Ultimately, we argue that framing the LLM agent loop as a CGDP can guide the design of modular, non-interfering improvements to agentic search harnesses.

大模型代理搜索优化强化学习多跳推理

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。