Search papers, labs, and topics across Lattice.
This paper explores the concept of Causal World Models (CWMs) by examining the necessary components that allow intelligent agents to predict and act within their environments. It emphasizes that effective WMs must not only generate data but also encapsulate the properties and interactions of entities within a system, thereby enhancing their predictive capabilities. The authors provide a formal definition of CWMs and connect them to existing frameworks in causal representation learning, offering insights into their identifiability and applicability in decision-making processes.
Causal World Models redefine how we understand agent-environment interactions, revealing that generative capabilities alone are insufficient for effective decision-making.
World Models (WM) are increasingly seen as a foundation for intelligent agents that can predict, plan, and act beyond their training distribution. In this paper, we study WMs from a causal perspective across multiple levels of abstraction, ranging from perceptual observations to building a conceptual representation of the structure governing the environment dynamics. We argue that useful WMs must go beyond generative capabilities alone: they should also capture entity properties, entity-to-entity interactions, and entity-to-environment interactions that determine and explain the dynamics of a system. We provide a formal definition of Causal WMs (CWMs) grounded in the tasks they are intended to support, connecting world modelling with existing work in causal representation learning, object-centric learning, causal discovery, structural causal models, and model-based decision-making. Finally, we relate CWMs to the literature on identifiability, clarifying when the components of a WM can be recovered from data and up to which equivalence. With this, we ground WMs in representations and structures that support causal reasoning and informed decision-making.