Models / Diffusion Policy

Diffusion Policy (UNet)

Columbia / TRI / MIT · Mar 2023 · MIT

Formulates visuomotor control as conditional denoising diffusion over action sequences, with receding-horizon replanning. Averaged a 46.9% relative improvement over prior state of the art across 15 tasks.

Success rate
76%
Downloads
93K
Control rate
10 Hz
Training episodes
5.5K

Benchmarks

Numbers as reported by the authors on the named evaluation suite.

EvaluationSuiteScore

15-task average improvement

Robomimic + Push-T + kitchen46.9%relative gain

Push-T

Push-T91%max IoU

Robomimic Square (PH)

Robomimic98%success

Real bimanual tasks

Real Franka76%success

Training

Architecture

Conditional DDPM visuomotor policy

Parameters

110M

Hardware

8x A100 40GB

Compute

320 accelerator-hours

Training data hours

60 h

Training episodes

5.5K

Recipe

100-step DDPM (or 10-step DDIM) with FiLM conditioning, EMA weights, receding-horizon control.

Fine-tuning notes

Position control beats velocity control; EMA and observation horizon 2 are critical for stability.

Linked datasets, robots & papers

Resolving related models, datasets and papers…