← Paper Digest

DROID: A Large-Scale In-The-Wild Robot Manipulation Dataset

Alexander Khazatsky, Karl Pertsch, Suraj Nair, Ashwin Balakrishna, Sudeep Dasari, Siddharth Karamcheti, Soroush Nasiriany, Mohan Kumar Srirama, Lawrence Yunliang Chen, Kirsty Ellis, Peter David Fagan, Joey Hejna, Masha Itkina, Marion Lepert, Yecheng Jason Ma, Patrick Tree Miller, Jimmy Wu, Suneel Belkhale, Shivin Dass, Huy Ha, Arhan Jain, Abraham Lee, Youngwoon Lee, Marius Memmel, Sungjae Park, Ilija Radosavovic, Kaiyuan Wang, Albert Zhan, Kevin Black, Cheng Chi, Kyle Beltran Hatch, Shan Lin, Jingpei Lu, Jean Mercat, Abdul Rehman, Pannag R Sanketi, Archit Sharma, Cody Simpson, Quan Vuong, Homer Rich Walke, Blake Wulfe, Ted Xiao, Jonathan Heewon Yang, Arefeh Yavary, Tony Z. Zhao, Christopher Agia, Rohan Baijal, Mateo Guaman Castro, Daphne Chen, Qiuyu Chen, Trinity Chung, Jaimyn Drake, Ethan Paul Foster, Jensen Gao, Vitor Guizilini, David Antonio Herrera, Minho Heo, Kyle Hsu, Jiaheng Hu, Muhammad Zubair Irshad, Donovon Jackson, Charlotte Le, Yunshuang Li, Kevin Lin, Roy Lin, Zehan Ma, Abhiram Maddukuri, Suvir Mirchandani, Daniel Morton, Tony Nguyen, Abigail O'Neill, Rosario Scalise, Derick Seale, Victor Son, Stephen Tian, Emi Tran, Andrew E. Wang, Yilin Wu, Annie Xie, Jingyun Yang, Patrick Yin, Yunchu Zhang, Osbert Bastani, Glen Berseth, Jeannette Bohg, Ken Goldberg, Abhinav Gupta, Abhishek Gupta, Dinesh Jayaraman, Joseph J Lim, Jitendra Malik, Roberto Martín-Martín, Subramanian Ramamoorthy, Dorsa Sadigh, Shuran Song, Jiajun Wu, Michael C. Yip, Yuke Zhu, Thomas Kollar, Sergey Levine, Chelsea Finn

Seminal March 2024

Teaser of DROID scenes grouped by room type, with dataset statistics.

TL;DR A teleoperated manipulation dataset of 76k demonstrations across 564 scenes and 86 tasks, collected on 18 identical Franka setups at 13 institutions. Co-training a diffusion policy on DROID beats the next best training mix by 22 points of success in distribution and 17 points out of distribution.

What they do

DROID is a teleoperated robot manipulation dataset built for scene diversity: 76k successful demonstrations (350 hours) across 564 scenes in 52 buildings and 86 tasks. Fifty data collectors at 13 institutions in North America, Asia and Europe recorded it over 12 months. Each episode has three stereo camera views, depth, camera calibration and crowdsourced language instructions. The paper measures the dataset’s diversity and tests whether co-training on it improves policies.

How it works

  • Shared platform. 18 copies of one setup: a Franka Panda with a Robotiq 2F-85 gripper on a wheeled, height-adjustable desk, two tripod-mounted Zed 2 cameras plus a wrist Zed Mini, and Meta Quest 2 controllers for teleoperation. Actions are recorded at 15 Hz.
  • Collection protocol. A GUI picks each episode’s task at random from those the collector listed for the scene and periodically asks for scene changes (base, cameras, lighting, objects). Collectors move on after about 20 minutes per scene.
  • Policy test. A language-conditioned diffusion policy (two external cameras, ResNet-50 encoder, U-Net head) trained on 50-150 in-domain demos per task. Training batches are mixed 50/50 with DROID, with Open X-Embodiment (OXE), or not mixed at all.

Results

  • 564 scenes, compared with 311 for the whole OXE aggregation and 24 for Bridge V2, plus 1417 distinct camera viewpoints.
  • Across 6 tasks in 4 locations (10 rollouts per setting), DROID co-training beats the next best method by 22 points of success in distribution and 17 points out of distribution. On the three-step Cook Lentils task, it is the only method that consistently finishes all steps.
  • At an equal size of 7,362 trajectories, a scene-diverse sample of DROID gives better out-of-distribution performance than the 20 scenes with the most demonstrations.

Limitations

All data comes from one robot, the Franka Panda. DROID is only tested as co-training data next to in-domain demonstrations; whether policies work in new scenes without in-domain data is left open. Each setting gets 10 rollouts, and distractors inside the desk made every policy fail. The post-hoc camera calibrations still contain false positives and struggle in cluttered, low-overlap scenes.

Videos

A DROID co-trained policy cooks lentils in a home kitchen.
A DROID co-trained policy puts an eraser in a desk drawer and closes it.

Figures

From Figure 0: The DROID robot platform, shared by all 13 institutions: a Franka Panda arm, two adjustable Zed 2 ster...
From Figure 0: The DROID robot platform, shared by all 13 institutions: a Franka Panda arm, two adjustable Zed 2 stereo cameras, a wrist-mounted Zed Mini and an Oculus Quest 2 headset for teleoperation, all on a portable height-adjustable desk.
From Figure 2: Number of scenes per scene type in DROID, Bridge V2, RT-1 and RH20T. DROID has an order of magnitude m...
From Figure 2: Number of scenes per scene type in DROID, Bridge V2, RT-1 and RH20T. DROID has an order of magnitude more scenes, across a wider range of scene types.
From Figure 6: Success rates per task, in distribution (top) and out of distribution (bottom), for no co-training, co...
From Figure 6: Success rates per task, in distribution (top) and out of distribution (bottom), for no co-training, co-training with Open X-Embodiment (OXE) and co-training with DROID. On average DROID beats the next best method by 22 points in distribution and 17 points out of distribution.
Abstract

The creation of large, diverse, high-quality robot manipulation datasets is an important stepping stone on the path toward more capable and robust robotic manipulation policies. However, creating such datasets is challenging: collecting robot manipulation data in diverse environments poses logistical and safety challenges and requires substantial investments in hardware and human labour. As a result, even the most general robot manipulation policies today are mostly trained on data collected in a small number of environments with limited scene and task diversity. In this work, we introduce DROID (Distributed Robot Interaction Dataset), a diverse robot manipulation dataset with 76k demonstration trajectories or 350 hours of interaction data, collected across 564 scenes and 84 tasks by 50 data collectors in North America, Asia, and Europe over the course of 12 months. We demonstrate that training with DROID leads to policies with higher performance and improved generalization ability. We open source the full dataset, policy learning code, and a detailed guide for reproducing our robot hardware setup.

Summaries are written automatically from each paper's text. Figures and videos come from the paper and its project page; all credit goes to the authors.