atlas

Research map / Video / Video generation

VidMan: Exploiting Implicit Dynamics from Video Diffusion Model for Effective Robot Manipulation

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

VidMan pre-trains a video diffusion transformer (Open-Sora STDiT) on Open X-Embodiment to predict future frames, then adapts it into an implicit inverse-dynamics action model via layer-wise self-attention adapters and a diffusion action head, evaluated on CALVIN and OXE offline.

Problem

Robot manipulation action prediction is limited by scarce robot data; directly fitting actions from observations ignores the relation between visual dynamics and actions, leading to poor data utilization.

Approach

A two-stage dual-process-inspired training: first pre-train a video diffusion transformer to predict future robot frames (dynamics awareness), then reuse its parameters with a layer-wise self-attention adapter and a diffusion action head to predict actions without generating frames (implicit inverse dynamics).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
use VidMan instead of GR-1CALVIN zero-shot evaluation · CALVINAvg. Len.: 3.06 -> 3.42Helped
pretrain on OXE instead of no video pretraining (ablation)VidMan · CALVINAvg. Len.: 2.89 -> 3.42Helped
pretrain on Ego4d instead of no video pretraining (ablation)VidMan · CALVINAvg. Len.: 2.89 -> 3.29Hurt
co-train video generation and action prediction in stage 2 instead of action-only supervision (ablation)VidMan · CALVINAvg. Len.: 3.42 -> 2.70Hurt
add the layer-wise adapter (ablation)VidMan · CALVINAvg. Len.: 1.54 -> 3.42Helped
train Open-Sora blocks as well as the adapter instead of freezing the blocks (ablation)VidMan · CALVINAvg. Len.: 2.98 -> 3.42Helped
increase frame sampling interval from 1 to 3 (ablation)VidMan · BridgeMSE: 2.80 -> 0.89Helped
increase frame sampling interval from 3 to 4 (ablation)VidMan · CALVINAvg. Len.: 3.42 -> 3.03Hurt
use pure noise instead of zero embeddings in stage 2 (ablation)VidMan · BridgeMSE: 12.1 -> 4.8Helped
use two historical frames instead of one (ablation)VidMan · BridgeAvg. Len.: 2.80 -> 3.42Helped
increase future frames from one to two (ablation)VidMan · BridgeAvg. Len.: 2.80 -> 3.42Helped

Surprising findings

Limitations

Experimental setup

Datasets: Open X-Embodiment (25 selected datasets) CALVIN Bridge Taco Play Cable Routing Autolab UR5 Ego4D RLBench

Benchmarks: CALVIN Bridge Taco Play Cable Routing Autolab UR5 RLBench

Metrics: average task length success rate per chain position MSE xyz accuracy angle accuracy FID FVD

Baselines: GR-1 RoboFlamingo SuSIE HULC MCIL RT-1 3D Diffuser Actor 3D Diffusion Policy RT-1-X Octo-small Octo-base VidMan-GPT

Builds on: Open-Sora GR-1 Octo diffusion policy

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