Models / 3D Diffuser Actor

3D Diffuser Actor

CMU · Feb 2024 · MIT

Marries 3D scene representations with diffusion policies: lifts multi-view features into 3D tokens and denoises end-effector trajectories, setting state of the art on RLBench and CALVIN.

Success rate
81%
Downloads
21K
Control rate
20 Hz
Training episodes
3.2K

Benchmarks

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

EvaluationSuiteScore

RLBench PerAct setup (18 tasks)

RLBench81.3%success

RLBench GNFactor setup

RLBench78.4%success

CALVIN long-horizon

CALVIN ABC->D3.4avg tasks per chain

Training

Architecture

3D scene-token diffusion policy

Parameters

180M

Hardware

8x A100 80GB

Compute

400 accelerator-hours

Training data hours

45 h

Training episodes

3.2K

Recipe

Relative 3D attention over scene tokens, DDPM over trajectories, keyframe supervision.

Fine-tuning notes

Depth quality dominates: noisy point clouds cost more than smaller models.

Linked datasets, robots & papers

Resolving related models, datasets and papers…