atlas

Research map / Video / Video generation

Learning an Actionable Discrete Diffusion Policy via Large-Scale Actionless Video Pre-Training

NeurIPS 2024 · Video › Video generation · approach: World-model video generation · 3 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.

Pre-trains a unified discrete (mask-and-replace) diffusion model on VQ-VAE video tokens from actionless Ego4D human videos and robot videos, then fine-tunes a separate action denoising network on few robot demonstrations for Meta-World and RLBench manipulation.

Problem

Robot policy learning is limited by scarce action-labeled robot data, and the human-robot domain gap and noisy multimodal human videos make it hard to transfer knowledge from actionless human videos.

Approach

Compress human and robot videos into a shared VQ-VAE token space, pre-train a unified discrete diffusion model to predict future video tokens, and use the predicted-video hidden representations to guide action denoising during few-shot fine-tuning.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Pre-train on human videos in addition to robot videos (ablation)VPDD · Meta-World MT50-randsuccess rate: 6.9% relative improvementHelped
Pre-train on increasing amounts of human videos (ablation)VPDD · Meta-World policy fine-tuningsuccess rateHelped
Use VPDD rather than UniPi for video predictionMeta-World video prediction · Meta-WorldFVD: 264.66 -> 235.81Helped
Use VPDD rather than RVTRLBench multi-task policy · RLBench 16 taskssuccess rate: 1.24× average improvementMixed

Surprising findings

Limitations

Experimental setup

Datasets: Ego4D Meta-World RLBench RoboSet

Benchmarks: MT50-rand RLBench (16 tasks)

Metrics: success rate FVD

Baselines: PerAct RVT R3M-Diffusion VC-1-Diffusion MTDiff-p Video-MTDT SODA UniPi GR-1

Builds on: VQ-VAE VideoGPT D3PM discrete diffusion UniD3 Perceiver Transformer

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