How do World Models and Policies Compose in LLM Agents? A Joint Spectral and Behavioral Account
On the map: World models → World models for language agents
also in 1 more
Abstract (arXiv)
How do LLM agents come to both understand environments they act in and master tasks set within them? Through controlled experiments combining world-model training (next-state prediction) and policy training (reward maximization), we investigate this question. We dissect the resulting models through their additive parameter updates. Geometrically, we find effective world-model updates are low-rank and share an input-feature subspace with policy updates while writing to nearly orthogonal output directions, whether trained separately or sequentially. However, we find that, in projection interventions, the sequential update induces more robustness than separate policy RL when removing the world model's leading input directions, suggesting that it has learned alternative input pathways. Behaviorally, we find the sequentially trained agent explores a wider range of states and actions. Based on this, we ask: does policy training preserve world knowledge as well as it could? We probe this with training-free merging built on the geometrically motivated input basis plus an online world-model loss during policy RL, and show both improve over the untreated baseline. Our findings suggest world knowledge and task-directed ability can be learned in geometrically complementary forms, and that future post-training pipelines should consider how best to engineer the interface between them.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in World models for language agents, one of 11 directions in World models. Builds on 13 of its 46 references that are in the map. First posted on arXiv 30 Aug 2026.
Builds on
13 of 46 references.
Closest ancestors · 4
Same direction, newest first
- 2026ECHO: Terminal Agents Learn World Models for Free23 May
- 2026Reinforcement World Model Learning for LLM-based Agents5 Feb
- 2025Why Do LLM Agents Fail in Exploring New Environments? A World-Modeling Perspective16 Oct
- 2025Dyna-Think: Synergizing Reasoning, Acting, and World Model Simulation in AI Agents31 May
Other references · 9
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2021Evaluating Large Language Models Trained on Codecited by 8,320
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
- 2025Qwen3 Embedding: Advancing Text Embedding and Reranking Through Foundation Modelscited by 1,261
- 2023Mastering Diverse Domains through World Modelscited by 1,253
- 2025Dyna-Mind: Learning to Simulate from Experience for Better AI Agentscited by 8
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026Towards a Belief-Based World Model for LLM Agents
- 2026When Does Multi-Agent RL Improve LLM Workflows? Workflow, Scale, and Policy-Sharing Tradeoffs
- 2026Beyond Prediction: Steering VLM Agents with Retrospective World Modeling
- 2026Beyond Next-Observation Prediction: Agent-Authored World Modeling for Sequential Decision Making
- 2026World Models as an Intermediary between Agents and the Real World
- 2026EVOKE: Eliciting World Knowledge in Agents for Transferable Decision-Making
- 2025WALL-E 2.0: World Alignment by NeuroSymbolic Learning improves World Model-based LLM Agents
- 2025Language-conditioned world model improves policy generalization by reading environmental descriptions