atlas

Research map / Video / Action recognition

Genie: Generative Interactive Environments

ICML 2024 · Video › Action recognition · approach: Action anticipation and latent actions · 12 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.

Genie trains an 11B-parameter generative interactive environment from 30k hours of unlabelled Internet platformer videos, learning discrete latent actions that let users control generated frames frame-by-frame.

Problem

Interactive, action-controllable video generation normally requires action-labelled data, which is scarce for Internet videos; the paper asks whether controllable generative environments can be learned from video alone.

Approach

Learn a discrete latent action space unsupervised with a VQ-VAE-based latent action model over raw pixels, combined with an ST-transformer video tokenizer and a MaskGIT-style dynamics model, so any image prompt can be played frame-by-frame with latent actions.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use tokenized images rather than pixels as latent action model input (ablation)Genie latent action model · PlatformersPSNRdiff: 1.91 -> 1.33Hurt
Use tokenized images rather than pixels as latent action model input (ablation)Genie latent action model · RoboticsPSNRdiff: 2.07 -> 1.65Hurt
Use tokenized images rather than pixels as latent action model input (ablation)Genie latent action model · PlatformersFVD: 40.1 -> 38.8Helped
Use tokenized images rather than pixels as latent action model input (ablation)Genie latent action model · RoboticsFVD: 136.4 -> 257.8Hurt
Replace ST-ViViT tokenizer with spatial-only ViT (ablation)Genie video tokenizer · Video tokenizer architecture ablationFVD: 81.4 -> 114.5Hurt
Replace ST-ViViT tokenizer with spatial-only ViT (ablation)Genie video tokenizer · Video tokenizer architecture ablationPSNRdiff: 1.66 -> 1.39Hurt
Replace ST-ViViT tokenizer with C-ViViT (ablation)Genie video tokenizer · Video tokenizer architecture ablationFVD: 81.4 -> 272.7Hurt
Replace ST-ViViT tokenizer with C-ViViT (ablation)Genie video tokenizer · Video tokenizer architecture ablationPSNRdiff: 1.66 -> 1.37Hurt
Curate the training videos instead of using the original dataset (ablation)Genie · Platformers datasetFVD: 61.4 -> 54.8Helped
Increase tokenizer batch size from 64 to 384 (ablation)Video tokenizer · Video tokenizerPSNR: 35.7 -> 36.5Helped

Surprising findings

Limitations

Experimental setup

Datasets: Platformers (filtered Internet 2D platformer videos) Robotics (RT1 and related robot videos) Procgen CoinRun

Benchmarks: Platformers test split Robotics test split Procgen CoinRun behavioral cloning

Metrics: FVD PSNR-Delta (controllability) percentage of levels solved

Baselines: C-ViViT ViT tokenizer token-input latent action model oracle behavioral cloning random agent

Builds on: MaskGIT VQ-VAE ViViT Phenaki C-ViViT VPT

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