CRAFT: Video Diffusion for Bimanual Robot Data Generation
Organizations: Thomas Lord Department of Computer Science at the University of Southern California, USA
Abstract
Bimanual robot learning from demonstrations is fundamentally limited by the cost and narrow visual diversity of real-world data, which constrains policy robustness across viewpoints, object configurations, and embodiments. We present Canny-guided Robot Data Generation using Video Diffusion Transformers (CRAFT), a video diffusion-based framework for scalable bimanual demonstration generation that synthesizes temporally coherent manipulation videos while producing action labels. By conditioning video diffusion on edge-based structural cues extracted from simulator-generated trajectories, CRAFT produces physically plausible trajectory variations and supports a unified augmentation pipeline spanning object pose changes, camera viewpoints, lighting and background variations, cross-embodiment transfer, and multi-view synthesis. We leverage a pre-trained video diffusion model to convert simulated videos, along with action labels from the simulation trajectories, into action-consistent demonstrations. Starting from only a few real-world demonstrations, CRAFT generates a large, visually diverse set of photorealistic training data, bypassing the need to replay demonstrations on the real robot (Sim2Real). Across simulated and real-world bimanual tasks, CRAFT improves success rates over existing augmentation strategies and straightforward data scaling, demonstrating that diffusion-based video generation can substantially expand demonstration diversity and improve generalization for dual-arm manipulation tasks. Our project website is available at: https://craftaug.github.io/
Figures & tables
| Simulation (%) | Real-World ( / 20 ) | |||||
|---|---|---|---|---|---|---|
| Method | LP | PC | SB | LR | PC | SB |
| Collected Target | 55.0% | 69.0% | 59.0% | 5 / 20 | 2 / 20 | 3 / 20 |
| Shadow [ 10 ] | 2.0% | 2.3% | 6.0% | 2 / 20 | 1 / 20 | 1 / 20 |
| CRAFT (Target) | 11.3% | 6.0% | 21.6% | 4 / 20 | 1 / 20 | 2 / 20 |
| CRAFT (Ours) | 82.6% | 89.3% | 86.0% | 17 / 20 | 15 / 20 | 16 / 20 |
| Method | Stack Bowls (SB) |
|---|---|
| Collected Demos (Upper Bound) | 59.0% |
| CRAFT w/o Canny | 10.3% |
| CRAFT w/ Canny | 21.6% |
| Lighting | Background | Camera View | Object Color | Wrist + 3rd Person | |||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Method | LR | PC | SB | LR | PC | SB | LR | PC | SB | LR | PC | SB | LR | PC | SB |
| ACT w/o Aug | 3 / 20 | 1 / 20 | 0 / 20 | 4 / 20 | 0 / 20 | 0 / 20 | 6 / 20 | 3 / 20 | 2 / 20 | 2 / 20 | 0 / 20 | 1 / 20 | 15 / 20 | 11 / 20 | 13 / 20 |
| CRAFT Pose-Only | 5 / 20 | 3 / 20 | 2 / 20 | 7 / 20 | 2 / 20 | 3 / 20 | 13 / 20 | 5 / 20 | 7 / 20 | 5 / 20 | 2 / 20 | 3 / 20 | 13 / 20 | 8 / 20 | 10 / 20 |
| ACT w/ Baseline Aug | 13 / 20 | 9 / 20 | 8 / 20 | 4 / 20 | 5 / 20 | 6 / 20 | 14 / 20 | 8 / 20 | 6 / 20 | 15 / 20 | 9 / 20 | 11 / 20 | N/A † | N/A † | N/A † |
| CRAFT (Ours) | 17 / 20 | 14 / 20 | 12 / 20 | 18 / 20 | 15 / 20 | 10 / 20 | 19 / 20 | 18 / 20 | 18 / 20 | 18 / 20 | 18 / 20 | 17 / 20 | 20 / 20 | 19 / 20 | 20 / 20 |
Synthetic Data Generation for Robotics
Synthetic data generation for robotics creates simulated scenes, sensor readings, and action trajectories to train and evaluate robots.
See all 303 papers