开放 X-具身:机器人学习数据集与 RT-X 模型

Open X-Embodiment: Robotic Learning Datasets and RT-X Models

迈克尔·安 Michael Ahn · Google DeepMind · 2023-10-13 · arXiv:2310.08864 ↗ · 被引 1083

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

大规模、高容量模型在多样化数据集上训练后,在高效处理下游应用方面取得了显著成功。从自然语言处理到计算机视觉,这导致了预训练模型的整合,通用预训练骨干网络成为许多应用的起点。这种整合能否在机器人领域发生?传统上,机器人学习方法为每个应用、每个机器人甚至每个环境训练单独的模型。我们能否训练通用的跨机器人策略,使其能够高效适应新的机器人、任务和环境?在本文中,我们提供了标准化数据格式的数据集和模型,以探索在机器人操作领域实现这一可能性的途径,并附有实验结果为有效的跨机器人策略提供示例。我们通过 21 个机构的合作,从 22 个不同的机器人收集了数据集,展示了 527 种技能(160266 个任务)。我们表明,在此数据上训练的高容量模型(称为 RT-X)展现出正向迁移,并通过利用其他平台的经验提升了多个机器人的能力。更多详情请见项目网站 https://robotics-transformer-x.github.io。

Large, high-capacity models trained on diverse datasets have shown remarkable successes on efficiently tackling downstream applications. In domains from NLP to Computer Vision, this has led to a consolidation of pretrained models, with general pretrained backbones serving as a starting point for many applications. Can such a consolidation happen in robotics? Conventionally, robotic learning methods train a separate model for every application, every robot, and even every environment. Can we instead train generalist X-robot policy that can be adapted efficiently to new robots, tasks, and environments? In this paper, we provide datasets in standardized data formats and models to make it possible to explore this possibility in the context of robotic manipulation, alongside experimental results that provide an example of effective X-robot policies. We assemble a dataset from 22 different robots collected through a collaboration between 21 institutions, demonstrating 527 skills (160266 tasks). We show that a high-capacity model trained on this data, which we call RT-X, exhibits positive transfer and improves the capabilities of multiple robots by leveraging experience from other platforms. More details can be found on the project website https://robotics-transformer-x.github.io.

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →