Search papers, labs, and topics across Lattice.
This paper introduces RepWAM, a novel world action model that utilizes representation visual-action tokenizers to enhance instruction-following dynamics in robotic control. By shifting from traditional reconstruction-oriented video tokenizers to a semantic visual-action latent space, RepWAM effectively aligns visual inputs with latent action tokens, enabling improved future state prediction. Experimental results demonstrate that RepWAM outperforms existing models in real-world manipulation tasks, underscoring the advantages of semantic tokenization in developing generalist robot policies.
Semantic visual-action tokenization in RepWAM significantly enhances robotic manipulation performance, outperforming traditional reconstruction-based approaches.
This work presents RepWAM, a representation-centric world action model (WAM) built on representation visual-action tokenizers. Existing WAMs typically inherit reconstruction-oriented video tokenizers from pretrained video generation models. Although these tokenizers preserve visual fidelity, pixel reconstruction alone provides limited guidance for learning instruction-following dynamics that connect future prediction with robot control. To address this, we explore a semantic visual-action latent space for representation-centric world action modeling. Specifically, we train a representation visual-action tokenizer that maps visual inputs into aligned visual and latent action tokens. We then pretrain our WAM to jointly model future visual states and the latent actions that connect them under language instructions, followed by adaptation to real robot trajectories for closed-loop manipulation. Experiments on real-world manipulation tasks and simulation benchmarks show that RepWAM delivers strong performance across diverse manipulation settings, while ablations highlight the value of semantic visual-action tokenization over reconstruction-oriented alternatives. These results establish representation visual-action tokenization as a promising foundation for world action models and a step toward generalist robot policies. Code and weights will be available at https://github.com/wdrink/RepWAM.
