arXiv AI

When Should Models Change Their Minds? Contextual Belief Management in Large Language Models

arXiv AI
Oct 2

Beyond Memory: Harnessing Long-Horizon Agents with Explicit Belief States

Beyond Memory: Harnessing Long-Horizon Agents with Explicit Belief States introduces PoS, an inference-time framework that builds and maintains explicit belief states to guide large language model agents. Each belief state combines an estimate of the current world with unresolved task requirements, making clear what the agent still needs to learn and accomplish. PoS validates consistency, monitors task progress to detect Belief Trapping, and tailors recovery to the trapping pattern and unresolved requirements, achieving top performance across four benchmarks with all three LLM backbones.

By Yu Luo, Jiamin Jiang, Yimin Zuo, Xidao Wen, Rongchen Gao, Yongqian Sun, Shenglin Zhang, Guiyang Liu, Cheng Zhang, Fang Situ, Qi Zhou, Dan Pei
arXiv Computation and Language
Oct 21

LAMB: Latent Multi-Hypothesis Beliefs for Long-Horizon Large Language Agents

The paper introduces LAMB, a recurrent framework that stores latent multi‑hypothesis beliefs for long‑horizon language agents. LAMB separates a shared Common state containing interaction evidence from weighted Delta components representing competing hypotheses, and updates these hypotheses using relative evidence from action‑outcome predictions without generating textual memory. Experiments on ALFWorld, BabyAI, ScienceWorld, and Crafter show that LAMB matches or surpasses prior methods, notably improving success rates and reducing redundant actions.

By Peien Ding, Hui Li, Yuhan Liu, Jinsong Su
Hugging Face Trending Papers
Jul 7

Doomed from the Start: Early Abort of LLM Agent Episodes via a Recall-Controlled Probe Cascade

Large language model (LLM) agents solving multi-step tasks frequently commit to trajectories that are doomed to fail, yet continue to consume substantial inference compute before the failure becomes observable. We show that failure is predictable early from the agent's internal representations: lightweight per-round probes on hidden activations anticipate eventual episode failure as early as the first interaction round, where scorers reading only the agent's observable behavior are barely better than chance.

arXiv AI
Sep 11

Belief-State Engine: Augmenting LLMs for Principled Planning Under Partial Observability

The paper introduces the Belief-State Engine (BSE), an inference module that supplies a large language model (LLM) with a Bayesian posterior over hidden states in a partially observable Markov decision process (POMDP). By keeping the raw action‑observation log hidden from the LLM, the BSE ensures the agent behaves as a sound Markov policy on the belief MDP, thereby inheriting classical POMDP optimality guarantees. Experiments on the Tiger POMDP and a red‑team attack‑graph task show that BSE‑augmented agents outperform six baselines in task return, belief calibration, and decision consistency.

By Arnab Chattopadhayay, Debdipta Halder
Hugging Face Trending Papers
Aug 18

Whether LLMs Can Navigate Beliefs and Facts Depends on How You Phrase It

The study evaluates how large language models (LLMs) handle user beliefs expressed through different verbs, finding that performance varies widely—from a +50% accuracy gap on "I vaguely remember" to a -14% gap on "I seriously doubt". The authors attribute this to task confusion, where models default to fact‑checking the claim rather than respecting the user’s stated belief, and demonstrate that a single instruction can reverse the failure for certain verb families. Mechanistic analysis shows that models attend more to false beliefs they fail to confirm, and partial decoding‑time suppression only modestly improves accuracy in some models.

arXiv Machine Learning
Oct 19

Process Reward Modeling with Reliability- and Conservatism-Aware Bayesian Belief

Process Reward Models (PRMs) are crucial for scaling large language model reasoning at test time, but they suffer from two types of uncertainty: PRM uncertainty from limited supervision and imperfect training, and policy uncertainty from mismatched data collection policies. The paper introduces BayesianPRM, which treats process rewards as random variables, discretizes the reward space with an ensemble, and applies two belief updates—reliability‑aware updating for PRM uncertainty and conservatism‑aware calibration for policy uncertainty—to form a Bayesian belief whose expectation yields a more accurate reward estimate. Experiments on multimodal reasoning benchmarks show that BayesianPRM improves reward accuracy and test‑time scaling performance.

By Zhongjian Qiao, Zhixiang Zhou, Jiafei Lyu, Peisong Wang, Qiaosheng Zhang, Siyang Gao, Shuang Qiu
arXiv AI
Jul 7

ASK in the Dark: Uncertainty-Gated LLM Assistance under Partial Observability

arXiv:2607. 02686v1 Announce Type: new Abstract: Reinforcement learning agents operating under partial observability must act on incomplete information, making them natural candidates for guidance from small language models (SLMs) that carry broad reasoning priors.

By Juarez Monteiro, Nathan Gavenski, Guilherme Lima, Francisco Galuppo, Odinaldo Rodrigues, Adriano Veloso