VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model

Jingwen Sun, Wenyao Zhang, Zekun Qi, Shaojie Ren, Zezhi Liu +4 more
2/10/2026
cs.ROcs.CV

No official code found. Best implementation: ginwind/VLA-JEPA (422 stars, last commit 2026-03-20).

Abstract

Pretraining Vision-Language-Action (VLA) policies on internet-scale video is appealing, yet current latent-action objectives often learn the wrong thing: they remain anchored to pixel variation rather than action-relevant state transitions, making them vulnerable to appearance bias, nuisance motion, and information leakage. We introduce VLA-JEPA, a JEPA-style pretraining framework that sidesteps these pitfalls by design. The key idea is leakage-free state prediction: a target encoder produces latent representations from future frames, while the student pathway sees only the current observation -- future information is used solely as supervision targets, never as input. By predicting in latent space rather than pixel space, VLA-JEPA learns dynamics abstractions that are robust to camera motion and irrelevant background changes. This yields a simple two-stage recipe -- JEPA pretraining followed by action-head fine-tuning -- without the multi-stage complexity of prior latent-action pipelines. Experiments on LIBERO, LIBERO-Plus, SimplerEnv and real-world manipulation tasks show that VLA-JEPA achieves consistent gains in generalization and robustness over existing methods.

View on arXivView PDF

Code Implementations(1)

ginwind/VLA-JEPACommunity68%

VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model

42232Shell, PythonFeb 10, 20267 months ago

Similar Papers

Cite this paper

@article{sun2026vlajepa,
  title  = {VLA-JEPA: Enhancing Vision-Language-Action Model with Latent World Model},
  author = {Jingwen Sun and Wenyao Zhang and Zekun Qi and Shaojie Ren and Zezhi Liu and Hanxin Zhu and Guangzhong Sun and Xin Jin and Zhibo Chen},
  year   = {2026},
  eprint = {2602.10098},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.10098v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2602.10098.svg)](https://codeofpaper.com/paper/2602.10098)

Discussion