titleLaWAM: Latent World Action Models for Efficient Dynamics-Aware Robot Policies
authorsJialei Chen, Kai Wang, Kang Chen, Shuaihang Chen, Feng Gao, Wenhao Tang, Zhiyuan Li, Weilin Liu, Zhuyu Yao, Boxun Li, Yuanbo Xu, Chao Yu
date2026-06-14
venuearXiv (preprint, not peer reviewed)
idarXiv:2606.15768
affiliationRLinf
upvotes3 (Hugging Face Daily Papers, 2026-11-01)
linksabs pdf hf code
primerVision-language-action models World models
abstractVision-Language-Action models (VLAs) leverage large-scale vision-language pretraining for semantic robot control, but often lack explicit foresight into how robot actions change the scene. World-Action Models (WAMs) address this limitation by conditioning policies on predicted futures, yet existing approaches typically rely on computationally expensive video generation with substantial pixel-level redundancy. We present LaWAM, a Latent World Action Model that exposes predictive dynamics to robot policies through compact latent visual subgoals instead of reconstructed future video. At the core of LaWAM is a latent-action-conditioned Latent World Model (LaWM). We obtain LaWM by training a latent action model in the latent space of a pretrained vision foundation model and repurposing its forward decoder to predict future observation features for scene evolution. LaWAM then conditions action generation on these predicted latent visual subgoals to enable dynamics-aware robot control. LaWAM achieves state-of-the-art or competitive success rates (SRs) across LIBERO (98.6% SR), RoboTwin (91.22% SR), and real-world manipulation tasks while retaining low-latency inference. LaWAM runs in 187 ms per action-chunk prediction and achieves up to 24x lower wall-clock latency than pixel-space WAMs.

cite

@misc{chen2026lawam,
  title={LaWAM: Latent World Action Models for Efficient Dynamics-Aware Robot Policies},
  author={Jialei Chen and Kai Wang and Kang Chen and Shuaihang Chen and Feng Gao and Wenhao Tang and Zhiyuan Li and Weilin Liu and Zhuyu Yao and Boxun Li and Yuanbo Xu and Chao Yu},
  year={2026},
  eprint={2606.15768},
  archivePrefix={arXiv},
  url={https://arxiv.org/abs/2606.15768}
}