Learning from Next-Frame Prediction: Towards Unified and Scalable Visual World Modeling
AuthorsYifeng Yao, Jinghan Li, Zhihao Zhang, Yajiao Xiong, Yang Jin, Kun Xu, Huishuai Zhang, Dongyan Zhao
AffiliationsPeking University · ByteDance · UNC-Chapel Hill
VANE learns visual representations by predicting what comes next, using the same objective for both images and videos.
Key results
Parameters in the main model’s ViT-g encoder.
Images used for pretraining.
Video samples used for pretraining.
Action mean-class recall@5; the paper reports a 7.1-point gain over V-JEPA 2.1.
VANE’s semantic segmentation score, versus 47.8 for V-JEPA 2.1.
What the paper found
ByteDance’s VANE changes visual pretraining from filling in masked content to predicting the next frame’s meaning from past frames. A frame-causal encoder and state predictor forecast the full next-frame representation in the fixed semantic space of a frozen DINOv2 ViT-L tokenizer; an auxiliary flow-matching decoder helps preserve fine spatial detail. For images, VANE applies the same objective to local-to-global crop sequences, enabling unified image–video training. The 1B-parameter model was pretrained on VISTA, a corpus of 145M images and 24M video samples. On EPIC-KITCHENS-100, it achieves 45.5 action mean-class recall@5, 7.1 points above V-JEPA 2.1. It also reaches 51.0 mIoU on ADE20K, compared with 47.8 for V-JEPA 2.1. The results suggest that predicting future frame-level embeddings can produce representations useful for anticipating events as well as recognizing scenes and preserving spatial detail.
Original abstract
Video provides a natural source of supervision: what happens next serves as a learning signal for visual world modeling. Yet masked video pretraining focuses on recovering missing content from surrounding context rather than predicting the future from the past. Autoregressive methods instead predict forward one token at a time, but have yet to match joint-embedding methods in representation quality. This gap motivates us to revisit the granularity and space of future prediction. In this work, we introduce VANE, a self-supervised visual pretraining framework built on next-frame prediction in embedding space. A causal encoder summarizes past observations for a state predictor that infers the representation of the entire next frame, making temporal evolution the primary learning signal. This prediction is carried out in a fixed embedding space provided by a frozen semantic tokenizer, emphasizing scene content and structure. To retain fine-grained spatial detail alongside semantic content, a lightweight auxiliary generative decoder reconstructs future-frame patches conditioned on the predicted representations. To unify visual modeling across images and videos, we construct a local-to-global crop sequence for still images, allowing the same encoder to predict newly revealed regions from earlier crops under the same objective. Experiments show that VANE improves with model and data scaling, achieving SOTA on most anticipation, recognition, and dense perception benchmarks among unified encoders. Especially on EPIC-KITCHENS-100 action anticipation, it surpasses V-JEPA 2.1 by 7.1 points in mean-class recall@5. These results demonstrate that next-frame prediction in embedding space is effective and scalable for visual world modeling.
Read the original paperMore in World Models
Browse all 50 papers →LaLa-World: Scaling Latent World Models with Natural Language Actions
Junha Song, Geonmo Gu, Jeongeun Park, Heejeong Nam, Lucas Maes, Byeongho Heo, Dongyoon Han, Sangdoo Yun
LaLa-World uses language to train a scalable latent model of how the world changes, then uses its imagined futures to plan robot actions.
Training Controllable World Models via Action Influence Maximization
Samuel M. Bateman, Tenny Yin, Chongyi Zheng, Lei Huang, Brian Wang, Benjamin Eysenbach, Jaime Fernández Fisac, Dhruv Shah
A new active data-collection method helps video world models respond to commanded actions instead of merely imagining plausible futures.
Agent World Model in the Wild: Computer-Use Agent Environment Synthesis
Zhaoyang Wang, Qianhui Wu, Haorui Wang, Simon Yu, Baolin Peng, Canwen Xu, Hao Cheng, Wenlin Yao, Boyi Liu, Yite Wang, Jianfeng Gao, Huaxiu Yao
WildAWM builds synthetic, interactive computer worlds where agents can practice complex tasks and improve at unseen computer-use benchmarks.