ThinkJEPA: Empowering Latent World Models with Large Vision-Language Reasoning Model

Haichao Zhang, Yijiang Li, Shwai He, Tushar Nagarajan, Mingfei Chen +3 more
3/23/2026
cs.CVcs.AIcs.CL

No official code found. Best implementation: Hai-chao-Zhang/ThinkJEPA (42 stars, last commit 2026-04-04). 1 more below.

Abstract

Recent progress in latent world models (e.g., V-JEPA2) has shown promising capability in forecasting future world states from video observations. Nevertheless, dense prediction from a short observation window limits temporal context and can bias predictors toward local, low-level extrapolation, making it difficult to capture long-horizon semantics and reducing downstream utility. Vision--language models (VLMs), in contrast, provide strong semantic grounding and general knowledge by reasoning over uniformly sampled frames, but they are not ideal as standalone dense predictors due to compute-driven sparse sampling, a language-output bottleneck that compresses fine-grained interaction states into text-oriented representations, and a data-regime mismatch when adapting to small action-conditioned datasets. We propose a VLM-guided JEPA-style latent world modeling framework that combines dense-frame dynamics modeling with long-horizon semantic guidance via a dual-temporal pathway: a dense JEPA branch for fine-grained motion and interaction cues, and a uniformly sampled VLM \emph{thinker} branch with a larger temporal stride for knowledge-rich guidance. To transfer the VLM's progressive reasoning signals effectively, we introduce a hierarchical pyramid representation extraction module that aggregates multi-layer VLM representations into guidance features compatible with latent prediction. Experiments on hand-manipulation trajectory prediction show that our method outperforms both a strong VLM-only baseline and a JEPA-predictor baseline, and yields more robust long-horizon rollout behavior.

View on arXivView PDF

Code Implementations(2)

ThinkJEPA: Empowering Latent World Models with Large Vision-Language Reasoning Model

425Shell, PythonApr 1, 20266 months agoNOASSERTION
computer-visiondexterous-manipulationdexterous-robotic-handjepalarge-language-model+15 more

Unofficial MVP implementation of ThinkJEPA: Empowering Latent World Models with Large Vision-Language Reasoning Model (arXiv:2603.22281)

11PythonMar 25, 20266 months ago

Similar Papers

Cite this paper

@article{zhang2026thinkjepa,
  title  = {ThinkJEPA: Empowering Latent World Models with Large Vision-Language Reasoning Model},
  author = {Haichao Zhang and Yijiang Li and Shwai He and Tushar Nagarajan and Mingfei Chen and Jianglin Lu and Ang Li and Yun Fu},
  year   = {2026},
  eprint = {2603.22281},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2603.22281v1}
}

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/2603.22281.svg)](https://codeofpaper.com/paper/2603.22281)

Discussion