English

Co-Evolving Latent Action World Models

Machine Learning 2026-04-07 v2

Abstract

Adapting pretrained video generation models into controllable world models via latent actions is a promising step towards creating generalist world models. The dominant paradigm adopts a two-stage approach that trains latent action model (LAM) and the world model separately, resulting in redundant training and limiting their potential for co-adaptation. A conceptually simple and appealing idea is to directly replace the forward dynamic model in LAM with a powerful world model and training them jointly, but it is non-trivial and prone to representational collapse. In this work, we propose CoLA-World, which for the first time successfully realizes this synergistic paradigm, resolving the core challenge in joint learning through a critical warm-up phase that effectively aligns the representations of the from-scratch LAM with the pretrained world model. This unlocks a co-evolution cycle: the world model acts as a knowledgeable tutor, providing gradients to shape a high-quality LAM, while the LAM offers a more precise and adaptable control interface to the world model. Empirically, CoLA-World matches or outperforms prior two-stage methods in both video simulation quality and downstream visual planning, establishing a robust and efficient new paradigm for the field.

Keywords

Cite

@article{arxiv.2510.26433,
  title  = {Co-Evolving Latent Action World Models},
  author = {Yucen Wang and Fengming Zhang and De-Chuan Zhan and Li Zhao and Kaixin Wang and Jiang Bian},
  journal= {arXiv preprint arXiv:2510.26433},
  year   = {2026}
}
R2 v1 2026-07-01T07:13:43.850Z