RT-2:视觉-语言-动作模型将网络知识迁移至机器人控制

RT-2: Vision-Language-Action Models Transfer Web Knowledge to Robotic Control

安东尼·布罗汉 Anthony Brohan · Google DeepMind · 2023-07-28 · arXiv:2307.15818 ↗ · 被引 3509

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

摘要 · Abstract

我们研究了如何将基于互联网规模数据训练的视觉-语言模型直接融入端到端机器人控制,以提升泛化能力并实现涌现的语义推理。我们的目标是让单个端到端训练的模型既能学习将机器人观测映射到动作,又能受益于来自网络的语音和视觉-语言数据的大规模预训练。为此,我们提出在机器人轨迹数据和互联网规模的视觉-语言任务(如视觉问答)上共同微调最先进的视觉-语言模型。与其他方法不同,我们提出了一个简单通用的方案:为了将自然语言响应和机器人动作适配到同一格式,我们将动作表示为文本标记,并像自然语言标记一样直接将其纳入模型的训练集。我们将这类模型称为视觉-语言-动作模型(VLA),并实例化了一个名为 RT-2 的模型。我们的广泛评估(6000 次评估试验)表明,我们的方法产生了高性能的机器人策略,并使 RT-2 从互联网规模训练中获得了一系列涌现能力。这包括对新颖物体的显著改进的泛化能力,解释机器人训练数据中未出现的命令的能力(例如将物体放置在特定数字或图标上),以及根据用户命令执行基本推理的能力(例如拿起最小或最大的物体,或最接近另一个物体的物体)。我们进一步表明,结合思维链推理使 RT-2 能够执行多阶段语义推理,例如找出哪个物体适合用作临时锤子(岩石),或者哪种饮料最适合疲倦的人(能量饮料)。

We study how vision-language models trained on Internet-scale data can be incorporated directly into end-to-end robotic control to boost generalization and enable emergent semantic reasoning. Our goal is to enable a single end-to-end trained model to both learn to map robot observations to actions and enjoy the benefits of large-scale pretraining on language and vision-language data from the web. To this end, we propose to co-fine-tune state-of-the-art vision-language models on both robotic trajectory data and Internet-scale vision-language tasks, such as visual question answering. In contrast to other approaches, we propose a simple, general recipe to achieve this goal: in order to fit both natural language responses and robotic actions into the same format, we express the actions as text tokens and incorporate them directly into the training set of the model in the same way as natural language tokens. We refer to such category of models as vision-language-action models (VLA) and instantiate an example of such a model, which we call RT-2. Our extensive evaluation (6k evaluation trials) shows that our approach leads to performant robotic policies and enables RT-2 to obtain a range of emergent capabilities from Internet-scale training. This includes significantly improved generalization to novel objects, the ability to interpret commands not present in the robot training data (such as placing an object onto a particular number or icon), and the ability to perform rudimentary reasoning in response to user commands (such as picking up the smallest or largest object, or the one closest to another object). We further show that incorporating chain of thought reasoning allows RT-2 to perform multi-stage semantic reasoning, for example figuring out which object to pick up for use as an improvised hammer (a rock), or which type of drink is best suited for someone who is tired (an energy drink).

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →