English

DiLA: Disentangled Latent Action World Models

Computer Vision and Pattern Recognition 2026-05-18 v1 Artificial Intelligence Robotics

Abstract

Latent Action Models (LAMs) enable the learning of world models from unlabeled video by inferring abstract actions between consecutive frames. However, LAMs face a fundamental trade-off between action abstraction and generation fidelity. Existing methods typically circumvent this issue by using two-stage training with pre-trained world models or by limiting predictions to optical flow. In this paper, we introduce DiLA, a novel Disentangled Latent Action world model that aims to resolve this trade-off via content-structure disentanglement. Our key insight is that disentanglement and latent action learning are co-evolving: the predictive bottleneck inherent in latent action learning serves as a driving force for disentanglement, compelling the model to distill spatial layouts into the structure pathway while offloading visual details to a separate content pathway for generation. This synergy yields a continuous, semantically structured latent action space without compromising generative quality. DiLA achieves superior results in video generation quality, action transfer, visual planning, and manifold interpretability. These findings establish DiLA as a unified framework that simultaneously achieves high-level action abstraction and high-fidelity generation, advancing the frontier of self-supervised world model learning.

Keywords

Cite

@article{arxiv.2605.15725,
  title  = {DiLA: Disentangled Latent Action World Models},
  author = {Tianqiu Zhang and Muyang Lyu and Yufan Zhang and Fang Fang and Si Wu},
  journal= {arXiv preprint arXiv:2605.15725},
  year   = {2026}
}

Comments

Project Page: http://disentangled-latent-action-world-models.github.io

R2 v1 2026-07-22T07:13:56.316Z