A Comprehensive Survey on World Models for Embodied AI
Xinqing Li, Xin He, Le Zhang, Min Wu, Xiaoli Li, Yun Liu
TL;DR
This survey formalizes world models for embodied AI using a three-axis taxonomy to unify functionality, temporal modeling, and spatial representation. It surveys data resources, metrics, and quantitative performance across robotics, autonomous driving, and general video domains, highlighting RSSM/Dreamer style sequential models and global-difference approaches like transformer-based and diffusion-based methods. Key insights include long-horizon error accumulation as a core challenge, the trade-off between pixel fidelity and physical consistency, and the fragmentation of datasets and evaluation standards limiting cross-domain generalization. The work advocates for unified, physically grounded benchmarks and hybrid architectures that blend autoregressive and global prediction strengths to advance safe, real-time, and transferable embodied agents.
Abstract
Embodied AI requires agents that perceive, act, and anticipate how actions reshape future world states. World models serve as internal simulators that capture environment dynamics, enabling forward and counterfactual rollouts to support perception, prediction, and decision making. This survey presents a unified framework for world models in embodied AI. Specifically, we formalize the problem setting and learning objectives, and propose a three-axis taxonomy encompassing: (1) Functionality, Decision-Coupled vs. General-Purpose; (2) Temporal Modeling, Sequential Simulation and Inference vs. Global Difference Prediction; (3) Spatial Representation, Global Latent Vector, Token Feature Sequence, Spatial Latent Grid, and Decomposed Rendering Representation. We systematize data resources and metrics across robotics, autonomous driving, and general video settings, covering pixel prediction quality, state-level understanding, and task performance. Furthermore, we offer a quantitative comparison of state-of-the-art models and distill key open challenges, including the scarcity of unified datasets and the need for evaluation metrics that assess physical consistency over pixel fidelity, the trade-off between model performance and the computational efficiency required for real-time control, and the core modeling difficulty of achieving long-horizon temporal consistency while mitigating error accumulation. Finally, we maintain a curated bibliography at https://github.com/Li-Zn-H/AwesomeWorldModels.
