English

iVideoGPT: Interactive VideoGPTs are Scalable World Models

Computer Vision and Pattern Recognition 2024-11-01 v3 Machine Learning Robotics

Abstract

World models empower model-based agents to interactively explore, reason, and plan within imagined environments for real-world decision-making. However, the high demand for interactivity poses challenges in harnessing recent advancements in video generative models for developing world models at scale. This work introduces Interactive VideoGPT (iVideoGPT), a scalable autoregressive transformer framework that integrates multimodal signals--visual observations, actions, and rewards--into a sequence of tokens, facilitating an interactive experience of agents via next-token prediction. iVideoGPT features a novel compressive tokenization technique that efficiently discretizes high-dimensional visual observations. Leveraging its scalable architecture, we are able to pre-train iVideoGPT on millions of human and robotic manipulation trajectories, establishing a versatile foundation that is adaptable to serve as interactive world models for a wide range of downstream tasks. These include action-conditioned video prediction, visual planning, and model-based reinforcement learning, where iVideoGPT achieves competitive performance compared with state-of-the-art methods. Our work advances the development of interactive general world models, bridging the gap between generative video models and practical model-based reinforcement learning applications. Code and pre-trained models are available at https://thuml.github.io/iVideoGPT.

Keywords

Cite

@article{arxiv.2405.15223,
  title  = {iVideoGPT: Interactive VideoGPTs are Scalable World Models},
  author = {Jialong Wu and Shaofeng Yin and Ningya Feng and Xu He and Dong Li and Jianye Hao and Mingsheng Long},
  journal= {arXiv preprint arXiv:2405.15223},
  year   = {2024}
}

Comments

NeurIPS 2024. Code is available at project website: https://thuml.github.io/iVideoGPT

R2 v1 2026-06-28T16:38:21.692Z