English

World2Act: Latent Action Post-Training via Skill-Compositional World Models

Computer Vision and Pattern Recognition 2026-03-12 v1

Abstract

World Models (WMs) have emerged as a promising approach for post-training Vision-Language-Action (VLA) policies to improve robustness and generalization under environmental changes. However, most WM-based post-training methods rely on pixel-space supervision, making policies sensitive to pixel-level artifacts and hallucination from imperfect WM rollouts. We introduce World2Act, a post-training framework that aligns VLA actions directly with WM video-dynamics latents using a contrastive matching objective, reducing dependence on pixels. Post-training performance is tied to rollout quality, yet current WMs struggle with arbitrary-length video generation as they are mostly trained on fixed-length clips while robotic execution durations vary widely. To address this, we propose an automatic LLM-based skill-decomposition pipeline that segments high-level instructions into low-level prompts. Our pipeline produces RoboCasa-Skill and LIBERO-Skill, supporting skill-compositional WMs that remain temporally consistent across diverse task horizons. Empirically, applying World2Act to VLAs like GR00T-N1.6 and Cosmos Policy achieves state-of-the-art results on RoboCasa and LIBERO, and improves real-world performance by 6.7%, enhancing embodied agent generalization.

Keywords

Cite

@article{arxiv.2603.10422,
  title  = {World2Act: Latent Action Post-Training via Skill-Compositional World Models},
  author = {An Dinh Vuong and Tuan Van Vo and Abdullah Sohail and Haoran Ding and Liang Ma and Xiaodan Liang and Anqing Duan and Ivan Laptev and Ian Reid},
  journal= {arXiv preprint arXiv:2603.10422},
  year   = {2026}
}

Comments

Project page: https://wm2act.github.io/

R2 v1 2026-07-01T11:14:09.308Z