atlas

Research map / Video / Video segmentation and tracking

iVideoGPT: Interactive VideoGPTs are Scalable World Models

NeurIPS 2024 · Video › Video segmentation and tracking · approach: Physics-inspired motion prediction · 8 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes iVideoGPT, a scalable autoregressive transformer world model with compressive conditional VQGAN tokenization, pre-trained on ~1.4M robot and human manipulation trajectories and adapted to video prediction, visual planning, and visual MBRL.

Problem

Existing video generation models lack step-level action interactivity, while recurrent world models for RL lack scalability to complex, in-the-wild data; the paper asks how to combine interactivity and scalability for visual world models.

Approach

Tokenize future frames with a conditional VQGAN whose prediction encoder/decoder cross-attends to context-frame features, yielding 4x4 tokens per future frame (16x fewer than 16x16 per-frame tokens), and feed frames, actions, and rewards as one token sequence to a LLaMA-style GPT-like transformer trained with next-token prediction.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
condition the model on actions instead of using action-free predictioniVideoGPT · BAIR, 64×64FVD: 75.0±0.20 -> 60.8±0.08Helped
use iVideoGPT instead of FitVidaction-conditioned video prediction · RoboNet, 64×64FVD: 62.5 -> 63.2±0.01Hurt
use iVideoGPT instead of FitVidaction-conditioned video prediction · RoboNet, 64×64PSNR: 28.2 -> 27.8±0.01Hurt
use iVideoGPT instead of FitVidaction-conditioned video prediction · RoboNet, 64×64SSIM: 89.3 -> 90.6±0.02Helped
use iVideoGPT instead of FitVidaction-conditioned video prediction · RoboNet, 64×64LPIPS: 2.4 -> 4.9±0.00Hurt
use iVideoGPT instead of MaskViTaction-conditioned video prediction · RoboNet, 256×256FVD: 211.7 -> 197.9±0.66Helped
use iVideoGPT instead of MaskViTaction-conditioned video prediction · RoboNet, 256×256LPIPS: 17.0 -> 14.7±0.01Helped
pre-train on mixed manipulation videos before fine-tuningiVideoGPT · BAIR with 100 or 1,000 action-free trajectoriesvideo prediction performanceHelped
pre-train before fine-tuningiVideoGPT · BAIR with full downstream datavideo prediction performanceNo clear effect
increase the transformer from 138M to 436M parametersiVideoGPT · pre-training datasetvalidation lossHelped
use conditional 4×4 future-frame tokenization rather than a standard 16×16 tokenizeriVideoGPT tokenizer · RoboNet video predictioncomputational efficiency: 16×Helped
use conditional 4×4 future-frame tokenization rather than a standard 16×16 tokenizeriVideoGPT tokenizer · RoboNet video predictionreconstruction qualityHurt

Surprising findings

Limitations

Experimental setup

Datasets: Open X-Embodiment Something-Something-v2 BAIR RoboNet VP^2 Meta-World

Benchmarks: BAIR RoboNet VP^2 Meta-World

Metrics: FVD PSNR SSIM LPIPS success rate interquartile mean

Baselines: MaskViT MAGVIT FitVid MCVD SVG DreamerV3 VideoGPT SVG'

Builds on: VQGAN VideoGPT MBPO DrQ-v2 ContextWM

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy