DROID: A Large-Scale In-the-Wild Robot Manipulation Dataset
FeaturedAlexander Khazatsky, Karl Pertsch, Suraj Nair, Ashwin Balakrishna, Sudeep Dasari, Siddharth Karamcheti, Chelsea Finn, Sergey Levine · Stanford, UC Berkeley · 2024
Framework
PyTorch
License
Apache-2.0
Stars
362
Summary
A 350-hour dataset of diverse real-world robot manipulation across 22 robots in 71 scenes, designed to train scalable and generalist imitation learning policies.
Abstract Summary
Key Points
- 350+ hours of real-world manipulation across 22 robots and 71 diverse environments.
- Designed for diversity: many tasks, many scenes, many robot embodiments.
- Pretraining on DROID then fine-tuning outperforms training from scratch.
- Standard train/val splits, PyTorch dataloader, and evaluation server provided.
- Became a standard benchmark alongside Open X-Embodiment for VLA training.
Related Papers
X-Imitator: Spatial-Aware Imitation Learning via Bidirectional Action-Pose Interaction
Kai Xiong, Hongjie Fang, Lixin Yang et al. · arXiv · May 2026
Effectively handling the interplay between spatial perception and action generation remains a critical bottleneck in robotic manipulation. Existing methods typically treat spatial perception and action execution as decoupled or strictly...
LeRobot: A Library for Real-World Robot Learning
Remi Cadene, Simon Alibert, Alexander Soare et al. · NeurIPS 2024 Workshop · Jun 2024
Hugging Face's LeRobot is an open-source PyTorch framework providing pretrained models, datasets, and training scripts for imitation and reinforcement learning on real robots, lowering the entry barrier to robot learning.
Octo: An Open-Source Generalist Robot Policy
Dibya Ghosh, Homer Walke, Karl Pertsch et al. · RSS 2024 · Jul 2024
Octo is a large open-source transformer-based generalist robot policy trained on 800k trajectories, supporting language-conditioned and goal-image-conditioned control across 9 robotic platforms with efficient fine-tuning on consumer GPUs.
Open X-Embodiment: Robotic Learning Datasets and RT-X Models
Open X-Embodiment Collaboration · arXiv · Oct 2023
The largest collaborative robot learning dataset ever assembled, spanning 22 robots from 21 institutions, enabling training of generalist RT-X policies that exhibit positive cross-embodiment transfer.