BridgeData V2: A Dataset for Robot Learning at Scale

By Homer Rich Walke, Kevin Black, Abraham Lee, Moo Jin Kim, Maximilian Du, Chongyi Zheng, Tony Zhao, Philippe Hansen-Estruch, Quan Vuong, Andre Wang He and 4 more. Conference on Robot Learning, 2023

762 citations recorded by the source corpus.

We introduce BridgeData V2, a large and diverse dataset of robotic manipulation behaviors designed to facilitate research on scalable robot learning. BridgeData V2 contains 60,096 trajectories collected across 24 environments on a publicly available low-cost robot. BridgeData V2 provides extensive task and environment variability, leading to skills that can generalize across environments, domains, and institutions, making the dataset a useful resource for a broad range of researchers. Additionally, the dataset is compatible with a wide variety of open-vocabulary, multi-task learning methods conditioned on goal images or natural language instructions. In our experiments, we train 6 state-of-the-art imitation learning and offline reinforcement learning methods on our dataset, and find that they succeed on a suite of tasks requiring varying amounts of generalization. We also demonstrate that the performance of these methods improves with more data and higher capacity models, and that training on a greater variety of skills leads to improved generalization. By publicly sharing BridgeData V2 and our pre-trained models, we aim to accelerate research in scalable robot learning methods. Project page at https://rail-berkeley.github.io/bridgedata

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Health Informatics

View Paper: https://www.semanticscholar.org/paper/6a375be82efc01ec4ed73334655935a56ba82d38

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Sergey Levine
  2. Chelsea Finn
  3. Pieter Abbeel
  4. Aviral Kumar
  5. Yilun Du
  6. Karol Hausman
  7. Ofir Nachum
  8. Suraj Nair

In Computer Vision and Pattern Recognition

  1. Sergey Levine
  2. Luc Van Gool
  3. Li Fei-Fei
  4. Jitendra Malik
  5. Dacheng Tao
  6. Trevor Darrell
  7. Xiaogang Wang
  8. Andrea Vedaldi

Citations in the corpus