mlsanityData to 14 Oct 2026

Textual Belief States for World Models: Identifiable Representation Learning Under Strict Mediation

arXiv 2606.27681 ↗First posted 26 Jun 2026cs.LG cs.CLcited by 1

On the map: World models → World models for language agents

Abstract (arXiv)

World models in partially observed environments rely on latent representations that summarize interaction history, but in many modern LLM-based architectures predictive performance fails to reflect representation quality due to history bypass, rendering the latent state unidentifiable. Strict latent state mediation, requiring predictions to depend only on the latent state and action, is a classical principle that resolves this, but enforcing it in text-based settings is an open challenge: textual latent states are discrete and non-differentiable, precluding variational training, and expressive LLM decoders readily ignore the bottleneck. We show how to make strict mediation work in the text domain. We formalize why it is necessary, showing that strict mediation makes representation quality empirically testable while history-leaky architectures break this connection. We then introduce textual latent states, which are discrete, interpretable, and variable-length, and factorized GRPO (fGRPO), a tree-structured reinforcement learning method that enforces strict mediation during training. Experiments on TextWorld and ScienceWorld show preserved one-step prediction accuracy alongside up to 57% gains in representation quality and 98% improvements in rollout performance, increasing with task complexity and horizon.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Placed in World models for language agents, one of 11 directions in World models. Builds on 13 of its 23 references that are in the map. First posted on arXiv 26 Jun 2026.

Builds on

13 of 23 references.

Closest ancestors · 5

Same direction, newest first

  1. 2025From Word to World: Can Large Language Models be Implicit Text-based World Models?21 Dec
  2. 2025PAN: A World Model for General, Actionable, and Long-Horizon World Simulation12 Nov
  3. 2024LLM-Based World Models Can Make Decisions Solely, But Rigorous Evaluations are Needed13 Nov
  4. 2024WorldGPT: Empowering LLM as Multimodal World Model28 Apr
  5. 2023Reasoning with Language Model is Planning with World Model24 May

Other references · 8

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  2. 2019Dream to Control: Learning Behaviors by Latent Imaginationcited by 1,653
  3. 2018World Modelscited by 1,513
  4. 2023Mastering Diverse Domains through World Modelscited by 1,253
  5. 2020Mastering Atari with Discrete World Modelscited by 1,007
  6. 2015Generating Sentences from a Continuous Spacecited by 612
  7. 2025Training Agents Inside of Scalable World Modelscited by 136
  8. 2019Preventing Posterior Collapse with delta-VAEscited by 86

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.

Similar, not cited

Close in wording, no citation either way

  1. 2026Agent-BRACE: Decoupling Beliefs from Actions in Long-Horizon Tasks via Verbalized State Uncertainty
  2. 2026Why We Need World Models for AGI: Where LLMs Fail and How World Models May Outperform
  3. 2026Shaping Persistent Representations from Independent Interactions
  4. 2026Masked Diffusion Language Models are Strong and Steerable Text-Based World Models for Agentic RL
  5. 2026Latent State Design for World Models under Sufficiency Constraints
  6. 2026HyperWorld: Hypergraph-Structured State Serialization Improves Learned Textual World Models
  7. 2026Reinforcement World Model Learning for LLM-based Agents
  8. 2026Why Struggle with Continuous Latents? Interpretable Discrete Latent Reasoning via Rendered Compression