LaLa-World: Scaling Latent World Models with Natural Language Actions
Abstract
LaLa-World is a language-conditioned latent world model that jointly learns from diverse simulation, robot, and video data to enable efficient model-predictive planning and policy initialization.
This paper shows that JEPA-style latent world models (LWMs) can learn across diverse domains and data sources by adopting the language-conditioned interface of video generation models. We establish this approach as an effective and efficient basis for planning with imagination and downstream policy fine-tuning. To this end, we introduce LaLa-World, a Language-conditioned Latent World model that learns from simulation and robot trajectories together with broader image and video data. Two designs make this work. First, we describe both numerical actions and visual changes in language, enabling joint training of a single future predictor across heterogeneous datasets. Second, we build the predictor on a language model, allowing it to interpret language conditions while predicting future latent states. The resulting predictor achieves higher average MPC planning performance than existing LWMs trained separately for each domain or task. Adding image and video data further improves average planning performance and robustness to action phrasing. In a matched comparison with an RAEv2-based video generation model baseline, LaLa-World reaches comparable planning performance with 6times less training cost and 13.9times faster planning. Beyond planning, the pretrained LaLa-World serves as an initialization for LaLa-Policy, which jointly predicts actions and future visual representations. LaLa-Policy performs competitively with Cosmos Policy on LIBERO and RoboCasa, with up to 5.5times faster inference.
Community
TL;DR
A Language-conditioned Latent World model (LaLa-World) uses natural language as a shared action interface to learn from simulation, robotics, images, and videos. It predicts future visual representations for efficient planning across domains and can be fine-tuned into LaLa-Policy, which jointly predicts robot actions and future representations.
Key results:
- One LaLa-World predictor trained across domains beats per-domain LWMs on average MPC planning performance
- Adding image/video data further improves planning and robustness to action phrasing
- vs. an RAEv2-based video generation baseline: comparable planning with 6× less training cost and 13.9× faster planning
- LaLa-Policy, initialized from LaLa-World, is competitive with Cosmos Policy on LIBERO and RoboCasa with up to 5.5× faster inference
Project page: https://june-page.github.io/lala-world/
Code: https://github.com/junha1125/LaLa-World
alphaXiv: https://www.alphaxiv.org/abs/2610.27851
arXiv: https://arxiv.org/abs/2610.27851
Presentation: https://june-page.github.io/lala-world-slides/
Interesting approach to combining language and visual data to improve AI planning. If you need a little break after reading research papers, Space Waves is a fun browser game to try. The fast-paced levels and tricky obstacles make it a good challenge for your reflexes: https://spacewaves2.io/
Get this paper in your agent:
hf papers read 2610.27851 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper