Open X-Embodiment: Robotic Learning Datasets and RT-X Models
FeaturedOpen X-Embodiment Collaboration · Google DeepMind and 20+ institutions worldwide · 2023
Framework
Jupyter Notebook
License
Apache-2.0
Stars
1,853
Summary
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.
Abstract Summary
Key Points
- Largest open robot learning dataset: 160k+ tasks and 527 skills across 22 robots and 21 institutions.
- Standardized data format enabling training of single policies on heterogeneous robot hardware.
- RT-X generalist policy shows positive transfer and outperforms specialist policies on held-out tasks.
- Action tokenization and cross-embodiment batching are key to multi-robot pre-training stability.
- Open-sourced dataset, baselines, and evaluation protocols to accelerate community research.
The Open X-Embodiment Repository
We introduce the Open X-Embodiment Repository ( robotics-transformer-x.github.io ) – an open-source repository which includes large-scale data along with pre-trained model checkpoints for X-embodied robot learning research.
More specifically, we provide and maintain the following open-source resources to the broader community: Open X-Embodiment Dataset : robot learning dataset with 1M+ robot trajectories from $22$ robot embodiments . Pre-Trained Checkpoints : a selection of RT-X model checkpoints ready for inference and finetuning.
RT-X Design
To evaluate how much X-embodiment training can improve the performance of learned policies on individual robots, we require models that have sufficient capacity to productively make use of such large and heterogeneous datasets. To that end, our experiments will build on two recently proposed Transformer-based robotic policies: RT-1 [ 8 ] and RT-2 [ 9 ] .
We briefly summarize the design of these models in this section, and discuss how we adapted them to the X-embodiment setting in our experiments.
Experimental Results
Our experiments answer three questions about the effect of X-embodiment training: (1) Can policies trained on our X-embodiment dataset effectively enable positive transfer, such that co-training on data collected on multiple robots improves performance on the training task? (2) Does co-training models on data from multiple platforms and tasks improve generalization to new, unseen tasks?
(3) What is the influence of different design dimensions, such as model size, model architecture or dataset composition, on performance and generalization capab...
Sources and Demonstrations
Figures are reproduced from Collaboration See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.
Related Papers
MonoDuo: Using One Robot Arm to Learn Bimanual Policies
Sandeep Bajamahal, Lawrence Yunliang Chen, Toru Lin et al.
Bimanual coordination is essential for many real-world manipulation tasks, yet learning bimanual robot policies is limited by the scarcity of bimanual robots and datasets. Single-arm robots, however,
X-Imitator: Spatial-Aware Imitation Learning via Bidirectional Action-Pose Interaction
Kai Xiong, Hongjie Fang, Lixin Yang et al.
Effectively handling the interplay between spatial perception and action generation remains a critical bottleneck in robotic manipulation.
DROID: A Large-Scale In-the-Wild Robot Manipulation Dataset
Alexander Khazatsky, Karl Pertsch, Suraj Nair et al.
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.
LeRobot: A Library for Real-World Robot Learning
Remi Cadene, Simon Alibert, Alexander Soare et al.
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.