Belief Memory: Agent Memory Under Partial Observability

AuthorsJunfeng Liao, Qizhou Wang, Jianing Zhu et al.

arXiv 20262026

TL;DR

BeliefMem uses probabilistic candidate memories with noisy OR updates to achieve 42.38 F1 on LoCoMo vs 40.99 for Mem0 (+1.39).

SharePost on XLinkedIn

Read our summary here, or open the publisher PDF on the next tab.

THE PROBLEM

Self-reinforcing error from deterministic memory under partial observability

Existing agents store each observation as a single deterministic conclusion, permanently discarding uncertainty and alternative hypotheses about hidden state.

When memory records only “API X failed,” the agent avoids testing API X, reinforcing incorrect beliefs and conflicting with user instructions like “Use API X to ...”.

HOW IT WORKS

BeliefMem: Attribute-level belief memory with noisy OR updates

BeliefMem introduces an external Belief Memory Bank, with Add, Merge, and Belief-aware Retrieval to maintain attribute-level candidate hypotheses and probabilities.

Think of BeliefMem like a card catalog of beliefs: instead of one card per fact, it keeps multiple cards per attribute, each with a confidence score that updates over time.

This probabilistic representation lets BeliefMem preserve uncertainty beyond a plain context window, enabling agents to revisit alternatives and correct earlier errors as new evidence arrives.

DIAGRAM

Belief-aware update and retrieval flow in BeliefMem

This diagram shows how BeliefMem updates candidate probabilities with Add and Merge, then retrieves top K attributes with time decay for action selection.

DIAGRAM

Evaluation pipeline for BeliefMem on LoCoMo and ALFWorld

This diagram shows how BeliefMem is plugged into LLM agents and evaluated on LoCoMo and ALFWorld with baselines.

PROCESS

How BeliefMem Handles a Session in a Partially Observable Environment

  1. 01

    Belief-based Memory Formulation

    BeliefMem first represents each attribute c with a belief over hypotheses H(c), approximating b_t(c) using the Belief Memory Bank instead of point estimates.

  2. 02

    Belief Update in Memory

    BeliefMem applies Add to initialize candidates for new attributes and Merge with noisy OR to update p_t+1(h) based on evidence strength Δ(o_t+1, h).

  3. 03

    Belief-aware Retrieval

    BeliefMem scores attributes with α_t(c) = sim(o_t, c) · λ^{τ_t(c)}, selects top K, and returns full candidate distributions p_t(h) for decision making.

  4. 04

    Action Selection under Partial Observability

    The agent policy π chooses action a_t conditioned on current observation o_t and retrieved beliefs r_t, keeping alternative hypotheses visible to avoid self-reinforcing error.

KEY CONTRIBUTIONS

Key Contributions

  • 01

    Belief-based Memory Representation

    BeliefMem replaces deterministic entries with attribute-level belief states in the Belief Memory Bank, storing multiple candidate conclusions per attribute with probabilities p_t(h).

  • 02

    Noisy OR Evidence Merge

    BeliefMem introduces a noisy OR Merge update p_t+1(h) = min(1 − (1 − p_t(h))(1 − Δ(o_t+1, h)), 0.99), enabling robust belief refinement under noisy observations.

  • 03

    Empirical Gains on LoCoMo and ALFWorld

    BeliefMem achieves 42.38 F1 on LoCoMo with GPT-4o-mini and 59.88% SR on ALFWorld, exceeding Mem0 by 1.39 F1 and ReadAgent by 5.85 SR.

RESULTS

By the Numbers

LoCoMo F1 Avg GPT4o mini

42.38 F1

+1.39 over Mem0 (40.99 F1)

LoCoMo BLEU1 Avg GPT4o mini

36.30 BLEU-1

+4.31 over Mem0 (31.99 BLEU-1)

ALFWorld Avg SR

59.88%

+5.85 over ReadAgent (54.03% SR)

ALFWorld Avg Steps

29.56 steps

- -3.99 vs ReadAgent (27.65 steps, more steps for higher SR)

BeliefMem is evaluated on LoCoMo for long-term conversational memory and ALFWorld for embodied interaction. These results show BeliefMem’s probabilistic memory improves multi-hop reasoning and success rate under partial observability compared to deterministic baselines.

BENCHMARK

By the Numbers

BeliefMem is evaluated on LoCoMo for long-term conversational memory and ALFWorld for embodied interaction. These results show BeliefMem’s probabilistic memory improves multi-hop reasoning and success rate under partial observability compared to deterministic baselines.

BENCHMARK

LoCoMo GPT-4o-mini Average F1 Comparison

Average F1 on LoCoMo across all question categories using GPT-4o-mini.

BENCHMARK

ALFWorld Average Success Rate Comparison

Average success rate on ALFWorld seen and unseen splits with Qwen3-Next-80B-A3B-Instruct.

KEY INSIGHT

The Counterintuitive Finding

BeliefMem using only 50% of the ALFWorld memory corpus (BeliefMem*) still achieves 59.88% average SR, beating all baselines including ReadAgent at 54.03%.

This is counterintuitive because more memory data usually helps, yet BeliefMem’s probabilistic representation avoids overfitting seen trajectories and improves out-of-distribution generalization with less data.

WHY IT MATTERS

What this unlocks for the field

BeliefMem unlocks agents that can act under partial observability while keeping multiple hypotheses alive, updating beliefs instead of locking into brittle deterministic conclusions.

Builders can now design long-lived LLM agents whose memory corrects itself over time, supports temporal reasoning, and resists self-reinforcing errors in noisy, real-world environments.

~12 min read← Back to papers

Related papers

BenchmarkAgent Memory

Active Context Compression: Autonomous Memory Management in LLM Agents

Nikhil Verma

· 2026

Focus Agent adds start_focus, complete_focus, a persistent Knowledge block, and an optimized Persistent Bash plus String-Replace Editor scaffold to actively compress context during long software-engineering tasks. On five hard SWE-bench Lite instances against a Baseline ReAct agent, Focus Agent achieves 22.7% token reduction (14.9M → 11.5M) while matching 3/5 = 60% task success.

Agent Memory

ActMem: Bridging the Gap Between Memory Retrieval and Reasoning in LLM Agents

Xiaohui Zhang, Zequn Sun et al.

· 2026

ActMem transforms dialogue history into atomic facts via Memory Fact Extraction, groups them with Fact Clustering, links them through a Memory KG Construction module, and uses Counterfactual-based Retrieval and Reasoning for action-aware answers. On ActMemEval, ActMem reaches 76.52% QA accuracy with DeepSeek-V3, beating LightMem’s 63.97% by 12.55 points and NaiveRAG’s 61.54%.

Questions about this paper?

Paper: Belief Memory: Agent Memory Under Partial Observability

Answers use this explainer on Memory Papers.

Checking…