Learning Implicit Causal World Models from Multi-Agent Demonstrations

[Submitted on 28 Jul 2026]

View PDF

Abstract:In model-based reinforcement learning, world models exist as internal simulators, but their training often conflates statistical correlations with causal mechanisms. This problem is exacerbated in multi-agent systems where physical transitions are intertwined with strategic agent intents, causing world models to fail under distribution shift. We introduce Implicit Causal World Models to recover environmental dynamics from offline demonstrations without requiring pre-defined causal graphs. By incorporating policy variance, we render world models discoverable via the sequential backdoor condition. Evaluations across coordination tasks (Two-Door, Navigation, and Giveway) demonstrate that these models provide interpretable causal representations under both full and partial observability, with model accuracy scaling directly with interventional strength.

Submission history

From: Jasorsi Ghosh [view email]
[v1] Tue, 28 Jul 2026 23:18:38 UTC (25,902 KB)