OpenVLA: An Open-Source Vision-Language-Action Model | AGIDaily
AGI Daily
资讯日报词条论文人物新创
返回论文列表
摘要1 引言2 相关工作视觉条件语言模型通用机器人策略视觉-语言-动作模型3 OpenVLA 模型3.1 预备知识:视觉-语言模型3.2 OpenVLA 训练流程3.3 训练数据3.4 OpenVLA 的设计决策3.5 训练与推理基础设施4 OpenVLA 代码库5 实验5.1 在多个机器人平台上的直接评测5.2 数据高效地适配新的机器人配置5.3 参数高效微调5.4 通过量化实现省显存的推理6 讨论与局限致谢参考文献附录 A 数据配比细节附录 B 评测任务与详细结果B.1 BridgeData V2 WidowX 评测细节B.1.1 BridgeData V2 评测任务B.1.2 评测任务与原始 BridgeData V2 训练数据的比较B.1.3 BridgeData V2 详细评测结果B.2 Google 机器人评测细节B.2.2 Google 机器人详细评测结果B.3 数据高效适配实验细节B.3.1 Franka-Tabletop 与 Franka-DROID 任务B.3.2 Franka-Tabletop 与 Franka-DROID 详细评测结果附录 C BridgeData V2 评测中的 RT-2-X 与 OpenVLA附录 D 补充实验与消融D.1 OpenX 训练数据消融实验D.2 双视觉编码器与单视觉编码器实验D.3 微调视觉编码器与冻结视觉编码器实验D.4 补充量化推理实验:把策略性能与模型推理速度分离开附录 E LIBERO 仿真实验E.1 LIBERO 仿真实验设置E.2 LIBERO 仿真实验结果
返回论文列表
目录

OpenVLA: An Open-Source Vision-Language-Action Model

收录于 CoRL 2024,PMLR 270:2679–2713;arXiv:2406.09246被引 4,098 次
打开论文来源

OpenVLA: An Open-Source Vision-Language-Action Model

OpenVLA:一个开源的视觉-语言-动作模型

Moo Jin Kim    Karl Pertsch    Siddharth Karamcheti    Ted Xiao    Ashwin Balakrishna    Suraj Nair    Rafael Rafailov    Ethan Foster    Grace Lam    Pannag Sanketi    Quan Vuong    Thomas Kollar    Benjamin Burchfiel    Russ Tedrake    Dorsa Sadigh    Sergey Levine    Percy Liang    Chelsea Finn https://openvla.github.io

Moo Jin Kim∗、Karl Pertsch∗、Siddharth Karamcheti∗、Ted Xiao、Ashwin Balakrishna、Suraj Nair、Rafael Rafailov、Ethan Foster、Grace Lam、Pannag Sanketi、Quan Vuong、Thomas Kollar、Benjamin Burchfiel、Russ Tedrake、Dorsa Sadigh、Sergey Levine、Percy Liang、Chelsea Finn https://openvla.github.io

∗: denotes equal contribution. Correspondence to: moojink@stanford.edu, pertsch@berkeley.edu, skaramcheti@stanford.edu. 1 Stanford University, 2 UC Berkeley, 3 Toyota Research Institute, 4 Google Deepmind, 5 Physical Intelligence, 6 MIT. †Work done in part while at Google Deepmind.

∗:同等贡献。通讯地址:moojink@stanford.edu、pertsch@berkeley.edu、skaramcheti@stanford.edu。1 斯坦福大学,2 加州大学伯克利分校,3 丰田研究院,4 Google DeepMind,5 Physical Intelligence,6 麻省理工学院。†部分工作在 Google DeepMind 期间完成。

Abstract

摘要

Large policies pretrained on a combination of Internet-scale vision-language data and diverse robot demonstrations have the potential to change how we teach robots new skills: rather than training new behaviors from scratch, we can fine-tune such vision-language-action (VLA) models to obtain robust, generalizable policies for visuomotor control. Yet, widespread adoption of VLAs for robotics has been challenging as 1) existing VLAs are largely closed and inaccessible to the public, and 2) prior work fails to explore methods for efficiently fine-tuning VLAs for new tasks, a key component for adoption. Addressing these challenges, we introduce OpenVLA, a 7B-parameter open-source VLA trained on a diverse collection of 970k real-world robot demonstrations. OpenVLA builds on a Llama 2 language model combined with a visual encoder that fuses pretrained features from DINOv2 and SigLIP. As a product of the added data diversity and new model components, OpenVLA demonstrates strong results for generalist manipulation, outperforming closed models such as RT-2-X (55B) by 16.5% in absolute task success rate across 29 tasks and multiple robot embodiments, with 7x fewer parameters. We further show that we can effectively fine-tune OpenVLA for new settings, with especially strong generalization results in multi-task environments involving multiple objects and strong language grounding abilities, and outperform expressive from-scratch imitation learning methods such as Diffusion Policy by 20.4% We also explore compute efficiency; as a separate contribution, we show that OpenVLA can be fine-tuned on consumer GPUs via modern low-rank adaptation methods and served efficiently via quantization without a hit to downstream success rate. Finally, we release model checkpoints, fine-tuning notebooks, and our PyTorch codebase with built-in support for training VLAs at scale on Open X-Embodiment datasets.

在互联网规模的视觉-语言数据和多样化机器人示教数据上联合预训练的大型策略,有可能改变我们教机器人新技能的方式:与其从零训练新行为,不如微调这类视觉-语言-动作(VLA)模型,得到鲁棒、可泛化的视觉运动控制策略。然而,VLA 在机器人领域的广泛应用一直面临困难:1)现有 VLA 大多是封闭的,公众无法获取;2)先前工作没有探索如何为新任务高效微调 VLA,而这正是推广应用的关键一环。针对这些挑战,我们推出 OpenVLA:一个 70 亿参数的开源 VLA,在 97 万条多样化的真实世界机器人示教上训练。OpenVLA 以 Llama 2 语言模型为基础,结合一个融合了 DINOv2 与 SigLIP 预训练特征的视觉编码器。得益于更丰富的数据多样性和新的模型组件,OpenVLA 在通用操作上表现出色:在 29 个任务和多种机器人本体上,其绝对任务成功率比 RT-2-X(550 亿参数)等封闭模型高 16.5%,而参数量只有后者的七分之一。我们进一步表明,OpenVLA 可以被有效地微调到新的场景,在涉及多个物体的多任务环境中泛化效果尤其好,语言落地(language grounding)能力很强,并且比 Diffusion Policy 等表达力强的从零模仿学习方法高出 20.4%。我们还探讨了计算效率:作为一项独立贡献,我们表明 OpenVLA 可以借助现代低秩适配方法在消费级 GPU 上微调,并通过量化高效部署,而不损失下游成功率。最后,我们发布了模型检查点、微调用的 notebook,以及我们的 PyTorch 代码库;该代码库内置支持在 Open X-Embodiment 数据集上大规模训练 VLA。

Figure 1

Figure 1: We present OpenVLA, a 7B-parameter open-source vision-language-action model (VLA), trained on 970k robot episodes from the Open X-Embodiment dataset [1]. OpenVLA sets a new state of the art for generalist robot manipulation policies. It supports controlling multiple robots out of the box and can be quickly adapted to new robot domains via parameter-efficient fine-tuning. The OpenVLA checkpoints and PyTorch training pipeline are fully open-source and models can be downloaded and fine-tuned from HuggingFace.

图 1:我们推出 OpenVLA,一个 70 亿参数的开源视觉-语言-动作模型(VLA),在 Open X-Embodiment 数据集 [1] 的 97 万个机器人回合上训练。OpenVLA 为通用机器人操作策略树立了新的最高水平。它开箱即可控制多种机器人,并能通过参数高效微调快速适配新的机器人领域。OpenVLA 的检查点和 PyTorch 训练流水线完全开源,模型可以从 HuggingFace 下载并微调。

1 Introduction

1 引言

A key weakness of learned policies for robotic manipulation is their inability to generalize beyond their training data: while existing policies trained for individual skills or language instructions have the capacity to extrapolate behaviors to new initial conditions such as object positions or lighting [2, 3], they lack robustness to scene distractors or novel objects [4, 5] and struggle to execute unseen task instructions [6, 7]. Yet beyond robotics, existing foundation models for vision and language such as CLIP [8], SigLIP [9], and Llama 2 [10] are capable of these types of generalization and more, stemming from the priors captured by their Internet-scale pretraining datasets. While reproducing this scale of pretraining for robotics is still an open challenge — even the largest robot manipulation datasets [1, 11] only have 100K to 1M examples – this imbalance suggests an opportunity: using existing foundation models for vision and language as a core building block for training robotic policies that can generalize to objects, scenes, and tasks beyond their training data.

用于机器人操作的学习型策略有一个关键弱点:无法泛化到训练数据之外。现有的、针对单个技能或语言指令训练的策略,能够把行为外推到新的初始条件,例如物体位置或光照的变化 [2, 3],但它们对场景干扰物或新物体缺乏鲁棒性 [4, 5],也难以执行没见过的任务指令 [6, 7]。而在机器人领域之外,CLIP [8]、SigLIP [9]、Llama 2 [10] 等现有的视觉与语言基础模型具备这类泛化能力乃至更多,这源自它们互联网规模的预训练数据所蕴含的先验。在机器人领域复现这种规模的预训练仍是一个未解决的难题——即便最大的机器人操作数据集 [1, 11] 也只有 10 万到 100 万个样本——但这种不平衡也提示了一个机会:把现有的视觉与语言基础模型作为核心构件,来训练能够泛化到训练数据之外的物体、场景和任务的机器人策略。

Towards this goal, existing work has explored integrating pretrained language and vision-language models for robotic representation learning [12, 13, 14] and as a component in modular systems for task planning and execution [15, 16]. More recently, they have been used for directly learning vision-language-action models [VLAs; 7, 1, 17, 18] for control. VLAs provide a direct instantiation of using pretrained vision-and-language foundation models for robotics, directly fine-tuning visually-conditioned language models (VLMs) such as PaLI [19, 20] to generate robot control actions. By building off of strong foundation models trained on Internet-scale data, VLAs such as RT-2 [7] demonstrate impressive robustness results, as well as an ability to generalize to novel objects and tasks, setting a new standard for generalist robot policies. Yet, there are two key reasons preventing the widespread use of existing VLAs: 1) current models [7, 1, 17, 18] are closed, with limited visibility into model architecture, training procedures, and data mixture, and 2) existing works do not provide best practices for deploying and adapting VLAs to new robots, environments, and tasks — especially on commodity hardware (e.g., consumer-grade GPUs). We argue that to develop a rich foundation for future research and development, robotics needs open-source, generalist VLAs that support effective fine-tuning and adaptation, akin to the existing ecosystem around open-source language models [21, 22, 23, 24].

为此,已有工作探索了把预训练的语言模型和视觉-语言模型用于机器人表征学习 [12, 13, 14],以及作为模块化系统中负责任务规划和执行的组件 [15, 16]。最近,它们又被用来直接学习用于控制的视觉-语言-动作模型 [VLAs; 7, 1, 17, 18]。VLA 是把预训练视觉-语言基础模型用于机器人的一种直接实现:直接微调 PaLI [19, 20] 等视觉条件语言模型(VLM),让它生成机器人控制动作。借助在互联网规模数据上训练的强大基础模型,RT-2 [7] 等 VLA 表现出令人印象深刻的鲁棒性,并能泛化到新的物体和任务,为通用机器人策略树立了新标准。然而,有两个关键原因阻碍了现有 VLA 的广泛使用:1)当前的模型 [7, 1, 17, 18] 是封闭的,模型架构、训练流程和数据配比都不透明;2)现有工作没有给出把 VLA 部署并适配到新机器人、新环境和新任务的最佳实践——尤其是在普通硬件(例如消费级 GPU)上。我们认为,要为未来的研究与开发打下丰厚的基础,机器人领域需要支持有效微调与适配的开源通用 VLA,就像围绕开源语言模型已经形成的生态 [21, 22, 23, 24] 那样。

To this end, we introduce OpenVLA, a 7B-parameter open-source VLA that establishes a new state of the art for generalist robot manipulation policies.¹ OpenVLA consists of a pretrained visually-conditioned language model backbone that captures visual features at multiple granularities, fine-tuned on a large, diverse dataset of 970k robot manipulation trajectories from the Open-X Embodiment [1] dataset — a dataset that spans a wide range of robot embodiments, tasks, and scenes. As a product of increased data diversity and new model components, OpenVLA outperforms the 55B-parameter RT-2-X model [7, 1], the prior state-of-the-art VLA, by 16.5% absolute success rate across 29 evaluation tasks on the WidowX and Google Robot embodiments. We additionally investigate efficient fine-tuning strategies for VLAs, a new contribution not explored in prior work, across 7 diverse manipulation tasks spanning behaviors from object pick-and-place to cleaning a table. We find that fine-tuned OpenVLA policies clearly outperform fine-tuned pretrained policies such as Octo [5]. Compared to from-scratch imitation learning with diffusion policies [3], fine-tuned OpenVLA shows substantial improvement on tasks involving grounding language to behavior in multi-task settings with multiple objects. Following these results, we are the first to demonstrate the effectiveness of compute-efficient fine-tuning methods leveraging low-rank adaptation [LoRA; 26] and model quantization [27] to facilitate adapting OpenVLA models on consumer-grade GPUs instead of large server nodes without compromising performance. As a final contribution, we open-source all models, deployment and fine-tuning notebooks, and the OpenVLA codebase for training VLAs at scale, with the hope that these resources enable future work exploring and adapting VLAs for robotics.

为此,我们推出 OpenVLA,一个 70 亿参数的开源 VLA,它为通用机器人操作策略树立了新的最高水平。¹ OpenVLA 由一个预训练的视觉条件语言模型骨干组成,能够捕捉多种粒度的视觉特征,并在来自 Open-X Embodiment [1] 数据集的 97 万条大规模、多样化机器人操作轨迹上微调;该数据集涵盖了广泛的机器人本体、任务和场景。得益于更高的数据多样性和新的模型组件,OpenVLA 在 WidowX 和 Google 机器人两种本体的 29 个评测任务上,绝对成功率比此前最先进的 VLA——550 亿参数的 RT-2-X 模型 [7, 1]——高 16.5%。我们还研究了 VLA 的高效微调策略,这是先前工作未曾探索的新贡献,覆盖 7 个多样的操作任务,行为从物体抓放到擦桌子不等。我们发现,微调后的 OpenVLA 策略明显优于微调后的 Octo [5] 等预训练策略。与使用扩散策略 [3] 的从零模仿学习相比,微调后的 OpenVLA 在多物体、多任务、需要把语言落实到行为的任务上有显著提升。在这些结果之后,我们首次证明了利用低秩适配 [LoRA; 26] 和模型量化 [27] 的计算高效微调方法是有效的,使 OpenVLA 模型可以在消费级 GPU 而不是大型服务器节点上完成适配,且不牺牲性能。最后一项贡献是,我们开源了全部模型、部署与微调 notebook,以及用于大规模训练 VLA 的 OpenVLA 代码库,希望这些资源能推动未来探索和适配机器人 VLA 的工作。

¹ OpenVLA uses multiple pretrained model components: SigLIP [9] and DinoV2 [25] vision encoders and a Llama 2 [10] language model backbone. For all three models, weights are open, but not their training data or code. We release training data, code and model weights for reproducing OpenVLA on top of these components.

¹ OpenVLA 使用了多个预训练模型组件:SigLIP [9] 和 DinoV2 [25] 视觉编码器,以及 Llama 2 [10] 语言模型骨干。这三个模型的权重都是开放的,但训练数据和代码没有公开。我们发布了在这些组件之上复现 OpenVLA 所需的训练数据、代码和模型权重。

2 Related Work

2 相关工作

Visually-Conditioned Language Models

Visually-conditioned language models (VLMs), which are trained on Internet-scale data to generate natural language from input image(s) and language prompts, have been adopted for myriad applications from visual question answering [28, 29, 30, 31] to object localization [32, 33]. One of the key advances fueling recent VLMs are model architectures that bridge features from pretrained vision encoders [8, 9, 25] with pretrained language models [10, 23, 34, 35, 36], directly building on advances in both computer vision and natural language modelling to create powerful multimodal models. While early work explored various architectures for cross-attending between vision and language features [37, 38, 39, 40, 41], new open-source VLMs [42, 43, 20, 44] have converged on a simpler “patch-as-token” approach, in which patch features from pretrained visual transformers are treated as tokens, and are then projected into the input space of a language model. This simplicity makes it easy to repurpose existing tools for training language models at scale for VLM training. We employ these tools in our work to scale VLA training, and specifically use VLMs from Karamcheti et al. [44] as our pretrained backbone, as they are trained from multi-resolution visual features, fusing low-level spatial information from DINOv2 [25] with higher-level semantics from SigLIP [9] to aid in visual generalization.

视觉条件语言模型

视觉条件语言模型(VLM)在互联网规模的数据上训练,根据输入图像和语言提示生成自然语言,已被用于从视觉问答 [28, 29, 30, 31] 到物体定位 [32, 33] 的众多应用。推动近期 VLM 发展的关键进展之一,是把预训练视觉编码器 [8, 9, 25] 的特征与预训练语言模型 [10, 23, 34, 35, 36] 连接起来的模型架构,它直接建立在计算机视觉和自然语言建模两方面的进展之上,造就了强大的多模态模型。早期工作探索了多种在视觉与语言特征之间做交叉注意力的架构 [37, 38, 39, 40, 41],而新的开源 VLM [42, 43, 20, 44] 已收敛到一种更简单的“图块即词元(patch-as-token)”方法:把预训练视觉 Transformer 的图块特征当作词元,再投影到语言模型的输入空间。这种简单性使得大规模训练语言模型的现有工具很容易被改用于 VLM 训练。我们在工作中借助这些工具来扩展 VLA 训练,具体采用 Karamcheti 等人 [44] 的 VLM 作为预训练骨干,因为它们基于多分辨率视觉特征训练,融合了 DINOv2 [25] 的低层空间信息和 SigLIP [9] 的高层语义,有助于视觉泛化。

Generalist Robot Policies

A recent trend in robotics works towards training multi-task “generalist” robot policies [45, 46, 47, 6, 2, 48, 49] on large diverse robot datasets [50, 51, 45, 52, 53, 54, 55, 6, 2, 56, 49, 11, 1], spanning many different robot embodiments [57, 53, 58, 59, 60, 61, 62, 63, 64, 65, 1, 5, 66]. Notably, Octo [5] trains a generalist policy that can control multiple robots out-of-the-box and allows for flexible fine-tuning to new robot setups. A key difference between these approaches and OpenVLA is the model architecture. Prior works like Octo typically compose pretrained components such as language embeddings or visual encoders with additional model components initialized from scratch [6, 2, 5], learning to “stitch” them together during the course of policy training. Unlike these works, OpenVLA adopts a more end-to-end approach, directly fine-tuning VLMs to generate robot actions by treating them as tokens in the language model vocabulary. Our experimental evaluation shows that this simple yet scalable pipeline substantially boosts performance and generalization ability over prior generalist policies.

通用机器人策略

机器人领域近来的一个趋势,是在大规模、多样化的机器人数据集 [50, 51, 45, 52, 53, 54, 55, 6, 2, 56, 49, 11, 1] 上训练多任务“通用”机器人策略 [45, 46, 47, 6, 2, 48, 49],覆盖许多不同的机器人本体 [57, 53, 58, 59, 60, 61, 62, 63, 64, 65, 1, 5, 66]。值得注意的是,Octo [5] 训练了一个开箱即可控制多种机器人、并能灵活微调到新机器人配置的通用策略。这些方法与 OpenVLA 的一个关键区别在于模型架构。Octo 等先前工作通常把语言嵌入或视觉编码器等预训练组件,与额外的从零初始化的模型组件组合起来 [6, 2, 5],在策略训练过程中学习把它们“缝合”在一起。与这些工作不同,OpenVLA 采用更端到端的方式:把机器人动作当作语言模型词表中的词元,直接微调 VLM 来生成动作。我们的实验评估表明,这条简单而可扩展的流水线相比先前的通用策略大幅提升了性能和泛化能力。

Vision-Language-Action Models

A number of works have explored the use of VLMs for robotics, e.g., for visual state representations [12, 13], object detection [67], high-level planning [16], and for providing a feedback signal [68, 69, 70, 71]. Others integrate VLMs directly into end-to-end visuomotor manipulation policies [14, 15], but incorporate significant structure into the policy architecture or require calibrated cameras, which limits their applicability. A number of recent works have explored similar recipes to ours and directly fine-tuned large pretrained VLMs for predicting robot actions [7, 1, 72, 73, 17, 18, 74]. Such models are often referred to as vision-language-action models (VLAs), since they fuse robot control actions directly into VLM backbones. This has three key benefits: (1) it performs alignment of pretrained vision and language components on a large, Internet-scale vision-language dataset, (2) the use of a generic architecture, not custom-made for robot control, allows us to leverage the scalable infrastructure underlying modern VLM training [75, 76, 77] and scale to training billion-parameter policies with minimal code modifications, and (3) it provides a direct pathway for robotics to benefit from the rapid improvements in VLMs. Existing works on VLAs either focus on training and evaluating in single robot or simulated setups [72, 73, 78, 74] and thus lack generality, or are closed and do not support efficient fine-tuning to new robot setups [7, 1, 17, 18].

视觉-语言-动作模型

许多工作探索了 VLM 在机器人中的用途,例如用于视觉状态表征 [12, 13]、物体检测 [67]、高层规划 [16],以及提供反馈信号 [68, 69, 70, 71]。另一些工作把 VLM 直接整合进端到端的视觉运动操作策略 [14, 15],但在策略架构中引入了大量结构,或者需要标定过的相机,这限制了它们的适用范围。近来有不少工作探索了与我们类似的做法,直接微调大型预训练 VLM 来预测机器人动作 [7, 1, 72, 73, 17, 18, 74]。这类模型通常被称为视觉-语言-动作模型(VLA),因为它们把机器人控制动作直接融合进 VLM 骨干。这样做有三个关键好处:(1)预训练的视觉和语言组件已在大规模、互联网级的视觉-语言数据集上完成对齐;(2)使用并非为机器人控制专门定制的通用架构,使我们能利用现代 VLM 训练背后的可扩展基础设施 [75, 76, 77],只需极少的代码修改就能扩展到训练数十亿参数的策略;(3)它为机器人领域直接受益于 VLM 的快速进步提供了一条通路。现有的 VLA 工作要么只在单一机器人或仿真环境中训练和评测 [72, 73, 78, 74],因而缺乏通用性;要么是封闭的,不支持高效微调到新的机器人配置 [7, 1, 17, 18]。

Most closely related, RT-2-X [1] trains a 55B-parameter VLA policy on the Open X-Embodiment dataset and demonstrates state-of-the-art generalist manipulation policy performance. However, our work differs from RT-2-X in multiple important aspects: (1) by combining a strong open VLM backbone with a richer robot pretraining dataset, OpenVLA outperforms RT-2-X in our experiments while being an order of magnitude smaller; (2) we thoroughly investigate fine-tuning of OpenVLA models to new target setups, while RT-2-X does not investigate the fine-tuning setting; (3) we are the first to demonstrate the effectiveness of modern parameter-efficient fine-tuning and quantization approaches for VLAs; and (4) OpenVLA is the first generalist VLA that is open-source and thus supports future research on VLA training, data mixtures, objectives, and inference.

与我们最相近的是 RT-2-X [1],它在 Open X-Embodiment 数据集上训练了一个 550 亿参数的 VLA 策略,展示了最先进的通用操作策略性能。不过,我们的工作在几个重要方面与 RT-2-X 不同:(1)通过把强大的开放 VLM 骨干与更丰富的机器人预训练数据集结合,OpenVLA 在我们的实验中优于 RT-2-X,而规模小了一个数量级;(2)我们深入研究了把 OpenVLA 模型微调到新目标配置的问题,而 RT-2-X 没有研究微调场景;(3)我们首次证明了现代参数高效微调和量化方法对 VLA 的有效性;(4)OpenVLA 是第一个开源的通用 VLA,因此能支持未来关于 VLA 训练、数据配比、目标函数和推理的研究。

3 The OpenVLA Model

3 OpenVLA 模型

We introduce the OpenVLA model, a 7B-parameter vision-language-action model (VLA) trained on 970k robot demonstrations from the Open X-Embodiment dataset [1]. There are many, largely unexplored, questions around best practices for developing VLA models, e.g., what are the best model backbones, datasets, and hyperparameters to use for training. Below, we detail our approach for developing OpenVLA and summarize our key learnings. Concretely, we first provide a brief overview of modern VLMs, which form the backbone of OpenVLA (Section 3.1); then describe our basic training recipe and dataset (Section 3.2 and Section 3.3); discuss key design decisions (Section 3.4); and provide details of the used infrastructure for training and inference (Section 3.5).

我们介绍 OpenVLA 模型:一个在 Open X-Embodiment 数据集 [1] 的 97 万条机器人示教上训练的 70 亿参数视觉-语言-动作模型(VLA)。关于开发 VLA 模型的最佳实践,还有许多基本未被探索的问题,例如训练时最好用什么模型骨干、数据集和超参数。下面我们详细介绍开发 OpenVLA 的方法,并总结关键经验。具体来说,我们先简要概述构成 OpenVLA 骨干的现代 VLM(第 3.1 节),然后介绍基本训练方案和数据集(第 3.2 节和第 3.3 节),讨论关键的设计决策(第 3.4 节),最后给出训练与推理所用基础设施的细节(第 3.5 节)。

Figure 2

Figure 2: OpenVLA model architecture. Given an image observation and a language instruction, the model predicts 7-dimensional robot control actions. The architecture consists of three key components: (1) a vision encoder that concatenates Dino V2 [25] and SigLIP [79] features, (2) a projector that maps visual features to the language embedding space, and (3) the LLM backbone, a Llama 2 7B-parameter large language model [10].

图 2:OpenVLA 模型架构。给定一张图像观测和一条语言指令,模型预测 7 维的机器人控制动作。该架构由三个关键组件组成:(1)一个视觉编码器,把 Dino V2 [25] 和 SigLIP [79] 的特征拼接起来;(2)一个投影器,把视觉特征映射到语言嵌入空间;(3)LLM 骨干,即 70 亿参数的大语言模型 Llama 2 [10]。

3.1 Preliminaries: Vision-Language Models

3.1 预备知识:视觉-语言模型

The architecture of most recent VLMs [42, 43, 20, 44] consists of three main parts (see Fig. 2): (1) a visual encoder that maps image inputs to a number of “image patch embeddings”, (2) a projector that takes the output embeddings of the visual encoder and maps them into the input space of a language model, and (3) a large language model (LLM) backbone. During VLM training, the model is trained end-to-end with a next text token prediction objective on paired or interleaved vision and language data curated from various Internet sources.

近期大多数 VLM [42, 43, 20, 44] 的架构由三个主要部分组成(见图 2):(1)一个视觉编码器,把输入图像映射为若干“图像图块嵌入”;(2)一个投影器,接收视觉编码器的输出嵌入并把它们映射到语言模型的输入空间;(3)一个大语言模型(LLM)骨干。在 VLM 训练中,模型以预测下一个文本词元为目标,在从各种互联网来源整理的成对或交错的视觉-语言数据上端到端训练。

In this work, we build on the Prismatic-7B VLM [44]. Prismatic follows the same standard architecture described above, with a 600M-parameter visual encoder, a small 2-layer MLP projector, and a 7B-parameter Llama 2 language model backbone [10]. Notably, Prismatic uses a two-part visual encoder, consisting of pretrained SigLIP [79] and DinoV2 [25] models. Input image patches are passed separately through both encoders and the resulting feature vectors are concatenated channel-wise. In contrast to the more commonly used vision encoders such as CLIP- [80] or SigLIP-only encoders, the addition of DinoV2 features has been shown to be helpful for improved spatial reasoning [44], which can be particularly helpful for robot control.

在这项工作中,我们以 Prismatic-7B VLM [44] 为基础。Prismatic 采用上述标准架构,包含一个 6 亿参数的视觉编码器、一个小型两层 MLP 投影器,以及一个 70 亿参数的 Llama 2 语言模型骨干 [10]。值得注意的是,Prismatic 使用两部分组成的视觉编码器,由预训练的 SigLIP [79] 和 DinoV2 [25] 模型构成。输入图块分别经过两个编码器,得到的特征向量按通道拼接。与更常用的 CLIP [80] 或仅 SigLIP 的视觉编码器相比,加入 DinoV2 特征已被证明有助于提升空间推理能力 [44],这对机器人控制可能尤其有帮助。

SigLIP, DinoV2, and Llama 2 do not release details about their training data, which likely consists of trillions of tokens of Internet-sourced image-text, image-only, and text-only data respectively. The Prismatic VLM is fine-tuned on top of these components using the LLaVA 1.5 data mixture [43], which contains a total of approximately 1M image-text and text-only data samples from open-source datasets [81, 82, 29, 83, 42].

SigLIP、DinoV2 和 Llama 2 都没有公布训练数据的细节,这些数据很可能分别是数万亿词元的互联网图文数据、纯图像数据和纯文本数据。Prismatic VLM 在这些组件之上用 LLaVA 1.5 的数据配比 [43] 进行微调,该配比总共包含约 100 万个来自开源数据集 [81, 82, 29, 83, 42] 的图文和纯文本样本。

3.2 OpenVLA Training Procedure

3.2 OpenVLA 训练流程

To train OpenVLA, we fine-tune a pretrained Prismatic-7B VLM backbone for robot action prediction (see Fig. 2). We formulate the action prediction problem as a “vision-language” task, where an input observation image and a natural language task instruction are mapped to a string of predicted robot actions [7]. To enable the VLM’s language model backbone to predict robot actions, we represent the actions in the output space of the LLM by mapping continuous robot actions to discrete tokens used by the language model’s tokenizer. Following Brohan et al. [7], we discretize each dimension of the robot actions separately into one of 256 bins. For each action dimension, we set the bin width to uniformly divide the interval between the 1st and 99th quantile of the actions in the training data. Using quantiles instead of the min-max bounds Brohan et al. [7] used allows us to ignore outlier actions in the data that could otherwise drastically expand the discretization interval and reduce the effective granularity of our action discretization.

为训练 OpenVLA,我们微调一个预训练的 Prismatic-7B VLM 骨干来预测机器人动作(见图 2)。我们把动作预测问题表述为一个“视觉-语言”任务:把输入的观测图像和自然语言任务指令映射为一串预测的机器人动作 [7]。为了让 VLM 的语言模型骨干能够预测机器人动作,我们把连续的机器人动作映射为语言模型分词器所用的离散词元,从而在 LLM 的输出空间中表示动作。沿用 Brohan 等人 [7] 的做法,我们把机器人动作的每一维分别离散化到 256 个区间(bin)之一。对每个动作维度,区间宽度设为把训练数据中动作的第 1 和第 99 百分位数之间的区间均匀等分。用分位数而不是 Brohan 等人 [7] 所用的最小-最大边界,可以忽略数据中的离群动作;否则这些离群值可能大幅撑大离散化区间,降低动作离散化的有效粒度。

Using this discretization, we obtain N discrete integers ∈[0…255]\in[0\dots 255]∈[0…255] for an NNN-dimensional robot action. Unfortunately, the tokenizer used by OpenVLA’s language backbone, the Llama tokenizer [10], only reserves 100 “special tokens” for tokens newly introduced during fine-tuning, which is too few for the 256 tokens of our action discretization. Instead, we again opt for simplicity and follow Brohan et al. [7]’s approach by simply overwriting the 256 least used tokens in the Llama tokenizer’s vocabulary (which corresponds to the last 256 tokens) with our action tokens. Once the actions are processed into a sequence of tokens, OpenVLA is trained with a standard next-token prediction objective, evaluating the cross-entropy loss on the predicted action tokens only. We discuss key design decisions for implementing this training procedure in Section 3.4. Next, we describe the robot dataset we use for OpenVLA training.

通过这种离散化,对一个 NNN 维的机器人动作,我们得到 N 个离散整数 ∈[0…255]\in[0\dots 255]∈[0…255]。遗憾的是,OpenVLA 语言骨干所用的分词器——Llama 分词器 [10]——只为微调时新引入的词元预留了 100 个“特殊词元”,不够容纳我们动作离散化所需的 256 个词元。于是我们再次选择简单的方案,沿用 Brohan 等人 [7] 的做法,直接用动作词元覆盖 Llama 分词器词表中最少使用的 256 个词元(即词表最后 256 个词元)。动作被处理成词元序列后,OpenVLA 用标准的下一词元预测目标训练,只在预测的动作词元上计算交叉熵损失。我们在第 3.4 节讨论实现这一训练流程的关键设计决策。接下来介绍 OpenVLA 训练所用的机器人数据集。

3.3 Training Data

3.3 训练数据

The goal in constructing the OpenVLA training dataset is to capture a large diversity of robot embodiments, scenes, and tasks. This enables the final model to control various robots out of the box and admits efficient fine-tuning to new robot setups. We leverage the Open X-Embodiment dataset [1] (OpenX) as a base to curate our training dataset. The full OpenX dataset, at the time of writing, consists of more than 70 individual robot datasets, with more than 2M robot trajectories, that were pooled into a coherent and easy-to-use data format in a large community effort. To make training on this data practical, we apply multiple steps of data curation to the raw dataset.

构建 OpenVLA 训练数据集的目标,是覆盖大量多样的机器人本体、场景和任务。这样最终模型既能开箱控制多种机器人,又能高效微调到新的机器人配置。我们以 Open X-Embodiment 数据集 [1](OpenX)为基础来整理训练数据集。在撰写本文时,完整的 OpenX 数据集包含 70 多个单独的机器人数据集、超过 200 万条机器人轨迹,这些数据经过一项大型社区合作被汇集成统一、易用的数据格式。为了让在这些数据上训练变得可行,我们对原始数据集做了多步数据整理。

The goals of this curation are to ensure (1) a coherent input and output space across all training datasets, and (2) a balanced mix of embodiments, tasks, and scenes in the final training mixture.² To address (1), we follow [1, 5] and restrict our training dataset to contain only manipulation datasets with at least one 3rd person camera and use single-arm end-effector control. For (2), we leverage the data mixture weights of Octo [5] for all datasets that pass the first round of filtering. Octo heuristically down-weights or removes less diverse datasets and up-weights datasets with larger task and scene diversity; see Octo Model Team et al. [5] for details.

整理的目标是保证:(1)所有训练数据集的输入和输出空间一致;(2)最终训练配比中本体、任务和场景的平衡。² 针对(1),我们沿用 [1, 5] 的做法,只保留至少有一个第三人称相机、使用单臂末端执行器控制的操作数据集。针对(2),对所有通过第一轮筛选的数据集,我们沿用 Octo [5] 的数据配比权重。Octo 启发式地降低或移除多样性较低的数据集的权重,提高任务和场景多样性较大的数据集的权重;细节见 Octo Model Team 等人 [5]。

² Octo [5] demonstrated training across datasets with heterogeneous sensory inputs. While very promising, we leave an investigation of VLA training across heterogeneous sensor modalities and action spaces to future work.

² Octo [5] 展示了在传感输入异构的数据集之间进行训练。这虽然很有前景,但我们把跨异构传感模态和动作空间的 VLA 训练研究留待未来工作。

We also experimented with incorporating a few additional datasets into our training mixture that were added to the OpenX dataset since the release of Octo, including the DROID dataset [11], although at a conservative mixture weight of 10%. In practice, we found that the action token accuracy on DROID remained low throughout training, suggesting a larger mixture weight or model may be required to fit its diversity in the future. To not jeopardize the quality of the final model, we removed DROID from the data mixture for the final third of training. We provide a complete overview of the used datasets and mixture weights in Appendix A.

我们还尝试把 Octo 发布后才加入 OpenX 的几个数据集纳入训练配比,其中包括 DROID 数据集 [11],不过只给了 10% 这一保守的配比权重。实际中我们发现,DROID 上的动作词元准确率在整个训练过程中都很低,这表明将来可能需要更大的配比权重或更大的模型才能拟合它的多样性。为了不影响最终模型的质量,我们在训练的最后三分之一阶段把 DROID 从数据配比中移除了。所用数据集和配比权重的完整概览见附录 A。

3.4 OpenVLA Design Decisions

3.4 OpenVLA 的设计决策

When developing the OpenVLA model, we explored various design decisions in smaller-scale experiments before starting the final model training run. Concretely, we trained and evaluated OpenVLA models on BridgeData V2 [6] for our initial experiments, instead of training on the full OpenX mixture, to increase iteration speed and reduce computational cost. We summarize key learnings from these explorations below.

在开发 OpenVLA 模型时,我们先在较小规模的实验中探索了各种设计决策,再开始最终的模型训练。具体来说,初期实验中我们在 BridgeData V2 [6] 上训练和评测 OpenVLA 模型,而不是在完整的 OpenX 配比上训练,以加快迭代速度、降低计算成本。下面总结这些探索得到的关键经验。

VLM Backbone. Initially, we experimented with multiple VLM backbones. Apart from Prismatic [44], we tested fine-tuning IDEFICS-1 [84] and LLaVA [85] for robot action prediction. We found that LLaVA and IDEFICS-1 performed comparably on tasks with only one object in the scene, but LLaVA demonstrated stronger language grounding in tasks that involved multiple objects in the scene and required the policy to manipulate the correct object, i.e., the object specified in the language instruction. Concretely, LLaVA improved upon IDEFICS-1 by 35% in absolute success rate, averaged across five language grounding tasks in a BridgeData V2 sink environment. The fine-tuned Prismatic VLM policy achieved further improvements, outperforming the LLaVA policy by roughly 10% in absolute success rate across both simple single-object tasks and multi-object, language grounding tasks. We attribute this performance delta to improved spatial reasoning capabilities afforded by the fused SigLIP-DinoV2 backbones (see Section 3.1). In addition to the performance enhancements, Prismatic also provides a modular and easy-to-use codebase, so we ultimately chose it to be the backbone for the OpenVLA model.

VLM 骨干。起初我们试验了多种 VLM 骨干。除 Prismatic [44] 外,我们还测试了微调 IDEFICS-1 [84] 和 LLaVA [85] 来预测机器人动作。我们发现,在场景中只有一个物体的任务上,LLaVA 和 IDEFICS-1 表现相当;但在场景中有多个物体、需要策略操作正确物体(即语言指令中指定的物体)的任务上,LLaVA 表现出更强的语言落地能力。具体而言,在 BridgeData V2 水槽环境的五个语言落地任务上取平均,LLaVA 的绝对成功率比 IDEFICS-1 高 35%。微调后的 Prismatic VLM 策略又有进一步提升,在简单的单物体任务和多物体语言落地任务上,绝对成功率都比 LLaVA 策略高出约 10%。我们把这一性能差距归因于融合 SigLIP-DinoV2 骨干带来的更强空间推理能力(见第 3.1 节)。除了性能提升,Prismatic 还提供了模块化、易用的代码库,因此我们最终选它作为 OpenVLA 模型的骨干。

Image Resolution. The resolution of input images has significant impact on the computational requirements of VLA training, since higher-resolution images result in more image patch tokens and thus longer context lengths that quadratically increase training compute. We compared VLAs with 224×224224\times 224224×224px and 384×384384\times 384384×384px inputs, but found no performance difference in our evaluations, while the latter takes 3x longer to train. We thus opt for a resolution of 224×224224\times 224224×224px for the final OpenVLA model. Note that on many VLM benchmarks, increased resolution does improve performance [44, 86, 87], but we did not see this trend (yet) for VLAs.

图像分辨率。输入图像的分辨率对 VLA 训练的计算需求影响很大,因为更高分辨率的图像会产生更多图块词元,从而带来更长的上下文长度,而训练计算量随上下文长度呈平方增长。我们比较了 224×224224\times 224224×224px 和 384×384384\times 384384×384px 输入的 VLA,在我们的评测中没有发现性能差异,而后者的训练时间是前者的 3 倍。因此最终的 OpenVLA 模型采用 224×224224\times 224224×224px 的分辨率。需要注意的是,在许多 VLM 基准上,提高分辨率确实能提升性能 [44, 86, 87],但我们(目前)在 VLA 上没有看到这一趋势。

Fine-Tuning Vision Encoder. Prior work on VLMs found that freezing vision encoders during VLM training typically leads to higher performance [44]. Intuitively, a frozen vision encoder may better preserve the robust features learned from its Internet-scale pretraining. However, we found fine-tuning the vision encoder during VLA training to be crucial for good VLA performance. We hypothesize that the pretrained vision backbone may not capture sufficient fine-grained spatial details about important parts of the scene to enable precise robotic control.

微调视觉编码器。先前关于 VLM 的工作发现,VLM 训练时冻结视觉编码器通常能带来更高的性能 [44]。直观上,冻结的视觉编码器可能更好地保留其在互联网规模预训练中学到的鲁棒特征。然而我们发现,在 VLA 训练中微调视觉编码器对取得良好的 VLA 性能至关重要。我们推测,预训练的视觉骨干可能没有捕捉到场景关键部位足够细粒度的空间细节,不足以支撑精确的机器人控制。

Training Epochs. Typical LLM or VLM training runs complete at most one or two epochs through their training dataset. In contrast, we found it important for VLA training to iterate through the training dataset significantly more times, with real robot performance continually increasing until training action token accuracy surpasses 95%. Our final training run completes 27 epochs through its training dataset.

训练轮数。典型的 LLM 或 VLM 训练最多在训练数据集上跑一到两个轮次(epoch)。相比之下,我们发现 VLA 训练需要在训练数据集上迭代多得多的次数:真实机器人上的性能会持续提高,直到训练动作词元准确率超过 95%。我们最终的训练在训练数据集上完成了 27 个轮次。

Learning Rate. We swept the learning rate across multiple orders of magnitude for VLA training, and achieved the best results using a fixed learning rate of 2e-5 (the same learning rate used during VLM pretraining [44]). We did not find learning rate warmup to provide benefits.

学习率。我们在多个数量级范围内扫描了 VLA 训练的学习率,使用 2e-5 的固定学习率(与 VLM 预训练 [44] 时相同)取得了最好的结果。我们没有发现学习率预热(warmup)带来好处。

3.5 Infrastructure for Training and Inference

3.5 训练与推理基础设施

The final OpenVLA model is trained on a cluster of 64 A100 GPUs for 14 days, or a total of 21,500 A100-hours, using a batch size of 2048. During inference, OpenVLA requires 15GB of GPU memory when loaded in bfloat16 precision (i.e., without quantization) and runs at approximately 6Hz on one NVIDIA RTX 4090 GPU (without compilation, speculative decoding, or other inference speed-up tricks). We can further reduce the memory footprint of OpenVLA during inference via quantization, without compromising performance in real-world robotics tasks, as shown in Section 5.4. We report inference speed on various consumer- and server-grade GPUs in Fig. 6. For convenience, we implement a remote VLA inference server to allow real-time remote streaming of action predictions to the robot – removing the requirement of having access to a powerful local compute device to control the robot. We release this remote inference solution as part of our open-source code release (Section 4).

最终的 OpenVLA 模型在 64 块 A100 GPU 组成的集群上训练了 14 天,共计 21,500 A100 小时,批大小为 2048。推理时,OpenVLA 以 bfloat16 精度加载(即不量化)需要 15GB 显存,在一块 NVIDIA RTX 4090 GPU 上的运行速度约为 6Hz(未使用编译、推测解码或其他推理加速技巧)。如第 5.4 节所示,我们还可以通过量化进一步降低 OpenVLA 推理时的显存占用,而不影响真实机器人任务上的性能。图 6 报告了在各种消费级和服务器级 GPU 上的推理速度。为方便起见,我们实现了一个远程 VLA 推理服务器,可以把动作预测实时远程流式传给机器人——这样控制机器人时就不必在本地配备强大的计算设备。我们把这一远程推理方案作为开源代码的一部分发布(第 4 节)。

4 The OpenVLA Codebase

4 OpenVLA 代码库

Along with our model, we release the OpenVLA codebase, a modular PyTorch codebase for training VLA models (see https://openvla.github.io). It scales from fine-tuning VLAs on individual GPUs to training billion-parameter VLAs on multi-node GPU clusters, and supports modern techniques for large transformer model training such as automatic mixed precision (AMP, PyTorch [75]), FlashAttention [76], and fully sharded data parallelism (FSDP, Zhao et al. [77]). Out of the box, the OpenVLA codebase has full support for training on the Open X dataset, integrates with HuggingFace’s [21] AutoModel class, and supports LoRA fine-tuning [26] and quantized model inference [88, 27].

随模型一起,我们发布了 OpenVLA 代码库,一个用于训练 VLA 模型的模块化 PyTorch 代码库(见 https://openvla.github.io)。它既能在单块 GPU 上微调 VLA,也能扩展到在多节点 GPU 集群上训练数十亿参数的 VLA,并支持大型 Transformer 模型训练的现代技术,例如自动混合精度(AMP,PyTorch [75])、FlashAttention [76] 和全分片数据并行(FSDP,Zhao 等人 [77])。OpenVLA 代码库开箱即完整支持在 Open X 数据集上训练,与 HuggingFace [21] 的 AutoModel 类集成,并支持 LoRA 微调 [26] 和量化模型推理 [88, 27]。

5 Experiments

5 实验

The goal of our experimental evaluations is to test OpenVLA’s ability to serve as a powerful multi-robot control policy out of the box, as well as be a good initialization for fine-tuning to new robot tasks. Concretely, we aim to answer the following questions:

我们实验评估的目标,是检验 OpenVLA 能否开箱即用地充当强大的多机器人控制策略,以及能否作为微调到新机器人任务的良好初始化。具体来说,我们希望回答以下问题:

  1. How does OpenVLA compare to prior generalist robot policies, when evaluating on multiple robots and various types of generalization?
  2. Can OpenVLA be effectively fine-tuned on a new robot setup and task, and how does it compare to state-of-the-art data-efficient imitation learning approaches?
  3. Can we use parameter-efficient fine-tuning and quantization to reduce the computational requirements for training and inference of OpenVLA models and make them more accessible? What are the performance-compute trade-offs?
  1. 在多种机器人和多类泛化上评测时,OpenVLA 与先前的通用机器人策略相比表现如何?
  2. OpenVLA 能否有效地微调到新的机器人配置和任务上?与最先进的数据高效模仿学习方法相比如何?
  3. 能否用参数高效微调和量化来降低 OpenVLA 模型训练和推理的计算需求,让它更容易使用?性能与计算之间的权衡是怎样的?

5.1 Direct Evaluations on Multiple Robot Platforms

5.1 在多个机器人平台上的直接评测

Figure 3

Figure 3: BridgeData V2 WidowX robot evaluation tasks and results. We evaluate OpenVLA and prior state-of-the-art generalist robot policies on a comprehensive suite of tasks covering several axes of generalization, as well as tasks that specifically assess language conditioning ability. OpenVLA achieves highest overall performance and even outperforms closed-source model RT-2-X in all categories except for semantic generalization. Average success rates ±\pm± StdErr are computed across 170 total rollouts per approach. See Table 4 for detailed results.

图 3:BridgeData V2 WidowX 机器人评测任务与结果。我们在一套全面的任务上评测 OpenVLA 和先前最先进的通用机器人策略,这些任务覆盖多个泛化维度,另有专门评估语言条件能力的任务。OpenVLA 取得最高的总体性能,除语义泛化外,在所有类别上都超过了闭源模型 RT-2-X。平均成功率 ±\pm± 标准误基于每种方法共 170 次试验计算。详细结果见表 4。

Robot Setups and Tasks. We evaluate OpenVLA’s performance “out-of-the-box” on two robot embodiments: the WidowX robot from the BridgeData V2 evaluations [6] (see Fig. 1, left) and the mobile manipulation robot from the RT-1 and RT-2 evaluations [2, 7] (“Google robot”; see Fig. 1, middle). Both platforms have been extensively used in prior works for evaluating generalist robot policies [2, 7, 1, 5]. We define a comprehensive set of evaluation tasks in each environment that covers various axes of generalization, such as visual (unseen backgrounds, distractor objects, colors/appearances of objects); motion (unseen object positions/orientations); physical (unseen object sizes/shapes); and semantic (unseen target objects, instructions, and concepts from the Internet) generalization. We also assess language conditioning ability in scenes with multiple objects, testing whether the policy can manipulate the correct target object, as specified in the user’s prompt. See bottom row of Fig. 3 and Fig. 4 for example task images in the BridgeData V2 and Google robot evaluations, respectively. Overall, we evaluated each method in 170 rollouts (17 tasks with 10 trials each) for BridgeData V2 experiments and 60 rollouts (12 tasks with 5 trials each) for Google robot experiments. A detailed breakdown of all tasks and how they differ from the training data is in Appendix B. All evaluations in this and the following sections are conducted as A/B evaluations, using the same tasks with the same sets of initial robot and object states, to ensure fair comparison.

机器人配置与任务。我们在两种机器人本体上评测 OpenVLA“开箱即用”的性能:BridgeData V2 评测 [6] 中的 WidowX 机器人(见图 1 左),以及 RT-1 和 RT-2 评测 [2, 7] 中的移动操作机器人(“Google 机器人”;见图 1 中)。两个平台都在先前工作中被广泛用于评测通用机器人策略 [2, 7, 1, 5]。我们在每个环境中定义了一套全面的评测任务,覆盖多个泛化维度,例如视觉泛化(未见过的背景、干扰物体、物体颜色/外观)、运动泛化(未见过的物体位置/朝向)、物理泛化(未见过的物体尺寸/形状)和语义泛化(未见过的目标物体、指令以及来自互联网的概念)。我们还在多物体场景中评估语言条件能力,测试策略能否按用户提示操作正确的目标物体。BridgeData V2 和 Google 机器人评测的示例任务图像分别见图 3 和图 4 的底行。总的来说,BridgeData V2 实验中每种方法评测 170 次(17 个任务,每个 10 次),Google 机器人实验中评测 60 次(12 个任务,每个 5 次)。所有任务的详细说明以及它们与训练数据的差异见附录 B。本节及后续各节的所有评测都以 A/B 方式进行,使用相同的任务和相同的机器人与物体初始状态集合,以保证比较公平。

Comparisons. We compare OpenVLA’s performance to three prior generalist manipulation policies: RT-1-X [1], RT-2-X [1], and Octo [5]. RT-1-X (35M parameters) and Octo (93M parameters) are transformer policies trained from scratch on subsets of the OpenX dataset; Octo is the state-of-the-art model among open-source manipulation policies. RT-2-X (55B parameters) is a state-of-the-art, closed-source VLA that leverages Internet-pretrained vision and language backbones.

对比方法。我们把 OpenVLA 的性能与三种先前的通用操作策略比较:RT-1-X [1]、RT-2-X [1] 和 Octo [5]。RT-1-X(3500 万参数)和 Octo(9300 万参数)是在 OpenX 数据集子集上从零训练的 Transformer 策略;Octo 是开源操作策略中最先进的模型。RT-2-X(550 亿参数)是最先进的闭源 VLA,利用了互联网预训练的视觉和语言骨干。

The results are summarized in Fig. 3 for BridgeData V2 evaluations and Fig. 4 for Google robot evaluations (per-task breakdown in Appendix, Table 4 and Table 6). We find that both RT-1-X and Octo struggle on the tested tasks, often failing to manipulate the correct object, especially when distractors are present, and in some cases causing the robot to wave its arm around aimlessly. Note that our evaluations test even larger degrees of generalization than the evaluations performed in those prior works to challenge the Internet-pretrained VLA models. Thus, lower performance of models without Internet pretraining is expected. RT-2-X clearly outperforms both RT-1-X and Octo, demonstrating the benefits of large, pretrained VLMs for robotics.

BridgeData V2 评测的结果汇总在图 3,Google 机器人评测的结果汇总在图 4(逐任务细分见附录表 4 和表 6)。我们发现 RT-1-X 和 Octo 在所测任务上都表现吃力,常常操作错物体,尤其在有干扰物时;有时还会让机器人漫无目的地挥动手臂。需要注意,为了考验互联网预训练的 VLA 模型,我们的评测所测试的泛化程度比那些先前工作中的评测更大。因此,没有互联网预训练的模型表现较差是意料之中的。RT-2-X 明显优于 RT-1-X 和 Octo,展示了大型预训练 VLM 对机器人的好处。

Figure 4

Figure 4: Google robot evaluation results. We evaluate generalist robot policies on in-distribution and out-of-distribution (OOD) tasks on the mobile manipulator used in RT-1 and RT-2 evaluations [2, 7]. We find that OpenVLA and RT-2-X attain comparable performance and significantly outperform RT-1-X and Octo overall. Average success rates ±\pm± StdErr are computed across 60 total rollouts per approach. See Table 6 for detailed results.

图 4:Google 机器人评测结果。我们在 RT-1 和 RT-2 评测 [2, 7] 所用的移动操作机器人上,用分布内和分布外(OOD)任务评测通用机器人策略。我们发现 OpenVLA 和 RT-2-X 性能相当,总体上显著优于 RT-1-X 和 Octo。平均成功率 ±\pm± 标准误基于每种方法共 60 次试验计算。详细结果见表 6。

Notably, OpenVLA performs comparably to RT-2-X on Google robot evaluations and significantly outperforms RT-2-X on BridgeData V2 evaluations despite being an order of magnitude smaller (7B vs. 55B parameters). Qualitatively, we find that both RT-2-X and OpenVLA exhibit markedly more robust behaviors than the other tested models, such as approaching the correct object when distractor objects are present, properly orienting the robot’s end-effector to align with the orientation of the target object, and even recovering from mistakes such as insecurely grasping objects (see https://openvla.github.io for qualitative rollout examples). RT-2-X achieves higher performance in semantic generalization tasks, as shown in Fig. 3, which is expected given that it uses larger-scale Internet pretraining data and is co-fine-tuned with both robot action data and Internet pretraining data to better preserve the pretraining knowledge, rather than being fine-tuned solely on robot data, like OpenVLA. However, OpenVLA performs comparably or better in all other task categories in both BridgeData V2 and Google robot evaluations. The performance difference can be attributed to a combination of factors: we curated a much larger training dataset for OpenVLA with 970k trajectories (vs. 350k for RT-2-X); we performed more careful cleaning of the training dataset and, e.g., filtered out all-zero actions in the Bridge dataset (see Appendix C for a detailed discussion); and OpenVLA uses a fused vision encoder that combines pretrained semantic and spatial features. See Appendix D for ablation analyses of these components.

值得注意的是,OpenVLA 在 Google 机器人评测上与 RT-2-X 表现相当,在 BridgeData V2 评测上则显著优于 RT-2-X,尽管规模小了一个数量级(70 亿对 550 亿参数)。定性地看,我们发现 RT-2-X 和 OpenVLA 都比其他被测模型表现出明显更鲁棒的行为,例如在有干扰物时仍能接近正确的物体,正确调整机器人末端执行器的朝向以对齐目标物体的朝向,甚至能从没抓稳物体之类的错误中恢复(定性的试验示例见 https://openvla.github.io)。如图 3 所示,RT-2-X 在语义泛化任务上性能更高,这在意料之中:它使用了更大规模的互联网预训练数据,并且同时用机器人动作数据和互联网预训练数据做联合微调,以更好地保留预训练知识,而不是像 OpenVLA 那样只在机器人数据上微调。不过,在 BridgeData V2 和 Google 机器人评测的所有其他任务类别上,OpenVLA 的表现都相当或更好。这一性能差异可以归因于几方面因素的结合:我们为 OpenVLA 整理了大得多的训练数据集,包含 97 万条轨迹(RT-2-X 为 35 万条);我们对训练数据集做了更仔细的清洗,例如过滤掉了 Bridge 数据集中的全零动作(详细讨论见附录 C);此外 OpenVLA 使用融合视觉编码器,结合了预训练的语义特征和空间特征。这些组件的消融分析见附录 D。

5.2 Data-Efficient Adaptation to New Robot Setups

5.2 数据高效地适配新的机器人配置

While prior works mainly focused on directly evaluating VLAs “out-of-the-box” [16, 7, 1], effective fine-tuning of VLA models to new tasks and robot setups is largely unexplored, yet is key for their widespread adoption. In this section, we investigate OpenVLA’s ability to be quickly adapted to a new real-world robot setup. (See Appendix E for fine-tuning experiments in simulation.)

先前工作主要关注直接“开箱即用”地评测 VLA [16, 7, 1],而把 VLA 模型有效地微调到新任务和新机器人配置上,基本没有被研究过,但这恰恰是 VLA 得到广泛应用的关键。本节我们研究 OpenVLA 能否快速适配到新的真实世界机器人配置。(仿真中的微调实验见附录 E。)

Robot setups and tasks. We test a simple fine-tuning recipe for the OpenVLA model: full fine-tuning of all model parameters, using small datasets with 10–150 demonstrations of a target task (see Fig. 5; we explore parameter-efficient fine-tuning approaches in Section 5.3). We test OpenVLA in two setups: Franka-Tabletop, a stationary, table-mounted Franka Emika Panda 7-DoF robot arm; and Franka-DROID, the Franka robot arm setup from the recently released DROID dataset [11], mounted on a movable standing desk. The setups use 5Hz and 15 Hz non-blocking controllers, respectively. We choose Franka robot arms as the target embodiment for our fine-tuning experiments since they are widely used in the robot learning community and thus a likely “target” of OpenVLA fine-tuning. We test on setups with different control frequencies to test OpenVLA’s applicability to a range of use cases.

机器人配置与任务。我们为 OpenVLA 模型测试一种简单的微调方案:全参数微调,使用目标任务的 10–150 条示教组成的小数据集(见图 5;参数高效微调方法在第 5.3 节探讨)。我们在两种配置中测试 OpenVLA:Franka-Tabletop,一台固定安装在桌上的 Franka Emika Panda 7 自由度机械臂;以及 Franka-DROID,即最近发布的 DROID 数据集 [11] 中的 Franka 机械臂配置,安装在可移动的升降桌上。两种配置分别使用 5Hz 和 15Hz 的非阻塞控制器。我们选择 Franka 机械臂作为微调实验的目标本体,因为它在机器人学习社区中被广泛使用,因而很可能成为 OpenVLA 微调的“目标”。我们在控制频率不同的配置上测试,以检验 OpenVLA 对多种使用场景的适用性。

Figure 5

Figure 5: Adapting to new robot setups. We evaluate the state-of-the-art Diffusion Policy trained from scratch on seven Franka Emika Panda tasks (10–150 demonstrations each), as well as generalist robot policies Octo and OpenVLA fine-tuned on the same data. Diffusion Policy exhibits strong performance on narrow single-instruction tasks, while Octo and OpenVLA perform better on diverse fine-tuning tasks involving multiple instructions and distractor objects. Overall, OpenVLA achieves highest aggregate performance across both setups, suggesting that it is an effective default for learning a policy on a downstream task. Average success rates ±\pm± StdErr are computed across 129 rollouts per approach (99 for Franka-Tabletop tasks and 30 for Franka-DROID tasks). See Table 7 for detailed results.

图 5:适配新的机器人配置。我们在七个 Franka Emika Panda 任务(每个 10–150 条示教)上评测从零训练的最先进 Diffusion Policy,以及在相同数据上微调的通用机器人策略 Octo 和 OpenVLA。Diffusion Policy 在狭窄的单指令任务上表现强劲,而 Octo 和 OpenVLA 在涉及多条指令和干扰物的多样化微调任务上表现更好。总体而言,OpenVLA 在两种配置上的综合性能最高,表明它是在下游任务上学习策略的有效默认选择。平均成功率 ±\pm± 标准误基于每种方法 129 次试验计算(Franka-Tabletop 任务 99 次,Franka-DROID 任务 30 次)。详细结果见表 7。

Comparisons. We compare to Diffusion Policy [3], a state-of-the-art data-efficient imitation learning approach, trained from scratch. We also compare to Diffusion Policy (matched), a version of Diffusion Policy that matches the input and output specifications of OpenVLA.³ Additionally, we evaluate Octo [5] fine-tuned on the target dataset, since it is currently the best generalist policy that supports fine-tuning (fine-tuning of RT-2-X is not supported through its inference API). We also fine-tune OpenVLA on the same target dataset, and the resulting policy is denoted by OpenVLA. Finally, as an ablation experiment, we compare to OpenVLA (scratch), where we directly fine-tune the underlying base Prismatic VLM on the target robot setup – rather than fine-tuning the OpenX-pretrained OpenVLA model – to assess the benefit of large-scale robot pretraining.

对比方法。我们与从零训练的 Diffusion Policy [3] 比较,它是最先进的数据高效模仿学习方法。我们还与 Diffusion Policy (matched) 比较,这是一个输入和输出规格与 OpenVLA 一致的 Diffusion Policy 版本。³ 此外,我们评测在目标数据集上微调的 Octo [5],因为它是目前支持微调的最佳通用策略(RT-2-X 的推理 API 不支持微调)。我们也在同一目标数据集上微调 OpenVLA,得到的策略记作 OpenVLA。最后,作为消融实验,我们与 OpenVLA (scratch) 比较:直接在目标机器人配置上微调底层的 Prismatic 基础 VLM,而不是微调经过 OpenX 预训练的 OpenVLA 模型,以评估大规模机器人预训练的好处。

³ The full Diffusion Policy uses a two-step observation history with both images and proprioceptive state, and performs receding horizon control by predicting a chunk of TTT future actions and executing the first XXX actions in open-loop fashion before predicting the next chunk (for 15Hz control, we set T=16,X=8T=16,X=8T=16,X=8 like in the DROID prior work [11]; for 5Hz control, we reduce the chunk sizes to T=8,X=3T=8,X=3T=8,X=3). It is also the only method in Section 5.2 that predicts absolute Cartesian coordinates to control the robot; all other methods use relative position control. Diffusion Policy (matched) uses a single image as input, has no proprioceptive information and no observation history, and predicts a single relative position control action without action chunking.

³ 完整的 Diffusion Policy 使用包含图像和本体感知状态的两步观测历史,并进行滚动时域控制:预测一段 TTT 个未来动作,以开环方式执行前 XXX 个动作,再预测下一段(15Hz 控制时,我们像 DROID 的先前工作 [11] 那样设 T=16,X=8T=16,X=8T=16,X=8;5Hz 控制时,把分块大小减为 T=8,X=3T=8,X=3T=8,X=3)。它也是第 5.2 节中唯一预测绝对笛卡尔坐标来控制机器人的方法;其他所有方法都使用相对位置控制。Diffusion Policy (matched) 以单张图像为输入,没有本体感知信息和观测历史,并且不做动作分块,只预测单个相对位置控制动作。

We present the results in Fig. 5 (per-task breakdown in Appendix, Table 7). We find that both versions of Diffusion Policy are competitive with or outperform the generalist policies Octo and OpenVLA on narrower single-instruction tasks like “Put Carrot in Bowl” and “Pour Corn into Pot”, but the pretrained generalist policies perform better in more diverse fine-tuning tasks that involve multiple objects in the scene and require language conditioning. OpenX pretraining for Octo and OpenVLA enables the models to better adapt to these more diverse tasks where language grounding is important; we see evidence for this in the lower performance of OpenVLA (scratch).

结果见图 5(逐任务细分见附录表 7)。我们发现,在“把胡萝卜放进碗里”“把玉米倒进锅里”这类较窄的单指令任务上,两个版本的 Diffusion Policy 都与通用策略 Octo 和 OpenVLA 相当或更好;但在场景中有多个物体、需要语言条件的更多样化的微调任务上,预训练的通用策略表现更好。OpenX 预训练使 Octo 和 OpenVLA 能更好地适应这些语言落地很重要的多样化任务;OpenVLA (scratch) 较低的性能为此提供了证据。

Overall, we find that OpenVLA achieves the highest average performance. Notably, most prior works achieve strong performance only in either narrow single-instruction or diverse multi-instruction tasks, resulting in widely varying success rates. OpenVLA is the only approach that achieves at least 50% success rate across all tested tasks, suggesting that it can be a strong default option for imitation learning tasks, particularly if they involve a diverse set of language instructions. For narrower but highly dexterous tasks, Diffusion Policy still shows smoother and more precise trajectories; incorporating action chunking and temporal smoothing, as implemented in Diffusion Policy, may help OpenVLA attain the same level of dexterity and may be a promising direction for future work (see Section 6 for a detailed discussion of current limitations).

总体而言,我们发现 OpenVLA 取得了最高的平均性能。值得注意的是,大多数先前工作要么只在狭窄的单指令任务上、要么只在多样化的多指令任务上表现强劲,导致成功率波动很大。OpenVLA 是唯一在所有被测任务上成功率都至少达到 50% 的方法,表明它可以作为模仿学习任务的强有力默认选择,尤其是当任务涉及多样的语言指令时。对于较窄但对灵巧性要求很高的任务,Diffusion Policy 仍然表现出更平滑、更精确的轨迹;引入 Diffusion Policy 中实现的动作分块(action chunking)和时间平滑,可能帮助 OpenVLA 达到同样的灵巧水平,这可能是未来工作的一个有前景的方向(当前局限的详细讨论见第 6 节)。

5.3 Parameter-Efficient Fine-Tuning

5.3 参数高效微调

The full fine-tuning runs of OpenVLA in the previous section used 8 A100 GPUs for 5-15 hours per task (depending on the dataset size) to achieve high performance. While this is substantially less compute than what is required for VLA pretraining, in this section we explore even more compute- and parameter-efficient fine-tuning approaches and investigate their effectiveness.

上一节中 OpenVLA 的全参数微调,每个任务用 8 块 A100 GPU 训练 5–15 小时(取决于数据集大小)才达到高性能。虽然这比 VLA 预训练所需的计算少得多,但本节我们探索计算和参数上更高效的微调方法,并考察它们的效果。

Strategy Success Rate Train Params (×106\times 10^{6}×106) VRAM (batch 16)
Full FT 69.7 ±\pm± 7.2 % 7,188.1 163.3 GB*
Last layer only 30.3 ±\pm± 6.1 % 465.1 51.4 GB
Frozen vision 47.0 ±\pm± 6.9 % 6,760.4 156.2 GB*
Sandwich 62.1 ±\pm± 7.9 % 914.2 64.0 GB
LoRA, rank=32 68.2 ±\pm± 7.5% 97.6 59.7 GB
rank=64 68.2 ±\pm± 7.8% 195.2 60.5 GB

Table 1: Parameter-efficient fine-tuning evaluation. LoRA fine-tuning achieves the best performance-compute trade-off, matching full fine-tuning performance while training only 1.4% of the model parameters. Mean success ±\pm± StdErr computed across 33 rollouts per approach on select Franka-Tabletop tasks (see Table 8 for details). ∗: Sharded across 2 GPUs with FSDP [77].

策略 成功率 训练参数量(×106\times 10^{6}×106) 显存(批大小 16)
全参数微调 69.7 ±\pm± 7.2 % 7,188.1 163.3 GB*
仅最后一层 30.3 ±\pm± 6.1 % 465.1 51.4 GB
冻结视觉编码器 47.0 ±\pm± 6.9 % 6,760.4 156.2 GB*
三明治微调 62.1 ±\pm± 7.9 % 914.2 64.0 GB
LoRA,秩=32 68.2 ±\pm± 7.5% 97.6 59.7 GB
秩=64 68.2 ±\pm± 7.8% 195.2 60.5 GB

表 1:参数高效微调评测。LoRA 微调取得了最好的性能-计算权衡,只训练 1.4% 的模型参数就达到了全参数微调的性能。平均成功率 ±\pm± 标准误基于每种方法在选定的 Franka-Tabletop 任务上的 33 次试验计算(详见表 8)。∗:用 FSDP [77] 分片到 2 块 GPU 上。

Concretely, we compare the following fine-tuning approaches: full fine-tuning updates all weights during fine-tuning, as described in Section 5.2; last layer only fine-tunes only the last layer of OpenVLA’s transformer backbone and the token embedding matrix; frozen vision freezes the vision encoder but fine-tunes all other weights; sandwich fine-tuning unfreezes the vision encoder, token embedding matrix, and last layer; and LoRA uses the popular low-rank adaptation technique of Hu et al. [26] with multiple rank values rrr, applied to all linear layers of the model.

具体来说,我们比较以下微调方法:全参数微调在微调时更新全部权重,如第 5.2 节所述;仅最后一层只微调 OpenVLA Transformer 骨干的最后一层和词元嵌入矩阵;冻结视觉编码器冻结视觉编码器、微调其余所有权重;三明治微调解冻视觉编码器、词元嵌入矩阵和最后一层;LoRA 使用 Hu 等人 [26] 提出的流行低秩适配技术,取多个秩值 rrr,作用于模型的所有线性层。

We report fine-tuning success rates across multiple Franka-Tabletop tasks, as well as training parameter count and GPU memory requirements, in Table 1.⁴ We find that only fine-tuning the network’s last layer or freezing the vision encoder leads to poor performance, suggesting that further adaptation of the visual features to the target scene is crucial. In contrast, “sandwich fine-tuning” achieves better performance since it fine-tunes the vision encoder, and it consumes less GPU memory since it does not fine-tune the full LLM backbone. Lastly, LoRA achieves the best trade-off between performance and training memory consumption, outperforming “sandwich fine-tuning” and matching full fine-tuning performance while fine-tuning only 1.4% of the parameters. We find that the LoRA rank has negligible effect on policy performance and thus recommend using a default rank of r=32r=32r=32. With LoRA, we can fine-tune OpenVLA on a new task within 10-15 hours on a single A100 GPU – an 8x reduction in compute compared to full fine-tuning.

表 1 报告了多个 Franka-Tabletop 任务上的微调成功率,以及训练参数量和 GPU 显存需求。⁴ 我们发现,只微调网络最后一层或冻结视觉编码器都会导致性能很差,说明让视觉特征进一步适应目标场景至关重要。相比之下,“三明治微调”因为微调了视觉编码器而取得更好的性能,又因为没有微调完整的 LLM 骨干而占用更少的 GPU 显存。最后,LoRA 在性能和训练显存占用之间取得了最好的权衡,优于“三明治微调”,并且只微调 1.4% 的参数就达到了全参数微调的性能。我们发现 LoRA 的秩对策略性能的影响可以忽略,因此建议默认使用 r=32r=32r=32。借助 LoRA,我们可以在单块 A100 GPU 上于 10–15 小时内把 OpenVLA 微调到新任务——计算量比全参数微调减少了 8 倍。

⁴ In Section 5.3 and Section 5.4, we experiment with a version of the OpenVLA model that is pretrained with a smaller robot data mixture (the same OpenX dataset mixture as Octo) and has a slightly smaller architecture which only uses a SigLIP [79] vision backbone instead of the fused DinoSigLIP encoder. We find that this simpler architecture still achieves strong performance in both fine-tuning tasks and “out-of-the-box” tasks.

⁴ 在第 5.3 节和第 5.4 节中,我们实验的是 OpenVLA 模型的一个版本:它用更小的机器人数据配比(与 Octo 相同的 OpenX 数据配比)预训练,架构也略小,只使用 SigLIP [79] 视觉骨干,而不是融合的 DinoSigLIP 编码器。我们发现这种更简单的架构在微调任务和“开箱即用”任务上仍有强劲的表现。

5.4 Memory-Efficient Inference via Quantization

5.4 通过量化实现省显存的推理

Figure 6

Figure 6: OpenVLA inference speed for various GPUs. Both bfloat16 and int4 quantization achieve high throughput, especially on GPUs with Ada Lovelace architecture (RTX 4090, H100). Further speed-ups are possible with modern LLM inference frameworks like TensorRT-LLM [89]. ♠\spadesuit♠: Model sharded across two GPUs to fit.

图 6:OpenVLA 在各种 GPU 上的推理速度。bfloat16 和 int4 量化都能达到很高的吞吐,在 Ada Lovelace 架构的 GPU(RTX 4090、H100)上尤其如此。借助 TensorRT-LLM [89] 等现代 LLM 推理框架还能进一步加速。♠\spadesuit♠:模型需分片到两块 GPU 上才能装下。

Precision Bridge Success VRAM
bfloat16 71.3 ±\pm± 4.8% 16.8 GB
int8 58.1 ±\pm± 5.1% 10.2 GB
int4 71.9 ±\pm± 4.7% 7.0 GB

Table 2: Performance with quantized inference. 4-bit quantization matches the performance of bfloat16 inference (our default approach) while reducing the GPU memory footprint by more than half. Mean success ±\pm± StdErr computed across 8 representative BridgeData V2 tasks [6] and 80 rollouts per approach (see Table 5 for details).

精度 Bridge 成功率 显存
bfloat16 71.3 ±\pm± 4.8% 16.8 GB
int8 58.1 ±\pm± 5.1% 10.2 GB
int4 71.9 ±\pm± 4.7% 7.0 GB

表 2:量化推理的性能。4 比特量化与 bfloat16 推理(我们的默认方式)性能相当,同时把 GPU 显存占用减少了一半以上。平均成功率 ±\pm± 标准误基于 8 个有代表性的 BridgeData V2 任务 [6]、每种方法 80 次试验计算(详见表 5)。

OpenVLA, a 7B-parameter model, consumes more memory at inference time than prior open-source generalist policies such as Octo, which has <<<100M parameters. We follow best-practices from LLM serving by saving and loading OpenVLA in bfloat16 precision for inference (our default approach), which cuts the memory footprint in half, allowing us to serve OpenVLA on GPUs with only 16GB of GPU memory. In this section, we test whether we can further reduce the required memory for policy inference and broaden accessibility of VLA policies, by using modern quantization techniques developed for serving LLMs [88, 27]. These approaches load the weights of the network at lower precision, thereby trading off reduced memory requirements for potentially reduced inference speed and accuracy.

OpenVLA 是一个 70 亿参数的模型,推理时比 Octo(参数不到 1 亿)等先前的开源通用策略占用更多显存。我们遵循 LLM 部署的最佳实践,以 bfloat16 精度保存和加载 OpenVLA 进行推理(我们的默认方式),这把显存占用减半,使我们能在只有 16GB 显存的 GPU 上部署 OpenVLA。本节我们测试能否利用为部署 LLM 而开发的现代量化技术 [88, 27],进一步降低策略推理所需的显存,扩大 VLA 策略的可及性。这些方法以更低的精度加载网络权重,用可能降低的推理速度和精度换取更低的显存需求。

Concretely, we investigate serving the OpenVLA model with 8-bit and 4-bit precision on 8 representative BridgeData V2 tasks. We report memory footprint and rollout performance in Table 2. We also report achievable control frequencies on various consumer- and server-grade GPUs in Fig. 6. We observe that 8-bit quantization slows down inference across most GPUs, due to the overhead of the added quantization operations. 4-bit inference achieves higher throughput, since reduced GPU memory transfer compensates for the quantization overhead.

具体来说,我们在 8 个有代表性的 BridgeData V2 任务上研究以 8 比特和 4 比特精度部署 OpenVLA 模型。表 2 报告了显存占用和试验性能。图 6 还报告了在各种消费级和服务器级 GPU 上可达到的控制频率。我们观察到,由于新增量化操作的开销,8 比特量化在大多数 GPU 上会拖慢推理。4 比特推理的吞吐更高,因为 GPU 显存传输量的减少抵消了量化开销。

As a result of the reduced inference speed, we observe a substantial performance decrease with 8-bit quantization: on the A5000 GPU we use for our evaluations, we can only run the model at 1.2Hz, which significantly changes the system dynamics compared to the training dataset for the 5Hz non-blocking controller used in the BridgeData V2 tasks.⁵ Notably, 4-bit quantization results in similar performance as bfloat16 half-precision inference despite requiring less than half the amount of GPU memory. 4-bit quantized models can run at 3Hz on the A5000, thus more closely matching the system dynamics during data collection.

由于推理速度降低,我们观察到 8 比特量化导致性能大幅下降:在我们评测所用的 A5000 GPU 上,模型只能以 1.2Hz 运行,与 BridgeData V2 任务所用的 5Hz 非阻塞控制器下采集的训练数据相比,系统动力学发生了显著变化。⁵ 值得注意的是,4 比特量化的性能与 bfloat16 半精度推理相近,而所需 GPU 显存不到后者的一半。4 比特量化模型在 A5000 上能以 3Hz 运行,因而更接近数据采集时的系统动力学。

⁵ We attribute the performance loss to low inference speed, since both 8-bit and 4-bit quantization achieve comparable token accuracy to bfloat16 inference when evaluated offline on training data. See Section D.4 for supporting details.

⁵ 我们把性能损失归因于推理速度低,因为在训练数据上离线评估时,8 比特和 4 比特量化的词元准确率都与 bfloat16 推理相当。支持细节见 D.4 节。

6 Discussion and Limitations

6 讨论与局限

In this work, we presented OpenVLA, a state-of-the-art, open-source vision-language-action model that obtains strong performance for cross-embodiment robot control out-of-the-box. We also demonstrated that OpenVLA can be easily adapted to new robot setups via parameter-efficient fine-tuning techniques.

在这项工作中,我们推出了 OpenVLA,一个最先进的开源视觉-语言-动作模型,开箱即可在跨本体机器人控制上取得强劲性能。我们还证明了 OpenVLA 可以借助参数高效微调技术轻松适配到新的机器人配置。

The current OpenVLA model has several limitations. First, it currently only supports single-image observations. In reality, real-world robot setups are heterogeneous, with a wide range of possible sensory inputs [5]. Expanding OpenVLA to support multiple image and proprioceptive inputs as well as observation history is an important avenue for future work. Exploring the use of VLMs pretrained on interleaved image and text data may facilitate such flexible-input VLA fine-tuning.

当前的 OpenVLA 模型有若干局限。首先,它目前只支持单张图像观测。而现实中的机器人配置是异构的,可能的传感输入范围很广 [5]。扩展 OpenVLA 以支持多张图像、本体感知输入以及观测历史,是未来工作的重要方向。探索在交错图文数据上预训练的 VLM,可能有助于实现这种输入灵活的 VLA 微调。

Secondly, improving the inference throughput of OpenVLA is critical to enable VLA control for high-frequency control setups such as ALOHA [90], which runs at 50Hz. This will also enable testing VLAs on more dexterous, bi-manual manipulation tasks than what we investigated in this work. Exploring the use of action chunking or alternative inference-time optimization techniques such as speculative decoding [91] offer potential remedies.

其次,提高 OpenVLA 的推理吞吐,对于在 ALOHA [90](以 50Hz 运行)这类高频控制配置上实现 VLA 控制至关重要。这也将使我们能在比本文所研究的更灵巧的双臂操作任务上测试 VLA。探索动作分块或推测解码 [91] 等其他推理时优化技术,提供了可能的解决办法。

Additionally, there is room for further performance improvements. While OpenVLA outperforms prior generalist policies, it does not yet offer very high reliability on the tested tasks, typically achieving <90% success rate.

此外,性能还有进一步提升的空间。虽然 OpenVLA 优于先前的通用策略,但它在所测任务上的可靠性还不够高,成功率通常低于 90%。

Finally, due to compute limitations, many VLA design questions remain underexplored: What effect does the size of the base VLM have on VLA performance? Does co-training on robot action prediction data and Internet-scale vision-language data substantially improve VLA performance? What visual features are best-suited for VLA models? We hope that the release of the OpenVLA model and codebase will enable the community to jointly investigate these questions.

最后,由于算力限制,许多 VLA 设计问题仍未被充分探索:基础 VLM 的规模对 VLA 性能有什么影响?在机器人动作预测数据和互联网规模视觉-语言数据上联合训练,能否显著提升 VLA 性能?哪些视觉特征最适合 VLA 模型?我们希望 OpenVLA 模型和代码库的发布能让社区共同研究这些问题。

Acknowledgments

We are grateful to the Toyota Research Institute for providing significant funding and compute resources required to carry out this research. We also thank the Stanford Center for Research on Foundation Models for providing additional compute resources and Google DeepMind for alpha access to the RT-2-X API for our evaluations. We acknowledge additional support from Volkswagen, Physical Intelligence, ONR grants N00014-22-1-2621 and N00014-22-1-2293, the National Science Foundation through IIS-2246811, and DARPA ANSR.

致谢

感谢丰田研究院(Toyota Research Institute)为开展本研究提供了大量资金和计算资源。我们还感谢斯坦福基础模型研究中心(Stanford Center for Research on Foundation Models)提供额外的计算资源,以及 Google DeepMind 为我们的评测提供 RT-2-X API 的 alpha 访问权限。我们感谢大众汽车(Volkswagen)、Physical Intelligence、ONR 资助 N00014-22-1-2621 和 N00014-22-1-2293、美国国家科学基金会 IIS-2246811 以及 DARPA ANSR 提供的额外支持。

References

参考文献

[1] Open X-Embodiment Collaboration, A. Padalkar, A. Pooley, A. Jain, A. Bewley, A. Herzog, A. Irpan, A. Khazatsky, A. Rai, A. Singh, A. Brohan, A. Raffin, A. Wahid, B. Burgess-Limerick, B. Kim, B. Schölkopf, B. Ichter, C. Lu, C. Xu, C. Finn, C. Xu, C. Chi, C. Huang, C. Chan, C. Pan, C. Fu, C. Devin, D. Driess, D. Pathak, D. Shah, D. Büchler, D. Kalashnikov, D. Sadigh, E. Johns, F. Ceola, F. Xia, F. Stulp, G. Zhou, G. S. Sukhatme, G. Salhotra, G. Yan, G. Schiavi, H. Su, H.-S. Fang, H. Shi, H. B. Amor, H. I. Christensen, H. Furuta, H. Walke, H. Fang, I. Mordatch, I. Radosavovic, I. Leal, J. Liang, J. Kim, J. Schneider, J. Hsu, J. Bohg, J. Bingham, J. Wu, J. Wu, J. Luo, J. Gu, J. Tan, J. Oh, J. Malik, J. Tompson, J. Yang, J. J. Lim, J. Silvério, J. Han, K. Rao, K. Pertsch, K. Hausman, K. Go, K. Gopalakrishnan, K. Goldberg, K. Byrne, K. Oslund, K. Kawaharazuka, K. Zhang, K. Majd, K. Rana, K. Srinivasan, L. Y. Chen, L. Pinto, L. Tan, L. Ott, L. Lee, M. Tomizuka, M. Du, M. Ahn, M. Zhang, M. Ding, M. K. Srirama, M. Sharma, M. J. Kim, N. Kanazawa, N. Hansen, N. Heess, N. J. Joshi, N. Suenderhauf, N. D. Palo, N. M. M. Shafiullah, O. Mees, O. Kroemer, P. R. Sanketi, P. Wohlhart, P. Xu, P. Sermanet, P. Sundaresan, Q. Vuong, R. Rafailov, R. Tian, R. Doshi, R. Martín-Martín, R. Mendonca, R. Shah, R. Hoque, R. Julian, S. Bustamante, S. Kirmani, S. Levine, S. Moore, S. Bahl, S. Dass, S. Song, S. Xu, S. Haldar, S. Adebola, S. Guist, S. Nasiriany, S. Schaal, S. Welker, S. Tian, S. Dasari, S. Belkhale, T. Osa, T. Harada, T. Matsushima, T. Xiao, T. Yu, T. Ding, T. Davchev, T. Z. Zhao, T. Armstrong, T. Darrell, V. Jain, V. Vanhoucke, W. Zhan, W. Zhou, W. Burgard, X. Chen, X. Wang, X. Zhu, X. Li, Y. Lu, Y. Chebotar, Y. Zhou, Y. Zhu, Y. Xu, Y. Wang, Y. Bisk, Y. Cho, Y. Lee, Y. Cui, Y. hua Wu, Y. Tang, Y. Zhu, Y. Li, Y. Iwasawa, Y. Matsuo, Z. Xu, and Z. J. Cui. Open X-Embodiment: Robotic learning datasets and RT-X models. https://arxiv.org/abs/2310.08864 , 2023.

[2] A. Brohan, N. Brown, J. Carbajal, Y. Chebotar, J. Dabis, C. Finn, K. Gopalakrishnan, K. Hausman, A. Herzog, J. Hsu, J. Ibarz, B. Ichter, A. Irpan, T. Jackson, S. Jesmonth, N. Joshi, R. Julian, D. Kalashnikov, Y. Kuang, I. Leal, K.-H. Lee, S. Levine, Y. Lu, U. Malla, D. Manjunath, I. Mordatch, O. Nachum, C. Parada, J. Peralta, E. Perez, K. Pertsch, J. Quiambao, K. Rao, M. Ryoo, G. Salazar, P. Sanketi, K. Sayed, J. Singh, S. Sontakke, A. Stone, C. Tan, H. Tran, V. Vanhoucke, S. Vega, Q. Vuong, F. Xia, T. Xiao, P. Xu, S. Xu, T. Yu, and B. Zitkovich. Rt-1: Robotics transformer for real-world control at scale. In arXiv preprint arXiv:2212.06817 , 2022.

[3] C. Chi, S. Feng, Y. Du, Z. Xu, E. Cousineau, B. Burchfiel, and S. Song. Diffusion policy: Visuomotor policy learning via action diffusion. In Proceedings of Robotics: Science and Systems (RSS) , 2023.

[4] A. Xie, L. Lee, T. Xiao, and C. Finn. Decomposing the generalization gap in imitation learning for visual robotic manipulation. arXiv preprint arXiv:2307.03659 , 2023.

[5] Octo Model Team, D. Ghosh, H. Walke, K. Pertsch, K. Black, O. Mees, S. Dasari, J. Hejna, C. Xu, J. Luo, T. Kreiman, Y. Tan, D. Sadigh, C. Finn, and S. Levine. Octo: An open-source generalist robot policy. https://octo-models.github.io , 2023.

[6] H. Walke, K. Black, A. Lee, M. J. Kim, M. Du, C. Zheng, T. Zhao, P. Hansen-Estruch, Q. Vuong, A. He, V. Myers, K. Fang, C. Finn, and S. Levine. Bridgedata v2: A dataset for robot learning at scale, 2023.

[7] A. Brohan, N. Brown, J. Carbajal, Y. Chebotar, X. Chen, K. Choromanski, T. Ding, D. Driess, A. Dubey, C. Finn, P. Florence, C. Fu, M. G. Arenas, K. Gopalakrishnan, K. Han, K. Hausman, A. Herzog, J. Hsu, B. Ichter, A. Irpan, N. Joshi, R. Julian, D. Kalashnikov, Y. Kuang, I. Leal, L. Lee, T.-W. E. Lee, S. Levine, Y. Lu, H. Michalewski, I. Mordatch, K. Pertsch, K. Rao, K. Reymann, M. Ryoo, G. Salazar, P. Sanketi, P. Sermanet, J. Singh, A. Singh, R. Soricut, H. Tran, V. Vanhoucke, Q. Vuong, A. Wahid, S. Welker, P. Wohlhart, J. Wu, F. Xia, T. Xiao, P. Xu, S. Xu, T. Yu, and B. Zitkovich. Rt-2: Vision-language-action models transfer web knowledge to robotic control. In arXiv preprint arXiv:2307.15818 , 2023.

[8] A. Radford, J. W. Kim, C. Hallacy, A. Ramesh, G. Goh, S. Agarwal, G. Sastry, A. Askell, P. Mishkin, J. Clark, G. Krueger, and I. Sutskever. Learning transferable visual models from natural language supervision. In International Conference on Machine Learning (ICML) , volume 139, pages 8748–8763, 2021.

[9] X. Zhai, B. Mustafa, A. Kolesnikov, and L. Beyer. Sigmoid loss for language image pre-training. In International Conference on Computer Vision (ICCV) , 2023.

[10] H. Touvron, L. Martin, K. Stone, P. Albert, A. Almahairi, Y. Babaei, N. Bashlykov, S. Batra, P. Bhargava, S. Bhosale, et al. Llama 2: Open foundation and fine-tuned chat models. arXiv preprint arXiv:2307.09288 , 2023.

[11] A. Khazatsky, K. Pertsch, S. Nair, A. Balakrishna, S. Dasari, S. Karamcheti, S. Nasiriany, M. K. Srirama, L. Y. Chen, K. Ellis, P. D. Fagan, J. Hejna, M. Itkina, M. Lepert, Y. J. Ma, P. T. Miller, J. Wu, S. Belkhale, S. Dass, H. Ha, A. Jain, A. Lee, Y. Lee, M. Memmel, S. Park, I. Radosavovic, K. Wang, A. Zhan, K. Black, C. Chi, K. B. Hatch, S. Lin, J. Lu, J. Mercat, A. Rehman, P. R. Sanketi, A. Sharma, C. Simpson, Q. Vuong, H. R. Walke, B. Wulfe, T. Xiao, J. H. Yang, A. Yavary, T. Z. Zhao, C. Agia, R. Baijal, M. G. Castro, D. Chen, Q. Chen, T. Chung, J. Drake, E. P. Foster, J. Gao, D. A. Herrera, M. Heo, K. Hsu, J. Hu, D. Jackson, C. Le, Y. Li, K. Lin, R. Lin, Z. Ma, A. Maddukuri, S. Mirchandani, D. Morton, T. Nguyen, A. O’Neill, R. Scalise, D. Seale, V. Son, S. Tian, E. Tran, A. E. Wang, Y. Wu, A. Xie, J. Yang, P. Yin, Y. Zhang, O. Bastani, G. Berseth, J. Bohg, K. Goldberg, A. Gupta, A. Gupta, D. Jayaraman, J. J. Lim, J. Malik, R. Martín-Martín, S. Ramamoorthy, D. Sadigh, S. Song, J. Wu, M. C. Yip, Y. Zhu, T. Kollar, S. Levine, and C. Finn. Droid: A large-scale in-the-wild robot manipulation dataset. 2024.

[12] S. Nair, A. Rajeswaran, V. Kumar, C. Finn, and A. Gupta. R3m: A universal visual representation for robot manipulation. In CoRL , 2022.

[13] S. Karamcheti, S. Nair, A. S. Chen, T. Kollar, C. Finn, D. Sadigh, and P. Liang. Language-driven representation learning for robotics. ArXiv , abs/2302.12766, 2023. URL https://api.semanticscholar.org/CorpusID:257205716 .

[14] M. Shridhar, L. Manuelli, and D. Fox. Cliport: What and where pathways for robotic manipulation. In Conference on robot learning , pages 894–906. PMLR, 2022.

[15] A. Stone, T. Xiao, Y. Lu, K. Gopalakrishnan, K.-H. Lee, Q. Vuong, P. Wohlhart, B. Zitkovich, F. Xia, C. Finn, et al. Open-world object manipulation using pre-trained vision-language models. arXiv preprint arXiv:2303.00905 , 2023.

[16] D. Driess, F. Xia, M. S. Sajjadi, C. Lynch, A. Chowdhery, B. Ichter, A. Wahid, J. Tompson, Q. Vuong, T. Yu, et al. Palm-e: An embodied multimodal language model. arXiv preprint arXiv:2303.03378 , 2023.

[17] A. S. et al. Introducing rfm-1: Giving robots human-like reasoning capabilities, 2024. URL https://covariant.ai/insights/introducing-rfm-1-giving-robots-human-like-reasoning-capabilities/ .

[18] Wayve. Lingo-2: Driving with natural language. 2024. URL https://wayve.ai/thinking/lingo-2-driving-with-language/ .

[19] X. Chen, X. Wang, S. Changpinyo, A. J. Piergiovanni, P. Padlewski, D. M. Salz, S. Goodman, A. Grycner, B. Mustafa, L. Beyer, A. Kolesnikov, J. Puigcerver, N. Ding, K. Rong, H. Akbari, G. Mishra, L. Xue, A. V. Thapliyal, J. Bradbury, W. Kuo, M. Seyedhosseini, C. Jia, B. K. Ayan, C. Riquelme, A. Steiner, A. Angelova, X. Zhai, N. Houlsby, and R. Soricut. Pali: A jointly-scaled multilingual language-image model. ArXiv , abs/2209.06794, 2022. URL https://api.semanticscholar.org/CorpusID:252222320 .

[20] X. Chen, X. Wang, L. Beyer, A. Kolesnikov, J. Wu, P. Voigtlaender, B. Mustafa, S. Goodman, I. M. Alabdulmohsin, P. Padlewski, D. M. Salz, X. Xiong, D. Vlasic, F. Pavetic, K. Rong, T. Yu, D. Keysers, X.-Q. Zhai, and R. Soricut. PaLI-3 vision language models: Smaller, faster, stronger. arXiv preprint arXiv:2310.09199 , 2023.

[21] T. Wolf, L. Debut, V. Sanh, J. Chaumond, C. Delangue, A. Moi, P. Cistac, T. Rault, R. Louf, M. Funtowicz, J. Davison, S. Shleifer, and … Transformers: State-of-the-art natural language processing. In Proceedings of the 6th International Conference on Learning Representations , 2020. URL https://arxiv.org/abs/1910.03771 .

[22] H. Touvron, T. Lavril, G. Izacard, X. Martinet, M.-A. Lachaux, T. Lacroix, B. Rozière, N. Goyal, E. Hambro, F. Azhar, et al. Llama: Open and efficient foundation language models. arXiv preprint arXiv:2302.13971 , 2023.

[23] A. Q. Jiang, A. Sablayrolles, A. Mensch, C. Bamford, D. S. Chaplot, D. d. l. Casas, F. Bressand, G. Lengyel, G. Lample, L. Saulnier, et al. Mistral 7b. arXiv preprint arXiv:2310.06825 , 2023.

[24] G. Team, T. Mesnard, C. Hardin, R. Dadashi, S. Bhupatiraju, S. Pathak, L. Sifre, M. Rivière, M. S. Kale, J. Love, et al. Gemma: Open models based on gemini research and technology. arXiv preprint arXiv:2403.08295 , 2024.

[25] M. Oquab, T. Darcet, T. Moutakanni, H. Vo, M. Szafraniec, V. Khalidov, P. Fernandez, D. Haziza, F. Massa, A. El-Nouby, et al. Dinov2: Learning robust visual features without supervision. arXiv preprint arXiv:2304.07193 , 2023.

[26] E. J. Hu, Y. Shen, P. Wallis, Z. Allen-Zhu, Y. Li, S. Wang, L. Wang, and W. Chen. Lora: Low-rank adaptation of large language models. arXiv preprint arXiv:2106.09685 , 2021.

[27] T. Dettmers, A. Pagnoni, A. Holtzman, and L. Zettlemoyer. Qlora: Efficient finetuning of quantized llms. Advances in Neural Information Processing Systems , 36, 2024.

[28] Y. Goyal, T. Khot, D. Summers-Stay, D. Batra, and D. Parikh. Making the V in VQA matter: Elevating the role of image understanding in visual question answering. In Computer Vision and Pattern Recognition (CVPR) , 2017.

[29] D. A. Hudson and C. D. Manning. GQA: A new dataset for real-world visual reasoning and compositional question answering. In Computer Vision and Pattern Recognition (CVPR) , 2019.

[30] A. Singh, V. Natarajan, M. Shah, Y. Jiang, X. Chen, D. Batra, D. Parikh, and M. Rohrbach. Towards VQA models that can read. In Computer Vision and Pattern Recognition (CVPR) , 2019.

[31] J. P. Bigham, C. Jayant, H. Ji, G. Little, A. Miller, R. C. Miller, R. Miller, A. Tatarowicz, B. White, S. White, and T. Yeh. VizWiz: nearly real-time answers to visual questions. In User Interface Software and Technology (UIST) , pages 333–342, 2010.

[32] S. Kazemzadeh, V. Ordonez, M. Matten, and T. Berg. ReferItGame: Referring to objects in photographs of natural scenes. In Empirical Methods in Natural Language Processing (EMNLP) , pages 787–798, 2014.

[33] L. Yu, P. Poirson, S. Yang, A. C. Berg, and T. L. Berg. Modeling context in referring expressions. In European Conference on Computer Vision (ECCV) , 2016.

[34] T. Mesnard, C. Hardin, R. Dadashi, S. Bhupatiraju, S. Pathak, L. Sifre, M. Rivière, M. S. Kale, J. Love, P. Tafti, L. Hussenot, P. G. Sessa, A. Chowdhery, A. Roberts, A. Barua, A. Botev, A. Castro-Ros, A. Slone, A. Héliou, A. Tacchetti, A. Bulanova, A. Paterson, B. Tsai, B. Shahriari, C. L. Lan, C. A. Choquette-Choo, C. Crepy, D. Cer, D. Ippolito, D. Reid, E. Buchatskaya, E. Ni, E. Noland, G. Yan, G. Tucker, G.-C. Muraru, G. Rozhdestvenskiy, H. Michalewski, I. Tenney, I. Grishchenko, J. Austin, J. Keeling, J. Labanowski, J.-B. Lespiau, J. Stanway, J. Brennan, J. Chen, J. Ferret, J. Chiu, J. Mao-Jones, K. Lee, K. Yu, K. Millican, L. L. Sjoesund, L. Lee, L. Dixon, M. Reid, M. Mikuła, M. Wirth, M. Sharman, N. Chinaev, N. Thain, O. Bachem, O. Chang, O. Wahltinez, P. Bailey, P. Michel, P. Yotov, R. Chaabouni, R. Comanescu, R. Jana, R. Anil, R. McIlroy, R. Liu, R. Mullins, S. L. Smith, S. Borgeaud, S. Girgin, S. Douglas, S. Pandya, S. Shakeri, S. De, T. Klimenko, T. Hennigan, V. Feinberg, W. Stokowiec, Y. hui Chen, Z. Ahmed, Z. Gong, T. Warkentin, L. Peran, M. Giang, C. Farabet, O. Vinyals, J. Dean, K. Kavukcuoglu, D. Hassabis, Z. Ghahramani, D. Eck, J. Barral, F. Pereira, E. Collins, A. Joulin, N. Fiedel, E. Senter, A. Andreev, and K. Kenealy. Gemma: Open models based on gemini research and technology. arXiv preprint arXiv:2403.08295 , 2024.

[35] Y. Li, S. Bubeck, R. Eldan, A. D. Giorno, S. Gunasekar, and Y. T. Lee. Textbooks are all you need ii: phi-1.5 technical report. arXiv preprint arXiv:2309.05463 , 2023.

[36] J. Bai, S. Bai, Y. Chu, Z. Cui, K. Dang, X. Deng, Y. Fan, W. Ge, Y. Han, F. Huang, et al. Qwen technical report. arXiv preprint arXiv:2309.16609 , 2023.

[37] J. Li, D. Li, C. Xiong, and S. C. H. Hoi. BLIP: Bootstrapping language-image pre-training for unified vision-language understanding and generation. In International Conference on Machine Learning (ICML) , 2022.

[38] J. Li, D. Li, S. Savarese, and S. C. H. Hoi. BLIP-2: Bootstrapping language-image pre-training with frozen image encoders and large language models. In International Conference on Machine Learning (ICML) , 2023.

[39] W. Dai, J. Li, D. Li, A. M. H. Tiong, J. Zhao, W. Wang, B. A. Li, P. Fung, and S. C. H. Hoi. InstructBLIP: Towards general-purpose vision-language models with instruction tuning. arXiv preprint arXiv:2305.06500 , 2023.

[40] H. H. Tan and M. Bansal. LXMERT: Learning cross-modality encoder representations from transformers. In Empirical Methods in Natural Language Processing (EMNLP) , 2019.

[41] H. Laurençon, L. Saulnier, L. Tronchon, S. Bekman, A. Singh, A. Lozhkov, T. Wang, S. Karamcheti, A. M. Rush, D. Kiela, M. Cord, and V. Sanh. OBELICS: An open web-scale filtered dataset of interleaved image-text documents. In Neural Information Processing Systems Track on Datasets and Benchmarks (NeurIPS Datasets and Benchmarks) , 2023.

[42] H. Liu, C. Li, Q. Wu, and Y. J. Lee. Visual instruction tuning. In Advances in Neural Information Processing Systems (NeurIPS) , 2023a.

[43] H. Liu, C. Li, Y. Li, and Y. J. Lee. Improved baselines with visual instruction tuning. arXiv preprint arXiv:2310.03744 , 2023b.

[44] S. Karamcheti, S. Nair, A. Balakrishna, P. Liang, T. Kollar, and D. Sadigh. Prismatic vlms: Investigating the design space of visually-conditioned language models. arXiv preprint arXiv:2402.07865 , 2024.

[45] D. Kalashnikov, A. Irpan, P. Pastor, J. Ibarz, A. Herzog, E. Jang, D. Quillen, E. Holly, M. Kalakrishnan, V. Vanhoucke, et al. QT-Opt: Scalable deep reinforcement learning for vision-based robotic manipulation. arXiv preprint arXiv:1806.10293 , 2018.

[46] D. Kalashnkov, J. Varley, Y. Chebotar, B. Swanson, R. Jonschkowski, C. Finn, S. Levine, and K. Hausman. Mt-opt: Continuous multi-task robotic reinforcement learning at scale. arXiv , 2021.

[47] F. Ebert, Y. Yang, K. Schmeckpeper, B. Bucher, G. Georgakis, K. Daniilidis, C. Finn, and S. Levine. Bridge data: Boosting generalization of robotic skills with cross-domain datasets. arXiv preprint arXiv:2109.13396 , 2021.

[48] K. Ehsani, T. Gupta, R. Hendrix, J. Salvador, L. Weihs, K.-H. Zeng, K. P. Singh, Y. Kim, W. Han, A. Herrasti, et al. Imitating shortest paths in simulation enables effective navigation and manipulation in the real world. arXiv preprint arXiv:2312.02976 , 2023.

[49] H. Bharadhwaj, J. Vakil, M. Sharma, A. Gupta, S. Tulsiani, and V. Kumar. Roboagent: Generalization and efficiency in robot manipulation via semantic augmentations and action chunking. arXiv preprint arXiv:2309.01918 , 2023.

[50] L. Pinto and A. Gupta. Supersizing self-supervision: Learning to grasp from 50k tries and 700 robot hours. In 2016 IEEE international conference on robotics and automation (ICRA) , pages 3406–3413. IEEE, 2016.

[51] A. Mandlekar, Y. Zhu, A. Garg, J. Booher, M. Spero, A. Tung, J. Gao, J. Emmons, A. Gupta, E. Orbay, et al. Roboturk: A crowdsourcing platform for robotic skill learning through imitation. In Conference on Robot Learning , pages 879–893. PMLR, 2018.

[52] A. Gupta, A. Murali, D. P. Gandhi, and L. Pinto. Robot learning in homes: Improving generalization and reducing dataset bias. Advances in neural information processing systems , 31, 2018.

[53] S. Dasari, F. Ebert, S. Tian, S. Nair, B. Bucher, K. Schmeckpeper, S. Singh, S. Levine, and C. Finn. Robonet: Large-scale multi-robot learning. CoRL , 2019.

[54] S. Cabi, S. G. Colmenarejo, A. Novikov, K. Konyushkova, S. Reed, R. Jeong, K. Zolna, Y. Aytar, D. Budden, M. Vecerik, O. Sushkov, D. Barker, J. Scholz, M. Denil, N. de Freitas, and Z. Wang. Scaling data-driven robotics with reward sketching and batch reinforcement learning. RSS , 2019.

[55] E. Jang, A. Irpan, M. Khansari, D. Kappler, F. Ebert, C. Lynch, S. Levine, and C. Finn. Bc-z: Zero-shot task generalization with robotic imitation learning. In Conference on Robot Learning , pages 991–1002. PMLR, 2022.

[56] H.-S. Fang, H. Fang, Z. Tang, J. Liu, C. Wang, J. Wang, H. Zhu, and C. Lu. Rh20t: A comprehensive robotic dataset for learning diverse skills in one-shot. Towards Generalist Robots: Learning Paradigms for Scalable Skill Acquisition@ CoRL2023 , 3:5, 2023.

[57] C. Devin, A. Gupta, T. Darrell, P. Abbeel, and S. Levine. Learning modular neural network policies for multi-task and multi-robot transfer. In Proceedings of IEEE International Conference on Robotics and Automation , 2017.

[58] E. S. Hu, K. Huang, O. Rybkin, and D. Jayaraman. Know thyself: Transferable visual control policies through robot-awareness. In International Conference on Learning Representations , 2022.

[59] J. H. Yang, D. Sadigh, and C. Finn. Polybot: Training one policy across robots while embracing variability. In 7th Annual Conference on Robot Learning , 2023. URL https://openreview.net/forum?id=HEIRj51lcS .

[60] S. Reed, K. Zolna, E. Parisotto, S. G. Colmenarejo, A. Novikov, G. Barth-maron, M. Giménez, Y. Sulsky, J. Kay, J. T. Springenberg, T. Eccles, J. Bruce, A. Razavi, A. Edwards, N. Heess, Y. Chen, R. Hadsell, O. Vinyals, M. Bordbar, and N. de Freitas. A generalist agent. Transactions on Machine Learning Research , 2022. ISSN 2835-8856.

[61] G. Salhotra, I.-C. A. Liu, and G. Sukhatme. Bridging action space mismatch in learning from demonstrations. arXiv preprint arXiv:2304.03833 , 2023.

[62] I. Radosavovic, B. Shi, L. Fu, K. Goldberg, T. Darrell, and J. Malik. Robot learning with sensorimotor pre-training. In Conference on Robot Learning , 2023.

[63] D. Shah, A. Sridhar, A. Bhorkar, N. Hirose, and S. Levine. Gnm: A general navigation model to drive any robot. In 2023 IEEE International Conference on Robotics and Automation (ICRA) , pages 7226–7233. IEEE, 2023.

[64] K. Bousmalis, G. Vezzani, D. Rao, C. Devin, A. X. Lee, M. Bauza, T. Davchev, Y. Zhou, A. Gupta, A. Raju, et al. Robocat: A self-improving foundation agent for robotic manipulation. arXiv preprint arXiv:2306.11706 , 2023.

[65] D. Shah, A. Sridhar, N. Dashora, K. Stachowicz, K. Black, N. Hirose, and S. Levine. ViNT: A foundation model for visual navigation. In 7th Annual Conference on Robot Learning , 2023. URL https://arxiv.org/abs/2306.14846 .

[66] J. Yang, C. Glossop, A. Bhorkar, D. Shah, Q. Vuong, C. Finn, D. Sadigh, and S. Levine. Pushing the limits of cross-embodiment learning for manipulation and navigation. arXiv preprint arXiv:2402.19432 , 2024.

[67] S. Y. Gadre, M. Wortsman, G. Ilharco, L. Schmidt, and S. Song. Cows on pasture: Baselines and benchmarks for language-driven zero-shot object navigation. In Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition , pages 23171–23181, 2023.

[68] Y. Du, K. Konyushkova, M. Denil, A. Raju, J. Landon, F. Hill, N. de Freitas, and S. Cabi. Vision-language models as success detectors. arXiv preprint arXiv:2303.07280 , 2023.

[69] Y. J. Ma, V. Kumar, A. Zhang, O. Bastani, and D. Jayaraman. Liv: Language-image representations and rewards for robotic control. In International Conference on Machine Learning , pages 23301–23320. PMLR, 2023.

[70] X. Zhang, Y. Ding, S. Amiri, H. Yang, A. Kaminski, C. Esselink, and S. Zhang. Grounding classical task planners via vision-language models. arXiv preprint arXiv:2304.08587 , 2023.

[71] S. Sontakke, J. Zhang, S. Arnold, K. Pertsch, E. Bıyık, D. Sadigh, C. Finn, and L. Itti. Roboclip: One demonstration is enough to learn robot policies. Advances in Neural Information Processing Systems , 36, 2024.

[72] J. Huang, S. Yong, X. Ma, X. Linghu, P. Li, Y. Wang, Q. Li, S.-C. Zhu, B. Jia, and S. Huang. An embodied generalist agent in 3d world. In Proceedings of the International Conference on Machine Learning (ICML) , 2024.

[73] X. Li, M. Liu, H. Zhang, C. Yu, J. Xu, H. Wu, C. Cheang, Y. Jing, W. Zhang, H. Liu, et al. Vision-language foundation models as effective robot imitators. arXiv preprint arXiv:2311.01378 , 2023.

[74] H. Zhen, X. Qiu, P. Chen, J. Yang, X. Yan, Y. Du, Y. Hong, and C. Gan. 3d-vla: 3d vision-language-action generative world model. arXiv preprint arXiv:2403.09631 , 2024.

[75] PyTorch. Automatic mixed precision. URL https://pytorch.org/docs/stable/amp.html .

[76] T. Dao. Flashattention-2: Faster attention with better parallelism and work partitioning. arXiv preprint arXiv:2307.08691 , 2023.

[77] Y. Zhao, A. Gu, R. Varma, L. Luo, C.-C. Huang, M. Xu, L. Wright, H. Shojanazeri, M. Ott, S. Shleifer, et al. Pytorch fsdp: experiences on scaling fully sharded data parallel. arXiv preprint arXiv:2304.11277 , 2023.

[78] N. Dorka, C. Huang, T. Welschehold, and W. Burgard. What matters in employing vision language models for tokenizing actions in robot control? In First Workshop on Vision-Language Models for Navigation and Manipulation at ICRA 2024 .

[79] X. Zhai, B. Mustafa, A. Kolesnikov, and L. Beyer. Sigmoid loss for language image pre-training. In Proceedings of the IEEE/CVF International Conference on Computer Vision , pages 11975–11986, 2023.

[80] A. Radford, J. W. Kim, C. Hallacy, A. Ramesh, G. Goh, S. Agarwal, G. Sastry, A. Askell, P. Mishkin, J. Clark, et al. Learning transferable visual models from natural language supervision. In International conference on machine learning , pages 8748–8763. PMLR, 2021.

[81] P. Sharma, N. Ding, S. Goodman, and R. Soricut. Conceptual captions: A cleaned, hypernymed, image alt-text dataset for automatic image captioning. In Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers) , pages 2556–2565, 2018.

[82] C. Schuhmann, R. Vencu, R. Beaumont, R. Kaczmarczyk, C. Mullis, A. Katta, T. Coombes, J. Jitsev, and A. Komatsuzaki. Laion-400m: Open dataset of clip-filtered 400 million image-text pairs. arXiv preprint arXiv:2111.02114 , 2021.

[83] O. Sidorov, R. Hu, M. Rohrbach, and A. Singh. Textcaps: a dataset for image captioning with reading comprehension. In Computer Vision–ECCV 2020: 16th European Conference, Glasgow, UK, August 23–28, 2020, Proceedings, Part II 16 , pages 742–758. Springer, 2020.

[84] H. Face. Introducing idefics: An open reproduction of state-of-the-art visual langage model. Hugging Face Blog , 2024.

[85] H. Liu, C. Li, Q. Wu, and Y. J. Lee. Visual instruction tuning. Advances in neural information processing systems , 36, 2024.

[86] B. McKinzie, Z. Gan, J.-P. Fauconnier, S. Dodge, B. Zhang, P. Dufter, D. Shah, X. Du, F. Peng, F. Weers, et al. Mm1: Methods, analysis & insights from multimodal llm pre-training. arXiv preprint arXiv:2403.09611 , 2024.

[87] J. Lin, H. Yin, W. Ping, Y. Lu, P. Molchanov, A. Tao, H. Mao, J. Kautz, M. Shoeybi, and S. Han. Vila: On pre-training for visual language models. arXiv preprint arXiv:2312.07533 , 2023.

[88] T. Dettmers, M. Lewis, Y. Belkada, and L. Zettlemoyer. Gpt3. int8 (): 8-bit matrix multiplication for transformers at scale. Advances in Neural Information Processing Systems , 35:30318–30332, 2022.

[89] NVIDIA. Tensorrt-llm. URL https://github.com/NVIDIA/TensorRT-LLM .

[90] T. Z. Zhao, V. Kumar, S. Levine, and C. Finn. Learning fine-grained bimanual manipulation with low-cost hardware. arXiv preprint arXiv:2304.13705 , 2023.

[91] Y. Leviathan, M. Kalman, and Y. Matias. Fast inference from transformers via speculative decoding. In International Conference on Machine Learning , pages 19274–19286. PMLR, 2023.

[92] A. Brohan, N. Brown, J. Carbajal, Y. Chebotar, J. Dabis, C. Finn, K. Gopalakrishnan, K. Hausman, A. Herzog, J. Hsu, et al. Rt-1: Robotics transformer for real-world control at scale. arXiv preprint arXiv:2212.06817 , 2022.

[93] E. Rosete-Beas, O. Mees, G. Kalweit, J. Boedecker, and W. Burgard. Latent plans for task agnostic offline reinforcement learning. In Proceedings of the 6th Conference on Robot Learning (CoRL) , 2022.

[94] O. Mees, J. Borja-Diaz, and W. Burgard. Grounding language with visual affordances over unstructured data. In Proceedings of the IEEE International Conference on Robotics and Automation (ICRA) , London, UK, 2023.

[95] S. Dass, J. Yapeter, J. Zhang, J. Zhang, K. Pertsch, S. Nikolaidis, and J. J. Lim. CLVR jaco play dataset, 2023. URL https://github.com/clvrai/clvr_jaco_play_dataset .

[96] J. Luo, C. Xu, X. Geng, G. Feng, K. Fang, L. Tan, S. Schaal, and S. Levine. Multi-stage cable routing through hierarchical imitation learning. arXiv preprint arXiv:2307.08927 , 2023.

[97] A. Mandlekar, Y. Zhu, A. Garg, J. Booher, M. Spero, A. Tung, J. Gao, J. Emmons, A. Gupta, E. Orbay, S. Savarese, and L. Fei-Fei. RoboTurk: A crowdsourcing platform for robotic skill learning through imitation. CoRR , abs/1811.02790, 2018. URL http://arxiv.org/abs/1811.02790 .

[98] Y. Zhu, A. Joshi, P. Stone, and Y. Zhu. Viola: Imitation learning for vision-based manipulation with object proposal priors, 2023.

[99] L. Y. Chen, S. Adebola, and K. Goldberg. Berkeley UR5 demonstration dataset. https://sites.google.com/view/berkeley-ur5/home .

[100] G. Zhou, V. Dean, M. K. Srirama, A. Rajeswaran, J. Pari, K. Hatch, A. Jain, T. Yu, P. Abbeel, L. Pinto, C. Finn, and A. Gupta. Train offline, test online: A real robot learning benchmark, 2023.

[101] C. Lynch, A. Wahid, J. Tompson, T. Ding, J. Betker, R. Baruch, T. Armstrong, and P. Florence. Interactive language: Talking to robots in real time. IEEE Robotics and Automation Letters , 2023.

[102] S. Belkhale, Y. Cui, and D. Sadigh. Hydra: Hybrid robot actions for imitation learning. arxiv , 2023.

[103] Y. Zhu, P. Stone, and Y. Zhu. Bottom-up skill discovery from unsegmented demonstrations for long-horizon robot manipulation. IEEE Robotics and Automation Letters , 7(2):4126–4133, 2022.

[104] Z. J. Cui, Y. Wang, N. M. M. Shafiullah, and L. Pinto. From play to policy: Conditional behavior generation from uncurated robot data. arXiv preprint arXiv:2210.10047 , 2022.

[105] M. Heo, Y. Lee, D. Lee, and J. J. Lim. Furniturebench: Reproducible real-world benchmark for long-horizon complex manipulation. In Robotics: Science and Systems , 2023.

[106] G. Yan, K. Wu, and X. Wang. ucsd kitchens Dataset. August 2023.

[107] S. Nasiriany, T. Gao, A. Mandlekar, and Y. Zhu. Learning and retrieval from prior data for skill-based imitation learning. In Conference on Robot Learning (CoRL) , 2022.

[108] H. Liu, S. Nasiriany, L. Zhang, Z. Bao, and Y. Zhu. Robot learning on the job: Human-in-the-loop autonomy and learning during deployment. In Robotics: Science and Systems (RSS) , 2023.

[109] G. Quere, A. Hagengruber, M. Iskandar, S. Bustamante, D. Leidner, F. Stulp, and J. Vogel. Shared Control Templates for Assistive Robotics. In 2020 IEEE International Conference on Robotics and Automation (ICRA) , page 7, Paris, France, 2020.

[110] S. Saxena, M. Sharma, and O. Kroemer. Multi-resolution sensing for real-time control with vision-language models. In 7th Annual Conference on Robot Learning , 2023. URL https://openreview.net/forum?id=WuBv9-IGDUA .

[111] R. Shah, R. Martín-Martín, and Y. Zhu. MUTEX: Learning unified policies from multimodal task specifications. In 7th Annual Conference on Robot Learning , 2023. URL https://openreview.net/forum?id=PwqiqaaEzJ .

[112] X. Zhu, R. Tian, C. Xu, M. Ding, W. Zhan, and M. Tomizuka. Fanuc manipulation: A dataset for learning-based manipulation with fanuc mate 200id robot. 2023.

[113] R. Mendonca, S. Bahl, and D. Pathak. Structured world models from human videos. CoRL , 2023.

[114] J. Luo, C. Xu, F. Liu, L. Tan, Z. Lin, J. Wu, P. Abbeel, and S. Levine. Fmb: a functional manipulation benchmark for generalizable robotic learning. arXiv preprint arXiv:2401.08553 , 2024.

[115] N. M. M. Shafiullah, A. Rai, H. Etukuru, Y. Liu, I. Misra, S. Chintala, and L. Pinto. On bringing robots home, 2023.

[116] B. Liu, Y. Zhu, C. Gao, Y. Feng, Q. Liu, Y. Zhu, and P. Stone. Libero: Benchmarking knowledge transfer for lifelong robot learning. Advances in Neural Information Processing Systems , 36, 2024.

[117] V. Sanh, L. Debut, J. Chaumond, and T. Wolf. Distilbert, a distilled version of bert: smaller, faster, cheaper and lighter. arXiv preprint arXiv:1910.01108 , 2019.

Appendix A Data Mixture Details

附录 A 数据配比细节

We list our used data mixture in Table 3. The mixture mostly follows [5], with a few additional datasets.

表 3 列出了我们使用的数据配比。该配比大体沿用 [5],另外加入了几个数据集。

OpenVLA Training Dataset Mixture
Fractal [92] 12.7%
Kuka [45] 12.7%
Bridge[47, 6] 13.3%
Taco Play [93, 94] 3.0%
Jaco Play [95] 0.4%
Berkeley Cable Routing [96] 0.2%
Roboturk [97] 2.3%
Viola [98] 0.9%
Berkeley Autolab UR5 [99] 1.2%
Toto [100] 2.0%
Language Table [101] 4.4%
Stanford Hydra Dataset [102] 4.4%
Austin Buds Dataset [103] 0.2%
NYU Franka Play Dataset [104] 0.8%
Furniture Bench Dataset [105] 2.4%
UCSD Kitchen Dataset [106] <0.1%
Austin Sailor Dataset [107] 2.2%
Austin Sirius Dataset [108] 1.7%
DLR EDAN Shared Control [109] <0.1%
IAMLab CMU Pickup Insert [110] 0.9%
UTAustin Mutex [111] 2.2%
Berkeley Fanuc Manipulation [112] 0.7%
CMU Stretch [113] 0.2%
BC-Z [55] 7.5%
FMB Dataset [114] 7.1%
DobbE [115] 1.4%
DROID [11] 10.0%⁶

Table 3: OpenVLA training data mixture using datasets from the Open X-Embodiment dataset [1], following [5] with a few additions.

OpenVLA 训练数据配比
Fractal [92] 12.7%
Kuka [45] 12.7%
Bridge[47, 6] 13.3%
Taco Play [93, 94] 3.0%
Jaco Play [95] 0.4%
Berkeley Cable Routing [96] 0.2%
Roboturk [97] 2.3%
Viola [98] 0.9%
Berkeley Autolab UR5 [99] 1.2%
Toto [100] 2.0%
Language Table [101] 4.4%
Stanford Hydra Dataset [102] 4.4%
Austin Buds Dataset [103] 0.2%
NYU Franka Play Dataset [104] 0.8%
Furniture Bench Dataset [105] 2.4%
UCSD Kitchen Dataset [106] <0.1%
Austin Sailor Dataset [107] 2.2%
Austin Sirius Dataset [108] 1.7%
DLR EDAN Shared Control [109] <0.1%
IAMLab CMU Pickup Insert [110] 0.9%
UTAustin Mutex [111] 2.2%
Berkeley Fanuc Manipulation [112] 0.7%
CMU Stretch [113] 0.2%
BC-Z [55] 7.5%
FMB Dataset [114] 7.1%
DobbE [115] 1.4%
DROID [11] 10.0%⁶

表 3:OpenVLA 的训练数据配比,使用 Open X-Embodiment 数据集 [1] 中的数据集,沿用 [5] 并做了少量补充。

⁶ We remove DROID for the last third of training due to slow learning progress (see Section 3.3) and re-distribute its mixture weights across all other datasets.

⁶ 由于学习进展缓慢,我们在训练的最后三分之一阶段移除了 DROID(见第 3.3 节),并把它的配比权重重新分配给其他所有数据集。

Appendix B Evaluation Tasks and Detailed Results

附录 B 评测任务与详细结果

In this section, we provide more details on the BridgeData V2 WidowX and Google robot evaluations discussed in Section 5.1, as well as the Franka-Tabletop and Franka-DROID fine-tuning evaluations discussed in Section 5.2.

本节提供第 5.1 节讨论的 BridgeData V2 WidowX 和 Google 机器人评测,以及第 5.2 节讨论的 Franka-Tabletop 和 Franka-DROID 微调评测的更多细节。

B.1 BridgeData V2 WidowX Evaluation Details

B.1 BridgeData V2 WidowX 评测细节

Here we focus specifically on BridgeData V2 evaluations discussed in Section 5.1.

这里专门介绍第 5.1 节讨论的 BridgeData V2 评测。

B.1.1 BridgeData V2 Evaluation Tasks

As described in Section 5.1, we evaluate each generalist robot manipulation policy on 17 tasks with 10 trials each. In this section, we provide details on the task categories and individual tasks.

B.1.1 BridgeData V2 评测任务

如第 5.1 节所述,我们在 17 个任务上评测每个通用机器人操作策略,每个任务 10 次试验。本节给出任务类别和各个任务的细节。

Figure 7

Figure 7: BridgeData V2 WidowX robot evaluation tasks. We evaluate every generalist robot policy on 4 types out-of-distribution (OOD) generalization tasks: visual, motion, physical, and semantic (as defined in Section 5.1). Every pair of images shows the start state and an example end state after the robot completes the task. We also rigorously assess language grounding in the 3 tasks shown in the bottom 3 rows, by changing the prompt while fixing the initial state and testing whether the policy can approach the correct target object.

图 7:BridgeData V2 WidowX 机器人评测任务。我们在 4 类分布外(OOD)泛化任务上评测每个通用机器人策略:视觉、运动、物理和语义泛化(定义见第 5.1 节)。每对图像展示起始状态,以及机器人完成任务后的一个示例结束状态。我们还在最下面 3 行所示的 3 个任务中严格评估语言落地能力:固定初始状态、改变提示语,测试策略能否接近正确的目标物体。

In total, we evaluate on 5 visual generalization tasks, 2 motion generalization tasks, 3 physical generalization tasks, 4 semantic generalization tasks, and 3 language grounding tasks. Note that all tasks we evaluate on introduce some form of distribution shift since we are unable to procure the exact objects used in the original dataset (other distribution shifts naturally arise as we reproduce a real-world test environment originally constructed at a different location; see Section B.1.2 for a detailed discussion on such distribution shifts). All 17 tasks are depicted in Fig. 7. Each rollout is marked as a failure (0) or success (1). In some more difficult tasks, we record partial successes (0.5); we describe the conditions for partial credit in the task descriptions below.

我们总共评测 5 个视觉泛化任务、2 个运动泛化任务、3 个物理泛化任务、4 个语义泛化任务和 3 个语言落地任务。注意,我们评测的所有任务都引入了某种形式的分布偏移,因为我们无法获得原始数据集中使用的那些物体(我们在别处重建一个原本搭建在另一个地点的真实测试环境,也会自然产生其他分布偏移;关于这类分布偏移的详细讨论见 B.1.2 节)。全部 17 个任务见图 7。每次试验记为失败(0)或成功(1)。在一些较难的任务中,我们记录部分成功(0.5);获得部分分数的条件在下面的任务描述中说明。

Below we describe each of the 17 tasks, in the order shown in Fig. 7:

下面按图 7 中的顺序逐一描述这 17 个任务:

  1. Put Eggplant into Pot (Easy Version): The robot’s goal is to pick up the eggplant and drop it into the pot. This is a visual generalization task because we use a handcrafted paper pot that has a different appearance than the pot used in the original BridgeData V2 training dataset (since we are unable to procure the original pot). Unlike all 16 other tasks, for this particular task we initialize the robot’s end-effector directly above the eggplant before rolling out the policy; hence, we call this the “Easy Version” of the “Put Eggplant into Pot” task.
  2. Put Eggplant into Pot: This is the same task as described above, except that the robot’s end-effector is not initialized directly above the eggplant. Instead, we initialize it in a position that is fixed across all rollouts, which means that the robot must horizontally reach for the eggplant first before manipulating it. (Note: The same applies to all other tasks described below.) This is a visual generalization task for the same reason as above.
  3. Put Cup from Counter into Sink: The robot’s goal is to pick up the pink cup from either the kitchen countertop or drying rack and place it into the sink on the right. This is a visual generalization task because we use a pink cup rather than a blue cup (a blue cup is used in the original BridgeData V2 dataset, but we find that none of the methods we evaluate is able to manipulate it reliably – most likely because the color of the cup blends in with the color of the sink).
  4. Put Eggplant into Pot (w/ Clutter): This is the same task as the “Put Eggplant into Pot” task, except that it is more difficult due to the presence of several distractor objects. It is a visual generalization task for the same reason discussed in the normal “Put Eggplant into Pot” task, and even more so given unseen distractors in the scene. Partial credit (0.5 out of 1) is rewarded when the robot moves towards the correct target object.
  5. Put Yellow Corn on Pink Plate: The robot’s goal is to pick up the yellow corn and place it on the pink plate. This is a visual generalization task due to the presence of unseen distractor objects in the scene, such as a green dinosaur on the countertop in the back section of the sink. Partial credit (0.5 out of 1) is rewarded when the robot moves towards the correct target object.
  6. Lift Eggplant: The robot’s goal is to grasp and lift the eggplant into the air. This is a motion generalization task because the eggplant is initialized in unseen positions and/or orientations, and the robot is forced to move beyond its training distribution of positions and/or orientations and often perform long-range reaching in order to complete the task. (Note: Long-range reaching is not demonstrated in this environment in the original BridgeData V2 demonstrations; see Section B.1.2 for details.) We find that this task, though seemingly simple, is deceptively challenging for many policies. Partial credit (0.5 out of 1) is rewarded when the robot makes contact with the eggplant.
  7. Put Carrot on Plate (w/ Height Change): The robot’s goal is to pick up the carrot and place it on the yellow plate. This is a motion generalization task because the plate is elevated from its usual position at the bottom of the sink, and the robot must adjust its trajectory to correctly place the carrot on the elevated platform (without knocking down the plate in the process). Partial credit (0.5 out of 1) is rewarded when the robot grasps the carrot and touches the plate with it.
  8. Put Carrot on Plate: This is the same task as above, except that the plate is at its normal position (at the bottom of the sink or drying rack). We consider this a physical generalization task because the carrot has a different size and shape than the one used in the original BridgeData V2 dataset, which is shorter and narrower. (Note that the previous version of this task listed above would also technically be a physical generalization task since it involves the same carrot, but we list it under the “motion generalization” category since that is the focus there.)
  9. Flip Pot Upright: The robot’s goal is to manipulate the pot such that it is oriented upright in the sink at the end of the episode. This is a physical generalization task because this pot has a different size and shape than the one used in the original BridgeData V2 training demonstrations (the pot we use is wider and shorter).
  10. Lift AAA Battery: The robot’s goal is simply to grasp the AAA battery and lift it up into the air. This is considered a physical generalization task because the battery is much smaller and thinner than target objects seen in the BridgeData V2 training demonstrations in this environment; see Section B.1.2 for details. (Note that this target object does not exist in the original BridgeData V2 demonstrations in this environment, so this is also an instance of “semantic generalization”, but we classify it solely as “physical generalization” since that is the main focus here).
  11. Move Skull into Drying Rack: The robot’s goal is to grasp the skull windup toy and drop it into the yellow drying rack in the left part of the sink. This is a semantic generalization task since the skull is an unseen target object (does not appear in the BridgeData V2 training demonstrations).
  12. Lift White Tape: The robot’s goal is to grasp and lift the white roll of tape into the air. This is a semantic generalization task since the white tape roll is an unseen target object (does not appear in the BridgeData V2 training demonstrations). (Note that this task may also be considered as “physical generalization” because of its shape being different than the objects seen in the training demonstrations in this environment; most policies struggle to grasp objects with this ring structure, and they often move the robot’s end-effector directly into the center region.)
  13. Take Purple Grapes out of Pot: The robot’s goal is to grasp the purple grapes lying inside the steel pot and remove it from the pot (by lifting it out and/or dropping it anywhere outside the pot). This is a semantic generalization task because it is an unseen language instruction; the robot has never seen this task in the original BridgeData V2 training dataset.
  14. Stack Blue Cup on Pink Cup: The robot’s goal is to grasp the blue cup and place it securely on top of the pink cup. This is a semantic generalization task because it is an unseen language instruction; the robot has never seen this task in this environment in the original BridgeData V2 training dataset. Partial credit (0.5 out of 1) is rewarded when the robot grasps the blue cup and touches the pink cup with the blue cup.
  15. Put {Eggplant, Red Bottle} into Pot: This is a language grounding task. The robot’s goal is to put the specified target object into the pot. Both the eggplant and red bottle are present in the scene. We conduct paired evaluations: for the same initial state, we prompt the policy to target the eggplant in one episode, and then the red bottle in the next episode. We test each method 5 times with the eggplant and 5 times with the red bottle, using the same set of 5 initial states for both target objects. Partial credit (0.5 out of 1) is rewarded when the robot moves towards the correct target object.
  16. Lift {Cheese, Red Chili Pepper}: This is a language grounding task. The robot’s goal is to grasp and lift the specified target object. We conduct paired evaluations as described in the task above. Partial credit (0.5 out of 1) is rewarded when the robot moves towards the correct target object.
  17. Put {Blue Cup, Pink Cup} on Plate: This is a language grounding task. The robot’s goal is to grasp the specified target object and place it onto the plate. We conduct paired evaluations as described in other language grounding tasks. Partial credit (0.5 out of 1) is rewarded when the robot moves towards the correct target object.
  1. 把茄子放进锅里(简单版):机器人的目标是拿起茄子并把它丢进锅里。这是一个视觉泛化任务,因为我们用的是手工制作的纸锅,外观与原始 BridgeData V2 训练数据集中的锅不同(因为我们无法获得原来的锅)。与其他 16 个任务都不同,这个任务在运行策略之前,我们把机器人末端执行器直接初始化在茄子正上方;因此我们称之为“把茄子放进锅里”任务的“简单版”。
  2. 把茄子放进锅里:与上面的任务相同,只是机器人末端执行器不再初始化在茄子正上方,而是初始化在一个所有试验都固定的位置,这意味着机器人必须先水平伸向茄子,再操作它。(注:下面描述的所有其他任务也是如此。)出于与上面相同的原因,这是一个视觉泛化任务。
  3. 把杯子从台面放进水槽:机器人的目标是从厨房台面或沥水架上拿起粉色杯子,放进右边的水槽。这是一个视觉泛化任务,因为我们用的是粉色杯子而不是蓝色杯子(原始 BridgeData V2 数据集中用的是蓝色杯子,但我们发现所评测的方法都无法可靠地操作它——很可能是因为杯子的颜色与水槽的颜色混在一起)。
  4. 把茄子放进锅里(有杂物):与“把茄子放进锅里”任务相同,只是因为场景中有若干干扰物体而更难。它是一个视觉泛化任务,理由与普通的“把茄子放进锅里”任务相同,而场景中还有未见过的干扰物,更是如此。机器人朝正确的目标物体移动时给予部分分数(满分 1 分中的 0.5 分)。
  5. 把黄玉米放到粉色盘子上:机器人的目标是拿起黄玉米并放到粉色盘子上。这是一个视觉泛化任务,因为场景中有未见过的干扰物体,例如水槽后部台面上的一只绿色恐龙。机器人朝正确的目标物体移动时给予部分分数(满分 1 分中的 0.5 分)。
  6. 举起茄子:机器人的目标是抓住茄子并把它举到空中。这是一个运动泛化任务,因为茄子被初始化在未见过的位置和/或朝向,机器人被迫超出其训练分布中的位置和/或朝向,常常需要远距离伸够才能完成任务。(注:在原始 BridgeData V2 示教中,这个环境里没有远距离伸够的示范;详见 B.1.2 节。)我们发现这个任务看似简单,对许多策略来说却出人意料地有挑战性。机器人接触到茄子时给予部分分数(满分 1 分中的 0.5 分)。
  7. 把胡萝卜放到盘子上(高度变化):机器人的目标是拿起胡萝卜并放到黄色盘子上。这是一个运动泛化任务,因为盘子被从它通常在水槽底部的位置抬高了,机器人必须调整轨迹,才能把胡萝卜正确地放到抬高的平台上(过程中不能把盘子碰倒)。机器人抓住胡萝卜并用它碰到盘子时给予部分分数(满分 1 分中的 0.5 分)。
  8. 把胡萝卜放到盘子上:与上面的任务相同,只是盘子位于正常位置(水槽底部或沥水架上)。我们把它视为物理泛化任务,因为这根胡萝卜的尺寸和形状与原始 BridgeData V2 数据集中所用的不同,后者更短更细。(注意,上面列出的前一个版本严格说来也属于物理泛化任务,因为用的是同一根胡萝卜,但我们把它列在“运动泛化”类别下,因为那才是它的重点。)
  9. 把锅翻正:机器人的目标是操作锅,使它在回合结束时在水槽里正立。这是一个物理泛化任务,因为这只锅的尺寸和形状与原始 BridgeData V2 训练示教中的锅不同(我们用的锅更宽更矮)。
  10. 举起 AAA 电池:机器人的目标只是抓住 AAA 电池并把它举到空中。这被视为物理泛化任务,因为这节电池比这个环境中 BridgeData V2 训练示教里出现的目标物体小得多、细得多;详见 B.1.2 节。(注意,这个目标物体在该环境的原始 BridgeData V2 示教中并不存在,所以这也是“语义泛化”的一个例子,但我们只把它归为“物理泛化”,因为那是这里的主要关注点。)
  11. 把骷髅放进沥水架:机器人的目标是抓住骷髅发条玩具,把它丢进水槽左侧的黄色沥水架。这是一个语义泛化任务,因为骷髅是未见过的目标物体(没有出现在 BridgeData V2 训练示教中)。
  12. 举起白色胶带:机器人的目标是抓住白色胶带卷并把它举到空中。这是一个语义泛化任务,因为白色胶带卷是未见过的目标物体(没有出现在 BridgeData V2 训练示教中)。(注意,这个任务也可以被视为“物理泛化”,因为它的形状与该环境训练示教中的物体不同;大多数策略难以抓取这种环形结构的物体,常常把机器人末端执行器直接伸进中间的空心区域。)
  13. 把紫葡萄从锅里拿出来:机器人的目标是抓住躺在钢锅里的紫葡萄,把它从锅里取出(举出来和/或丢到锅外任意位置)。这是一个语义泛化任务,因为这是一条未见过的语言指令;机器人在原始 BridgeData V2 训练数据集中从未见过这个任务。
  14. 把蓝杯子叠到粉杯子上:机器人的目标是抓住蓝色杯子,把它稳稳地放在粉色杯子上面。这是一个语义泛化任务,因为这是一条未见过的语言指令;机器人在原始 BridgeData V2 训练数据集的这个环境中从未见过这个任务。机器人抓住蓝杯子并用它碰到粉杯子时给予部分分数(满分 1 分中的 0.5 分)。
  15. 把 {茄子, 红瓶子} 放进锅里:这是一个语言落地任务。机器人的目标是把指定的目标物体放进锅里。茄子和红瓶子都在场景中。我们做成对评测:对同一个初始状态,在一个回合中提示策略去拿茄子,在下一个回合中提示它去拿红瓶子。每种方法用茄子测试 5 次、用红瓶子测试 5 次,两个目标物体使用同一组 5 个初始状态。机器人朝正确的目标物体移动时给予部分分数(满分 1 分中的 0.5 分)。
  16. 举起 {奶酪, 红辣椒}:这是一个语言落地任务。机器人的目标是抓住并举起指定的目标物体。我们按上一个任务所述的方式做成对评测。机器人朝正确的目标物体移动时给予部分分数(满分 1 分中的 0.5 分)。
  17. 把 {蓝杯子, 粉杯子} 放到盘子上:这是一个语言落地任务。机器人的目标是抓住指定的目标物体并把它放到盘子上。我们按其他语言落地任务所述的方式做成对评测。机器人朝正确的目标物体移动时给予部分分数(满分 1 分中的 0.5 分)。

B.1.2 Comparing Evaluation Tasks to Original BridgeData V2 Training Data

We conduct our evaluations in a sink environment used in the original BridgeData V2 dataset [6]. We reproduce the environment to match the original environment in the BridgeData V2 dataset with rough approximations for the robot’s location relative to the sink, as well as the camera’s placement relative to the scene. Given the lack of precise measurements of these positions in the original dataset, we are unable to reproduce the exact environment setup, and natural distribution shifts arise due to slightly different robot, sink, and camera placements. In addition, since we evaluate robot policies in a different location than where the training demonstrations were collected from, other natural distribution shifts arise. For example, the lighting conditions and background (e.g., visible areas behind the sink) are inevitably different than what was seen in the training dataset. Furthermore, we are unable to procure the exact set of objects used in the original BridgeData V2 dataset, so there are distribution shifts between the objects used at train time and those used at test time.

B.1.2 评测任务与原始 BridgeData V2 训练数据的比较

我们在原始 BridgeData V2 数据集 [6] 所用的一个水槽环境中进行评测。我们重建了这个环境,使之与 BridgeData V2 数据集中的原始环境相匹配,对机器人相对水槽的位置以及相机相对场景的摆放做了粗略近似。由于原始数据集中缺少这些位置的精确测量,我们无法重现完全一致的环境配置,机器人、水槽和相机的摆放略有不同,自然会产生分布偏移。此外,由于我们评测机器人策略的地点与采集训练示教的地点不同,还会产生其他自然的分布偏移。例如,光照条件和背景(例如水槽后面可见的区域)不可避免地与训练数据集中看到的不同。再者,我们无法获得原始 BridgeData V2 数据集中所用的那套物体,因此训练时和测试时所用的物体之间存在分布偏移。

Despite all these challenges, we find that certain generalist policies, such as OpenVLA and RT-2-X, can still generalize and perform various tasks fairly reliably “out-of-the-box”. Other generalist policies, such as RT-1-X and Octo, can also complete some tasks, though they struggle when tested with more difficult generalization tasks in our BridgeData V2 evaluation suite.

尽管有这些挑战,我们发现某些通用策略,例如 OpenVLA 和 RT-2-X,仍然能够“开箱即用”地泛化,并相当可靠地完成各种任务。其他通用策略,例如 RT-1-X 和 Octo,也能完成部分任务,但在我们 BridgeData V2 评测套件中较难的泛化任务上表现吃力。

The original BridgeData V2 dataset includes demonstrations of the following seven tasks in this specific sink environment: “Flip Pot Upright”, “Put Carrot on Plate”, “Put Cup from Counter (or Drying Rack) into Sink”, “Put Eggplant into Pot”, “Put Knife on Cutting Board”, “Put Spoon in Pot”, and “Turn Lever Vertical to Front”. See Fig. 8 for samples images of all these tasks from the original dataset. Note that all training demonstrations collected in this environment are initialized such that the robot’s end-effector is positioned directly above the target object in the beginning of the episode. (However, this is not the case across all environments in the BridgeData V2 dataset; in some other environments, the robot is initialized farther away from the target object, so it must horizontally reach for the object first before manipulating it.)

原始 BridgeData V2 数据集在这个水槽环境中包含以下七个任务的示教:“把锅翻正”“把胡萝卜放到盘子上”“把杯子从台面(或沥水架)放进水槽”“把茄子放进锅里”“把刀放到砧板上”“把勺子放进锅里”和“把拉杆转到竖直朝前”。原始数据集中所有这些任务的样例图像见图 8。注意,在这个环境中采集的所有训练示教,在回合开始时都把机器人末端执行器初始化在目标物体正上方。(不过,并不是 BridgeData V2 数据集的所有环境都如此;在其他一些环境中,机器人被初始化在离目标物体较远的位置,因此必须先水平伸向物体,再操作它。)

Figure 8

Figure 8: Original BridgeData V2 sink environment tasks. Images from sample demonstrations in the sink environment from the original BridgeData V2 dataset reveal that all demonstrations in this environment were initialized such that the robot’s end-effector was positioned immediately above the target object. Note that these initial states are different from the initial states we use in our BridgeData V2 evaluation tasks shown in Fig. 7. In our evaluations, we always initialize the robot’s end-effector to a fixed location above the sink, rather than positioning it directly above the target object (except for one task: “Put Eggplant into Pot (Easy Version)”).

图 8:原始 BridgeData V2 水槽环境中的任务。原始 BridgeData V2 数据集水槽环境中的示教样例图像显示,这个环境中所有示教的初始状态都是机器人末端执行器位于目标物体正上方。注意,这些初始状态与我们在图 7 所示的 BridgeData V2 评测任务中所用的初始状态不同。在我们的评测中,我们总是把机器人末端执行器初始化在水槽上方的一个固定位置,而不是直接位于目标物体上方(只有一个任务例外:“把茄子放进锅里(简单版)”)。

In our BridgeData V2 evaluation suite, only one task – “Put Eggplant into Pot (Easy Version”) – is initialized with the robot’s end-effector hovering directly over the target object; in all 16 other tasks, the end-effector is initialized at a fixed location above the sink such that the robot must horizontally reach towards the object. This initial condition, in combination with the distribution shifts we introduce in the various types of OOD generalization in our evaluation suite, challenges the generalist policies and requires a high degree of robustness in order to complete the tasks successfully. Hence, the success rates for policies like RT-1-X and Octo are lower than what is reported in prior works. However, we find that other policies such as RT-2-X and OpenVLA still achieve relatively strong performance despite all these distribution shifts and challenges.

在我们的 BridgeData V2 评测套件中,只有一个任务——“把茄子放进锅里(简单版)”——在初始化时让机器人末端执行器悬停在目标物体正上方;在其他全部 16 个任务中,末端执行器都被初始化在水槽上方的一个固定位置,机器人必须水平伸向物体。这种初始条件,加上我们在评测套件中各类 OOD 泛化里引入的分布偏移,对通用策略构成挑战,要成功完成任务需要很高的鲁棒性。因此,RT-1-X 和 Octo 等策略的成功率低于先前工作中报告的数字。不过我们发现,RT-2-X 和 OpenVLA 等其他策略尽管面临这些分布偏移和挑战,仍取得了相对强劲的性能。

B.1.3 Detailed BridgeData V2 Evaluation Results

See Table 4 for the full BridgeData V2 WidowX evaluation results. The number of successes for each method, out of 10 trials, is listed for each of 17 tasks. OpenVLA achieves strongest performance in the majority of the tasks and has the highest aggregate success rate among the generalist policies. RT-2-X also shows good performance, outperforming RT-1-X and Octo, though it does not perform as well as OpenVLA. RT-1-X and Octo generally experience difficulty in these generalization tasks.

B.1.3 BridgeData V2 详细评测结果

完整的 BridgeData V2 WidowX 评测结果见表 4。表中列出了每种方法在 17 个任务中每个任务 10 次试验里的成功次数。OpenVLA 在大多数任务上表现最强,在通用策略中综合成功率最高。RT-2-X 也表现不错,优于 RT-1-X 和 Octo,但不如 OpenVLA。RT-1-X 和 Octo 在这些泛化任务上普遍表现吃力。

Category Task # Trials RT-1-X # Successes Octo # Successes RT-2-X # Successes OpenVLA (ours) # Successes
Visual gen Put Eggplant into Pot (Easy Version) 10 1 5 7 10
Visual gen Put Eggplant into Pot 10 0 1 5 10
Visual gen Put Cup from Counter into Sink 10 1 1 0 7
Visual gen Put Eggplant into Pot (w/ Clutter) 10 1 3.5 6 7.5
Visual gen Put Yellow Corn on Pink Plate 10 1 4 8 9
Motion gen Lift Eggplant 10 3 0.5 6.5 7.5
Motion gen Put Carrot on Plate (w/ Height Change) 10 2 1 4.5 4.5
Physical gen Put Carrot on Plate 10 1 0 1 8
Physical gen Flip Pot Upright 10 2 6 5 8
Physical gen Lift AAA Battery 10 0 0 2 7
Semantic gen Move Skull into Drying Rack 10 1 0 5 5
Semantic gen Lift White Tape 10 3 0 0 1
Semantic gen Take Purple Grapes out of Pot 10 6 0 5 4
Semantic gen Stack Blue Cup on Pink Cup 10 0.5 0 5.5 4.5
Language grounding Put {Eggplant, Red Bottle} into Pot 10 2.5 4 8.5 7.5
Language grounding Lift {Cheese, Red Chili Pepper} 10 1.5 2.5 8.5 10
Language grounding Put {Blue Cup, Pink Cup} on Plate 10 5 5.5 8.5 9.5
Mean Success Rate 18.5±\pm±2.7% 20.0±\pm±2.6% 50.6±\pm±3.5% 70.6±\pm±3.2%

Table 4: Detailed BridgeData V2 WidowX evaluation results. We report performance on the full evaluation suite of 17 tasks (discussed in Section 5.1), including visual/motion/physical/semantic generalization tasks and language grounding tasks. Note that partial success (score of 0.5) is possible for some tasks; see Section B.1.1 for details. We find that OpenVLA performs best in most tasks and achieves highest performance overall, followed by RT-2-X. On the other hand, RT-1-X and Octo struggle in the evaluations, only getting 0–2 successes in several tasks. See Fig. 7 for illustrations of all tasks.

类别 任务 试验次数 RT-1-X 成功次数 Octo 成功次数 RT-2-X 成功次数 OpenVLA (ours) 成功次数
视觉泛化 把茄子放进锅里(简单版) 10 1 5 7 10
视觉泛化 把茄子放进锅里 10 0 1 5 10
视觉泛化 把杯子从台面放进水槽 10 1 1 0 7
视觉泛化 把茄子放进锅里(有杂物) 10 1 3.5 6 7.5
视觉泛化 把黄玉米放到粉色盘子上 10 1 4 8 9
运动泛化 举起茄子 10 3 0.5 6.5 7.5
运动泛化 把胡萝卜放到盘子上(高度变化) 10 2 1 4.5 4.5
物理泛化 把胡萝卜放到盘子上 10 1 0 1 8
物理泛化 把锅翻正 10 2 6 5 8
物理泛化 举起 AAA 电池 10 0 0 2 7
语义泛化 把骷髅放进沥水架 10 1 0 5 5
语义泛化 举起白色胶带 10 3 0 0 1
语义泛化 把紫葡萄从锅里拿出来 10 6 0 5 4
语义泛化 把蓝杯子叠到粉杯子上 10 0.5 0 5.5 4.5
语言落地 把 {茄子, 红瓶子} 放进锅里 10 2.5 4 8.5 7.5
语言落地 举起 {奶酪, 红辣椒} 10 1.5 2.5 8.5 10
语言落地 把 {蓝杯子, 粉杯子} 放到盘子上 10 5 5.5 8.5 9.5
平均成功率 18.5±\pm±2.7% 20.0±\pm±2.6% 50.6±\pm±3.5% 70.6±\pm±3.2%

表 4:BridgeData V2 WidowX 详细评测结果。我们报告在完整的 17 个任务评测套件(见第 5.1 节)上的性能,包括视觉/运动/物理/语义泛化任务和语言落地任务。注意部分任务可以获得部分成功(0.5 分),详见 B.1.1 节。我们发现 OpenVLA 在大多数任务上表现最好,总体性能最高,RT-2-X 次之。另一方面,RT-1-X 和 Octo 在评测中表现吃力,在若干任务上只有 0–2 次成功。所有任务的示意图见图 7。

Additionally, in Table 5, we provide the full evaluation results for the quantized inference experiments that were summarized in Table 2. For these evaluations, we test policies on 8 representative BridgeData V2 tasks spanning all task categories in the full evaluation suite.

此外,我们在表 5 中给出量化推理实验的完整评测结果,其汇总见表 2。这些评测在 8 个有代表性的 BridgeData V2 任务上测试策略,这些任务覆盖完整评测套件中的所有任务类别。

Category Task # Trials bfloat16 # Successes int8 # Successes int4 # Successes
Visual gen Put Eggplant into Pot (Easy Version) 10 9 7 9
Visual gen Put Eggplant into Pot 10 7 7 7
Visual gen Put Cup from Counter into Sink 10 5 3 7
Motion gen Lift Eggplant 10 6 4 7.5
Physical gen Put Carrot on Plate 10 6 5 7
Physical gen Lift AAA Battery 10 7 5 3
Semantic gen Take Purple Grapes out of Pot 10 8 8 9
Language grounding Put {Eggplant, Red Bottle} into Pot 10 9 7.5 8
Mean Success Rate 71.3 ±\pm± 4.8% 58.1 ±\pm± 5.1% 71.9 ±\pm± 4.7%

Table 5: Full quantized inference results. Here we present the detailed version of the results shown in Table 2.

类别 任务 试验次数 bfloat16 成功次数 int8 成功次数 int4 成功次数
视觉泛化 把茄子放进锅里(简单版) 10 9 7 9
视觉泛化 把茄子放进锅里 10 7 7 7
视觉泛化 把杯子从台面放进水槽 10 5 3 7
运动泛化 举起茄子 10 6 4 7.5
物理泛化 把胡萝卜放到盘子上 10 6 5 7
物理泛化 举起 AAA 电池 10 7 5 3
语义泛化 把紫葡萄从锅里拿出来 10 8 8 9
语言落地 把 {茄子, 红瓶子} 放进锅里 10 9 7.5 8
平均成功率 71.3 ±\pm± 4.8% 58.1 ±\pm± 5.1% 71.9 ±\pm± 4.7%

表 5:完整的量化推理结果。这里给出表 2 所示结果的详细版本。

B.2 Google Robot Evaluation Details

B.2 Google 机器人评测细节

In this section, we provide more details on the Google robot evaluations introduced in Section 5.1.

本节提供第 5.1 节介绍的 Google 机器人评测的更多细节。

Figure 9

B.2.1 Google Robot Evaluation Tasks

Figure 9: Google robot evaluation tasks. We evaluate every generalist robot policy on in-distribution tasks and out-of-distribution (OOD) generalization tasks. OOD tasks involve unseen backgrounds, target objects, instructions/object relations, and semantic concepts (e.g., photos from the Internet that do not appear in robot action data).

B.2.1 Google 机器人评测任务

图 9:Google 机器人评测任务。我们在分布内任务和分布外(OOD)泛化任务上评测每个通用机器人策略。OOD 任务涉及未见过的背景、目标物体、指令/物体关系,以及语义概念(例如机器人动作数据中没有出现过的互联网照片)。

On the Google robot, we evaluate each generalist robot policy on 12 tasks with 5 rollouts each, for a total of 60 rollouts. The first five tasks test on in-distribution conditions, and the last seven tasks test on more difficult out-of-distribution (OOD) conditions. All tasks are depicted in Fig. 9. Each rollout is marked as a failure (0) or success (1).

在 Google 机器人上,我们在 12 个任务上评测每个通用机器人策略,每个任务 5 次试验,共 60 次。前五个任务在分布内条件下测试,后七个任务在更难的分布外(OOD)条件下测试。所有任务见图 9。每次试验记为失败(0)或成功(1)。

We describe the 12 tasks below:

下面描述这 12 个任务:

  1. Pick Coke Can (in-distribution): The robot is positioned in front of a platform with a can of Coke on top of it. The robot’s goal is to grasp and lift the Coke can.
  2. Move Apple near Green Can (in-distribution): The robot is positioned in front of a platform with an apple and a green soda can on top of it. The robot’s goal is to grasp the apple and move it next to the green can.
  3. Move Blue Chip Bag near Apple (in-distribution): The robot is positioned in front of a platform with a blue bag of chips and an apple on top of it. The robot’s goal is to grasp the blue bag of chips and move it close to the apple.
  4. Place Coke Can Upright (in-distribution): The robot is positioned in front of a platform with a can of Coke on top of it, and the can is oriented horizontally on its side. The robot’s goal is to grasp the Coke can and orient it to be in a vertical position.
  5. Open Middle Drawer (in-distribution): The robot is positioned in front of a set of three drawers. The robot’s goal is to grasp the middle drawer handle and pull the drawer open.
  6. Move Orange near Brown Chip Bag (OOD): The robot is positioned in front of a platform with a brown bag of chips and an orange on top of it. A tablecloth with blue sky and white cloud patterns covers the platform underneath the objects. The robot’s goal is to grasp the orange and bring it next to the bag of chips. This task is OOD because the orange is an unseen object relative to the training dataset, and the tablecloth is an unseen background.⁷
  7. Pick Pepsi Can (OOD): The robot is positioned in front of a platform with a can of Pepsi on top of it. A tablecloth with bright yellow/brown patterns covers the platform underneath the can. The robot’s goal is to grasp and lift the can. This task is OOD because the Pepsi can is an unseen object, and the tablecloth is an unseen background.
  8. Pick Banana (OOD): The robot is positioned in front of a platform with an apple, a can of Coke, and a banana. The robot’s goal is to grasp and lift the banana. This task is OOD because the banana is an unseen target object.
  9. Pick Green Cup (OOD): The robot is positioned in front of a platform with a banana, a can of Pepsi, and a green cup. The robot’s goal is to grasp and lift the green cup. This task is OOD because all objects in the scene are unseen in the training data.
  10. Place Apple on Plate (OOD): The robot is positioned in front of a platform with a plate and an apple. The robot’s goal is to grasp the apple and move it onto the plate. This task is OOD because it is a novel instruction describing an unseen object relation: training demonstrations only cover moving the apple near the plate, rather than placing it on top of the plate.
  11. Place Banana in Pan (OOD): The robot is positioned in front of a platform with a pan and a banana. The robot’s goal is to grasp the banana and move it into the pan. This task is OOD because the banana is an unseen target object, and it is a novel instruction describing an unseen object relation, as explained in the previous task.
  12. Move Coke Can to Taylor Swift (OOD): The robot is positioned in front of a platform with a can of Coke and photos of three different celebrities, including Taylor Swift. The robot’s goal is to grasp the can and move it to the photo of Taylor Swift. This task is OOD because the photos of the celebrities are unseen in the robot interaction data.
  1. 拿起可乐罐(分布内):机器人位于一个平台前,平台上有一罐可乐。机器人的目标是抓住并举起可乐罐。
  2. 把苹果移到绿色罐子旁(分布内):机器人位于一个平台前,平台上有一个苹果和一个绿色汽水罐。机器人的目标是抓住苹果并把它移到绿色罐子旁边。
  3. 把蓝色薯片袋移到苹果旁(分布内):机器人位于一个平台前,平台上有一袋蓝色包装的薯片和一个苹果。机器人的目标是抓住蓝色薯片袋并把它移到苹果附近。
  4. 把可乐罐立起来(分布内):机器人位于一个平台前,平台上有一罐横躺着的可乐。机器人的目标是抓住可乐罐并把它调整为竖直放置。
  5. 打开中间的抽屉(分布内):机器人位于一组三个抽屉前。机器人的目标是抓住中间抽屉的把手,把抽屉拉开。
  6. 把橙子移到棕色薯片袋旁(OOD):机器人位于一个平台前,平台上有一袋棕色包装的薯片和一个橙子。物体下方的平台上铺着一块印有蓝天白云图案的桌布。机器人的目标是抓住橙子并把它拿到薯片袋旁边。这个任务属于 OOD,因为相对于训练数据集,橙子是未见过的物体,桌布是未见过的背景。⁷
  7. 拿起百事可乐罐(OOD):机器人位于一个平台前,平台上有一罐百事可乐。罐子下方的平台上铺着一块印有亮黄色/棕色图案的桌布。机器人的目标是抓住并举起罐子。这个任务属于 OOD,因为百事可乐罐是未见过的物体,桌布是未见过的背景。
  8. 拿起香蕉(OOD):机器人位于一个平台前,平台上有一个苹果、一罐可乐和一根香蕉。机器人的目标是抓住并举起香蕉。这个任务属于 OOD,因为香蕉是未见过的目标物体。
  9. 拿起绿色杯子(OOD):机器人位于一个平台前,平台上有一根香蕉、一罐百事可乐和一个绿色杯子。机器人的目标是抓住并举起绿色杯子。这个任务属于 OOD,因为场景中的所有物体在训练数据中都没有出现过。
  10. 把苹果放到盘子上(OOD):机器人位于一个平台前,平台上有一个盘子和一个苹果。机器人的目标是抓住苹果并把它移到盘子上。这个任务属于 OOD,因为这是一条描述未见过的物体关系的新指令:训练示教只涵盖把苹果移到盘子附近,而不是放到盘子上面。
  11. 把香蕉放进平底锅(OOD):机器人位于一个平台前,平台上有一个平底锅和一根香蕉。机器人的目标是抓住香蕉并把它移进平底锅。这个任务属于 OOD,因为香蕉是未见过的目标物体,而且如上一个任务所述,这是一条描述未见过的物体关系的新指令。
  12. 把可乐罐移到泰勒·斯威夫特那里(OOD):机器人位于一个平台前,平台上有一罐可乐和三位不同名人的照片,其中包括泰勒·斯威夫特。机器人的目标是抓住罐子并把它移到泰勒·斯威夫特的照片处。这个任务属于 OOD,因为这些名人照片在机器人交互数据中没有出现过。

⁷ See Appendix of Brohan et al. [7] for a detailed list of OOD conditions in Google robot evaluations.

⁷ Google 机器人评测中 OOD 条件的详细列表见 Brohan 等人 [7] 的附录。

B.2.2 Detailed Google Robot Evaluation Results

Category Task # Trials RT-1-X # Successes Octo # Successes RT-2-X # Successes OpenVLA (ours) # Successes
In-distribution Pick Coke Can 5 5 1 5 5
In-distribution Move Apple near Green Can 5 3 3 3 5
In-distribution Move Blue Chip Bag near Apple 5 0 3 4 5
In-distribution Place Coke Can Upright 5 0 0 4 4
In-distribution Open Middle Drawer 5 0 4 2 3
OOD Move Orange near Brown Chip Bag 5 1 2 5 5
OOD Pick Pepsi Can 5 3 0 5 4
OOD Pick Banana 5 5 3 5 5
OOD Pick Green Cup 5 1 0 5 5
OOD Place Apple on Plate 5 0 0 4 4
OOD Place Banana in Pan 5 0 0 2 4
OOD Move Coke Can near Taylor Swift 5 2 0 3 2
Mean Success Rate 33.3±\pm±6.1% 26.7±\pm±5.8% 78.3±\pm±5.4% 85.0±\pm±4.6%

Table 6: Detailed Google robot evaluation results. We report full evaluation results for Google robot evaluations discussed in Section 5.1. Each generalist policy is evaluated with 60 rollouts across 12 tasks, covering both in-distribution and out-of-distribution (OOD) testing conditions. In the bottom row, we report mean success rate ±\pm± StdErr for each policy. OpenVLA and RT-2-X both significantly outperform RT-1-X and Octo overall (we bold the mean success rate for both due to overlapping error bars). See Fig. 9 for illustrations of all tasks.

B.2.2 Google 机器人详细评测结果

类别 任务 试验次数 RT-1-X 成功次数 Octo 成功次数 RT-2-X 成功次数 OpenVLA (ours) 成功次数
分布内 拿起可乐罐 5 5 1 5 5
分布内 把苹果移到绿色罐子旁 5 3 3 3 5
分布内 把蓝色薯片袋移到苹果旁 5 0 3 4 5
分布内 把可乐罐立起来 5 0 0 4 4
分布内 打开中间的抽屉 5 0 4 2 3
分布外 把橙子移到棕色薯片袋旁 5 1 2 5 5
分布外 拿起百事可乐罐 5 3 0 5 4
分布外 拿起香蕉 5 5 3 5 5
分布外 拿起绿色杯子 5 1 0 5 5
分布外 把苹果放到盘子上 5 0 0 4 4
分布外 把香蕉放进平底锅 5 0 0 2 4
分布外 把可乐罐移到泰勒·斯威夫特附近 5 2 0 3 2
平均成功率 33.3±\pm±6.1% 26.7±\pm±5.8% 78.3±\pm±5.4% 85.0±\pm±4.6%

表 6:Google 机器人详细评测结果。我们报告第 5.1 节讨论的 Google 机器人评测的完整结果。每个通用策略在 12 个任务上评测 60 次,涵盖分布内和分布外(OOD)测试条件。最下一行报告每个策略的平均成功率 ±\pm± 标准误。OpenVLA 和 RT-2-X 总体上都显著优于 RT-1-X 和 Octo(由于误差棒重叠,我们把两者的平均成功率都加粗)。所有任务的示意图见图 9。

译注:表中最后一个任务写作“Move Coke Can near Taylor Swift”(把可乐罐移到泰勒·斯威夫特附近),与 B.2.1 节任务 12 的名称“Move Coke Can to Taylor Swift”略有出入,原文如此。

Full results for the Google robot evaluations are shown in Table 6. Overall, we find that RT-1-X and Octo experience difficulty on the evaluation tasks; they are often unable to achieve a single success out of five trials in several tasks. On the other hand, RT-2-X and OpenVLA demonstrate strong performance, completing every task at least two times out of five trials; these two VLA policies perform comparably with each other on this particular evaluation suite.

Google 机器人评测的完整结果见表 6。总体而言,我们发现 RT-1-X 和 Octo 在评测任务上表现吃力;在若干任务上,它们五次试验连一次都没能成功。另一方面,RT-2-X 和 OpenVLA 表现强劲,每个任务五次试验中至少完成两次;这两个 VLA 策略在这套评测上表现相当。

B.3 Data-Efficient Adaptation Experiment Details

B.3 数据高效适配实验细节

In this section, we provide more details on the data-efficient adaptation experiments discussed in Section 5.2, where we investigate the effectiveness of fine-tuned OpenVLA policies on new robot setups such as Franka-Tabletop and Franka-DROID.

本节提供第 5.2 节讨论的数据高效适配实验的更多细节,该实验研究微调后的 OpenVLA 策略在 Franka-Tabletop 和 Franka-DROID 等新机器人配置上的效果。

B.3.1 Franka-Tabletop and Franka-DROID Tasks

We collect 10–150 demonstrations of each of seven tasks. The first six tasks correspond to a robot setup which we denote as “Franka-Tabletop” (Franka Emika Panda robot mounted on top of a table), and the final task corresponds to a robot setup which we call “Franka-DROID”.

B.3.1 Franka-Tabletop 与 Franka-DROID 任务

我们为七个任务各采集 10–150 条示教。前六个任务对应我们称为“Franka-Tabletop”的机器人配置(安装在桌面上的 Franka Emika Panda 机器人),最后一个任务对应我们称为“Franka-DROID”的机器人配置。

In the Franka-Tabletop setup, the first three of six tasks correspond to single-instruction tasks and are narrow, while the last three tasks correspond to multi-instruction tasks in which multiple objects are present in the scene and the robot must manipulate the correct one depending on the language instruction.

在 Franka-Tabletop 配置中,六个任务里的前三个是单指令任务,范围较窄;后三个是多指令任务,场景中有多个物体,机器人必须根据语言指令操作正确的那个。

Figure 10

Figure 10: Franka-Tabletop fine-tuning tasks. Franka-Tabletop tasks used in the data-efficient adaptation experiments in Section 5.2 and described in detail in Fig. 10 are depicted above. The first three of six tasks, shown in the top three rows, only involve a single instruction, while the last three tasks in the bottom three rows involve multiple objects and instructions (the instructions specify the target object or target location). The first column shows sample initial states matching the training data distribution, while the second column shows out-of-distribution (OOD) initial states (e.g., unseen backgrounds, target objects, distractors, and object positions/orientations). Every policy in Section 5.2 is evaluated with 10–12 rollouts on in-distribution tasks and 5–6 rollouts on OOD tasks.

图 10:Franka-Tabletop 微调任务。上图展示了第 5.2 节数据高效适配实验中使用、并在图 10 中详细描述的 Franka-Tabletop 任务。六个任务中的前三个(上面三行)只涉及单条指令,后三个任务(下面三行)涉及多个物体和多条指令(指令指定目标物体或目标位置)。第一列展示与训练数据分布一致的示例初始状态,第二列展示分布外(OOD)初始状态(例如未见过的背景、目标物体、干扰物以及物体位置/朝向)。第 5.2 节中的每个策略在分布内任务上评测 10–12 次,在 OOD 任务上评测 5–6 次。

译注:图注中“在图 10 中详细描述”指向图 10 自身,原文如此;任务的文字描述实际在 B.3.1 节正文中。

Below we describe each of the six Franka-Tabletop tasks shown in Fig. 10:

下面描述图 10 所示的六个 Franka-Tabletop 任务:

  1. Put Carrot in Bowl (single-instruction): The robot’s goal is to grasp the carrot and place it into the bowl. We collect 50 demonstrations of this task for the training dataset, randomly placing the carrot and the bowl at different locations on the table in every episode. The carrot is always initialized on the left side of the bowl. During evaluation, each trial is recorded as a success (1) or failure (0); there is no partial credit.
  2. Pour Corn into Pot (single-instruction): The robot’s goal is to grasp the red bowl, move towards the steel pot, and pour the contents (a yellow corn) into the pot. We collect 50 demonstrations of this task for the training dataset, randomly placing the bowl and the pot at different locations on the table in every episode. The bowl is always initialized on the right side of the pot. During evaluation, each trial is recorded as a success (1) or failure (0); there is no partial credit.
  3. Flip Pot Upright (single-instruction): The robot’s goal is to grasp the steel pot (which is initially oriented vertically), rotate it to be in the upright position, and place it back onto the table. We collect only 10 demonstrations of this task for the training dataset, randomly placing the steel pot at various locations within a small section of the table. During evaluation, each trial is recorded as a success (1), failure (0), or partial success (0.5). Partial successes include grasping the pot but not orienting it upright, or knocking it over to the upright position but not carefully guiding it. The robot must release the pot at the end of the episode for full credit.
  4. Move <object> onto Plate (multi-instruction): The robot’s goal is to grasp one out of three objects (depending on the target specified in the language instruction) and place it on the plate on the right side of the table. We collect 150 demonstrations of this task for the training dataset, randomly placing different combinations of three objects on the table and selecting one as the target. The plate is always initialized on the right side of the table. During evaluation, each trial is recorded as a success (1), failure (0), or partial success (0.5). Partial success is recorded when the first object that the robot makes contact with is the correct target object (i.e., the object specified in the language instruction), but the robot does not complete the task.
  5. Knock <object> Over (multi-instruction): The robot’s goal is to approach one out of three objects (depending on the target specified in the language instruction) and push it until it falls over. We collect 70 demonstrations of this task for the training dataset, randomly placing different combinations of three objects on the table and selecting one as the target. During evaluation, each trial is recorded as a success (1), failure (0), or partial success (0.5). Partial success is recorded when the first object that the robot makes contact with is the correct target object (i.e., the object specified in the language instruction), but the robot does not complete the task.
  6. Cover <object> with Towel (multi-instruction): The robot’s goal is to grasp the blue towel and place it on one out of three objects (depending on the target specified in the language instruction). We collect 45 demonstrations of this task for the training dataset, randomly placing different combinations of three objects on the table. During evaluation, each trial is recorded as a success (1), failure (0), or partial success (0.5). Partial success is recorded when the first object that the robot touches with the towel is the correct target object (i.e., the object specified in the language instruction), but the robot does not complete the task (e.g., it drops the towel onto the table instead of on top of the target object). Full credit is given when any part of the towel is resting over the top surface of the target object, i.e., the object does not need to be fully covered.
  1. 把胡萝卜放进碗里(单指令):机器人的目标是抓住胡萝卜并放进碗里。我们为训练数据集采集了这个任务的 50 条示教,每个回合都把胡萝卜和碗随机放在桌上的不同位置。胡萝卜总是初始化在碗的左侧。评测时,每次试验记为成功(1)或失败(0),没有部分分数。
  2. 把玉米倒进锅里(单指令):机器人的目标是抓住红碗,移向钢锅,把里面的东西(一根黄玉米)倒进锅里。我们为训练数据集采集了这个任务的 50 条示教,每个回合都把碗和锅随机放在桌上的不同位置。碗总是初始化在锅的右侧。评测时,每次试验记为成功(1)或失败(0),没有部分分数。
  3. 把锅翻正(单指令):机器人的目标是抓住钢锅(初始时竖着侧立),把它转成正立,再放回桌上。我们为训练数据集只采集了这个任务的 10 条示教,把钢锅随机放在桌上一小块区域内的不同位置。评测时,每次试验记为成功(1)、失败(0)或部分成功(0.5)。部分成功包括抓住了锅但没有把它翻正,或者把锅碰倒成正立但没有小心地引导它。机器人必须在回合结束时松开锅才能得满分。
  4. 把 <object> 移到盘子上(多指令):机器人的目标是抓住三个物体中的一个(取决于语言指令指定的目标),放到桌子右侧的盘子上。我们为训练数据集采集了这个任务的 150 条示教,每次在桌上随机摆放三个物体的不同组合并选择其中一个作为目标。盘子总是初始化在桌子右侧。评测时,每次试验记为成功(1)、失败(0)或部分成功(0.5)。当机器人第一个接触的物体是正确的目标物体(即语言指令指定的物体)但没有完成任务时,记为部分成功。
  5. 把 <object> 推倒(多指令):机器人的目标是接近三个物体中的一个(取决于语言指令指定的目标),推它直到它倒下。我们为训练数据集采集了这个任务的 70 条示教,每次在桌上随机摆放三个物体的不同组合并选择其中一个作为目标。评测时,每次试验记为成功(1)、失败(0)或部分成功(0.5)。当机器人第一个接触的物体是正确的目标物体(即语言指令指定的物体)但没有完成任务时,记为部分成功。
  6. 用毛巾盖住 <object>(多指令):机器人的目标是抓住蓝色毛巾,把它盖到三个物体中的一个上(取决于语言指令指定的目标)。我们为训练数据集采集了这个任务的 45 条示教,每次在桌上随机摆放三个物体的不同组合。评测时,每次试验记为成功(1)、失败(0)或部分成功(0.5)。当机器人用毛巾第一个碰到的物体是正确的目标物体(即语言指令指定的物体)但没有完成任务时(例如把毛巾丢在桌上而不是目标物体上面),记为部分成功。只要毛巾的任何部分搭在目标物体的顶面上就给满分,也就是说物体不需要被完全盖住。

For every Franka-Tabletop task, we evaluate each method with 10–12 in-distribution trials and 5–6 OOD generalization trials. The in-distribution and OOD test conditions are depicted in Fig. 10 (second column).

对每个 Franka-Tabletop 任务,我们用 10–12 次分布内试验和 5–6 次 OOD 泛化试验评测每种方法。分布内和 OOD 测试条件见图 10(第二列)。

We describe the OOD test conditions for each of the six tasks below:

下面描述六个任务各自的 OOD 测试条件:

  1. Put Carrot in Bowl (OOD): An eggplant (unseen object) replaces the carrot.
  2. Pour Corn into Pot (OOD): An unseen brown tablecloth covers the tabletop.
  3. Flip Pot Upright (OOD): An unseen white tablecloth covers the tabletop
  4. Move <object> onto Plate (OOD): A set of three unseen objects are placed on the table.
  5. Knock <object> Over (OOD): Two unseen distractor objects (red plastic cup and brown box) are positioned behind the set of three seen objects.
  6. Cover <object> with Towel (OOD): The three objects on the table are placed upside-down and at unseen positions.
  1. 把胡萝卜放进碗里(OOD):用一根茄子(未见过的物体)代替胡萝卜。
  2. 把玉米倒进锅里(OOD):桌面上铺一块未见过的棕色桌布。
  3. 把锅翻正(OOD):桌面上铺一块未见过的白色桌布。
  4. 把 <object> 移到盘子上(OOD):桌上放一组三个未见过的物体。
  5. 把 <object> 推倒(OOD):在三个见过的物体后面放两个未见过的干扰物(红色塑料杯和棕色盒子)。
  6. 用毛巾盖住 <object>(OOD):桌上的三个物体倒扣放置,并处在未见过的位置。

Finally, in the Franka-DROID environment, we experiment with one task and variants of it: Wipe Table (see Fig. 11). In this task, the robot’s goal is to grab the brush and sweep all three small brown objects into the dustpan. We collect 70 demonstrations for this task for the training dataset, varying the positions of all the objects.

最后,在 Franka-DROID 环境中,我们用一个任务及其变体做实验:擦桌子(见图 11)。在这个任务中,机器人的目标是抓起刷子,把三个棕色小物体全部扫进簸箕。我们为训练数据集采集了这个任务的 70 条示教,每次改变所有物体的位置。

Figure 11

Figure 11: Franka-DROID fine-tuning task. The “Wipe Table” task shown here is the final task used in the data-efficient adaptation experiments in Section 5.2. The left image shows the initial conditions for an in-distribution trial. The right image shows an out-of-distribution trial in which unseen distractor objects are present on the table. To fully complete the task, the robot must grab the brush and sweep all three objects into the dustpan.

图 11:Franka-DROID 微调任务。这里展示的“擦桌子”任务是第 5.2 节数据高效适配实验中使用的最后一个任务。左图展示一次分布内试验的初始条件。右图展示一次分布外试验,桌上有未见过的干扰物体。要完全完成任务,机器人必须抓起刷子,把三个物体全部扫进簸箕。

At test time, we evaluate on in-distribution conditions matching the training data (Fig. 11, left), as well as out-of-distribution (OOD) conditions in which distractor objects are also present in the scene on the table (Fig. 11, right). Since there are various possible outcomes for each trial, we define a scoring rubric as follows: The maximum score for each trial is 2 points. The policy receives the full 2 points if the robot sweeps all three objects into the dustpan. It receives 1 point for successfully sweeping one or two objects into the dustpan. Otherwise, it receives 0 points. We evaluate each policy with 18 in-distribution trials and 12 OOD trials, so each policy receives an aggregate score out of 60 points.

测试时,我们既在与训练数据一致的分布内条件下评测(图 11 左),也在场景中桌上还有干扰物体的分布外(OOD)条件下评测(图 11 右)。由于每次试验可能有多种结果,我们定义如下评分标准:每次试验满分 2 分。如果机器人把三个物体全部扫进簸箕,策略得满分 2 分;成功把一个或两个物体扫进簸箕得 1 分;否则得 0 分。我们用 18 次分布内试验和 12 次 OOD 试验评测每个策略,因此每个策略的总分满分为 60 分。

B.3.2 Detailed Franka-Tabletop and Franka-DROID Evaluation Results

Full evaluation results for both Franka-Tabletop and Franka-DROID evaluations are shown in Table 7. We evaluate the methods discussed in Section 5.2. We find that Diffusion Policy demonstrates strong performance on the single-instruction Franka-Tabletop tasks (e.g., “Put Carrot in Bowl” and “Pour Corn in Pot”), outperforming other methods. However, OpenVLA and Octo achieve higher performance in the more diverse multi-instruction tasks (“Move <object> onto Plate”, “Knock <object> Over”, and “Cover <object> with Towel”). In the Franka-DROID environment, OpenVLA obtains best results. Overall, we find that OpenVLA achieves the highest average performance across both tasks.

B.3.2 Franka-Tabletop 与 Franka-DROID 详细评测结果

Franka-Tabletop 和 Franka-DROID 评测的完整结果见表 7。我们评测第 5.2 节讨论的方法。我们发现,Diffusion Policy 在单指令的 Franka-Tabletop 任务上(例如“把胡萝卜放进碗里”和“把玉米倒进锅里”)表现强劲,优于其他方法。然而在更多样化的多指令任务(“把 <object> 移到盘子上”“把 <object> 推倒”和“用毛巾盖住 <object>”)上,OpenVLA 和 Octo 性能更高。在 Franka-DROID 环境中,OpenVLA 取得了最好的结果。总体而言,我们发现 OpenVLA 在两类任务上的平均性能最高。

# trials Diffusion Policy Diffusion Policy (matched) Octo OpenVLA (scratch) OpenVLA (ours)
Franka-Tabletop (5Hz) “Put Carrot in Bowl” (in-distribution) 10 90.0% 80.0% 40.0% 70.0% 70.0%
“Put Carrot in Bowl” (OOD) 5 20.0% 0.0% 20.0% 0.0% 40.0%
“Pour Corn into Pot” (in-distribution) 10 100.0% 90.0% 0.0% 10.0% 50.0%
“Pour Corn into Pot” (OOD) 5 80.0% 60.0% 0.0% 20.0% 60.0%
“Flip Pot Upright” (in-distribution) 10 100.0% 85.0% 40.0% 85.0% 100.0%
“Flip Pot Upright” (OOD) 5 50.0% 20.0% 0.0% 40.0% 80.0%
“Move <object> onto Plate” (in-distribution) 12 25.0% 25.0% 41.7% 8.3% 75.0%
“Move <object> onto Plate” (OOD) 6 8.3% 33.3% 8.3% 33.3% 58.3%
“Knock <object> Over” (in-distribution) 12 33.3% 25.0% 83.3% 75.0% 75.0%
“Knock <object> Over” (OOD) 6 16.7% 16.7% 33.3% 58.3% 83.3%
“Cover <object> with Towel” (in-distribution) 12 16.7% 20.8% 91.7% 41.7% 50.0%
“Cover <object> with Towel” (OOD) 6 16.7% 33.3% 91.7% 50.0% 50.0%
Average 48.5±\pm±4.9% 43.4±\pm±4.7% 43.4±\pm±4.4% 43.4±\pm±4.6% 67.2±\pm±4.0%
Franka-DROID (15Hz) “Wipe Table” (in-distribution) 18 50.0% 27.8% 52.8% 25.0% 55.6%
“Wipe Table” + Distractors (OOD) 12 12.5% 25.0% 16.7% 16.7% 62.5%
Average 35.0±\pm±8.0% 26.7±\pm±7.5% 38.3±\pm±8.5% 21.7±\pm±6.6% 58.3±\pm±7.2%

Table 7: Detailed data-efficient adaptation experiment results. Here we present the full breakdown of results summarized in Fig. 5. We report the performance of Diffusion Policy trained from scratch on new robot tasks, as well as generalist policies fine-tuned on the same data. Each policy is tested against both in-distribution and out-of-distribution (OOD) generalization conditions (see Fig. 10 for Franka-Tabletop tasks and Fig. 11 for Franka-DROID tasks). We find that no single policy performs best on all tasks: Diffusion Policy achieves high success rates on single-instruction tasks, while OpenVLA and Octo performs well on diverse multi-instruction tasks. In terms of aggregate performance, however, OpenVLA obtains the highest average success rate across both environments.

试验次数 Diffusion Policy Diffusion Policy(matched) Octo OpenVLA(scratch) OpenVLA(本文)
Franka-Tabletop (5Hz) “把胡萝卜放进碗里”(分布内) 10 90.0% 80.0% 40.0% 70.0% 70.0%
“把胡萝卜放进碗里”(OOD) 5 20.0% 0.0% 20.0% 0.0% 40.0%
“把玉米倒进锅里”(分布内) 10 100.0% 90.0% 0.0% 10.0% 50.0%
“把玉米倒进锅里”(OOD) 5 80.0% 60.0% 0.0% 20.0% 60.0%
“把锅翻正”(分布内) 10 100.0% 85.0% 40.0% 85.0% 100.0%
“把锅翻正”(OOD) 5 50.0% 20.0% 0.0% 40.0% 80.0%
“把 <object> 移到盘子上”(分布内) 12 25.0% 25.0% 41.7% 8.3% 75.0%
“把 <object> 移到盘子上”(OOD) 6 8.3% 33.3% 8.3% 33.3% 58.3%
“把 <object> 推倒”(分布内) 12 33.3% 25.0% 83.3% 75.0% 75.0%
“把 <object> 推倒”(OOD) 6 16.7% 16.7% 33.3% 58.3% 83.3%
“用毛巾盖住 <object>”(分布内) 12 16.7% 20.8% 91.7% 41.7% 50.0%
“用毛巾盖住 <object>”(OOD) 6 16.7% 33.3% 91.7% 50.0% 50.0%
平均 48.5±\pm±4.9% 43.4±\pm±4.7% 43.4±\pm±4.4% 43.4±\pm±4.6% 67.2±\pm±4.0%
Franka-DROID (15Hz) “擦桌子”(分布内) 18 50.0% 27.8% 52.8% 25.0% 55.6%
“擦桌子” + 干扰物 (OOD) 12 12.5% 25.0% 16.7% 16.7% 62.5%
平均 35.0±\pm±8.0% 26.7±\pm±7.5% 38.3±\pm±8.5% 21.7±\pm±6.6% 58.3±\pm±7.2%

表 7:数据高效适配实验详细结果。这里给出图 5 所汇总结果的完整细分。我们报告在新机器人任务上从零训练的 Diffusion Policy,以及在相同数据上微调的通用策略的性能。每个策略都在分布内和分布外(OOD)泛化条件下测试(Franka-Tabletop 任务见图 10,Franka-DROID 任务见图 11)。我们发现没有哪一个策略在所有任务上都最好:Diffusion Policy 在单指令任务上成功率很高,而 OpenVLA 和 Octo 在多样化的多指令任务上表现良好。不过就综合性能而言,OpenVLA 在两个环境中的平均成功率都是最高的。

Additionally, in Table 8, we show the detailed version of the parameter-efficient fine-tuning experiment results summarized in Table 1. In these experiments, we use a representative subset of two Franka-Tabletop tasks, with both in-distribution and OOD variants: one narrow single-instruction task (“Put Carrot in Bowl”) and one diverse multi-instruction task (“Move <object> onto Plate”). We use the same number of training demonstrations used in Section 5.2 (50 and 150, respectively), which is delineated in Section B.3.1.

此外,我们在表 8 中给出表 1 所汇总的参数高效微调实验结果的详细版本。在这些实验中,我们使用两个 Franka-Tabletop 任务组成的代表性子集,每个任务都有分布内和 OOD 两种变体:一个是狭窄的单指令任务(“把胡萝卜放进碗里”),一个是多样化的多指令任务(“把 <object> 移到盘子上”)。我们使用与第 5.2 节相同数量的训练示教(分别为 50 条和 150 条),具体见 B.3.1 节。

# trials Full FT Last layer only Frozen vision Sandwich LoRA, r=32 LoRA, r=64
Franka-Tabletop (5Hz) “Put Carrot in Bowl” (in-distribution) 10 90.0 40.0 40.0 90.0 60.0 90.0
“Put Carrot in Bowl” (OOD) 5 40.0 0.0 40.0 0.0 60.0 40.0
“Move <object> onto Plate” (in-distribution) 12 79.2 33.3 50.0 75.0 75.0 62.5
“Move <object> onto Plate” (OOD) 6 41.7 33.3 58.3 41.7 75.0 66.7
Average 69.7±\pm±7.2% 30.3±\pm±6.1% 47.0±\pm±6.9% 62.1±\pm±7.9% 68.2±\pm±7.5% 68.2±\pm±7.8%

Table 8: Detailed parameter-efficient fine-tuning experiment results. Here we present the detailed task performance results summarized in Table 1.

试验次数 全参数微调 仅最后一层 冻结视觉编码器 三明治微调 LoRA,r=32 LoRA,r=64
Franka-Tabletop (5Hz) “把胡萝卜放进碗里”(分布内) 10 90.0 40.0 40.0 90.0 60.0 90.0
“把胡萝卜放进碗里”(OOD) 5 40.0 0.0 40.0 0.0 60.0 40.0
“把 <object> 移到盘子上”(分布内) 12 79.2 33.3 50.0 75.0 75.0 62.5
“把 <object> 移到盘子上”(OOD) 6 41.7 33.3 58.3 41.7 75.0 66.7
平均 69.7±\pm±7.2% 30.3±\pm±6.1% 47.0±\pm±6.9% 62.1±\pm±7.9% 68.2±\pm±7.5% 68.2±\pm±7.8%

表 8:参数高效微调实验详细结果。这里给出表 1 所汇总的各任务详细性能结果。

Appendix C RT-2-X vs. OpenVLA in BridgeData V2 Evaluations

附录 C BridgeData V2 评测中的 RT-2-X 与 OpenVLA

In this section, we provide additional details on RT-2-X vs. OpenVLA comparisons in BridgeData V2 evaluations discussed in Section 5.1. As discussed previously, OpenVLA is pretrained on a larger subset of OpenX data than RT-2-X and uses a fused SigLIP-DinoV2 vision backbone rather than a single visual encoder. However, in addition to these factors, we believe that OpenVLA’s significant improvement upon RT-2-X specifically in BridgeData V2 evaluations (as shown in Fig. 3) also stems from more careful preprocessing of the Bridge dataset.

本节补充第 5.1 节讨论的 BridgeData V2 评测中 RT-2-X 与 OpenVLA 比较的更多细节。如前所述,OpenVLA 在比 RT-2-X 更大的 OpenX 数据子集上预训练,并使用融合的 SigLIP-DinoV2 视觉骨干而非单一视觉编码器。不过,除了这些因素之外,我们认为 OpenVLA 在 BridgeData V2 评测中相对 RT-2-X 的显著提升(如图 3 所示)还源于对 Bridge 数据集更仔细的预处理。

During the development of the OpenVLA model, we discovered that the original version of the BridgeData V2 dataset contained many transitions with all-zero (no-op) actions. For instance, in every demonstration, an all-zero action was recorded as the ground-truth action in the first timestep. Consequently, training a highly expressive VLA model on the original dataset without any data preprocessing led to a policy that frequently predicted all-zero actions and froze during evaluations. Therefore, we simply filtered out the first transition in every demonstration when training the OpenVLA model, and this was sufficient for mitigating the freezing behavior in most cases.

在开发 OpenVLA 模型的过程中,我们发现原始版本的 BridgeData V2 数据集包含许多全零(无操作,no-op)动作的转移。例如,每条示教的第一个时间步都把一个全零动作记录为真值动作。因此,不做任何数据预处理、直接在原始数据集上训练一个表达力很强的 VLA 模型,会得到一个经常预测全零动作、在评测中卡住不动的策略。于是,我们在训练 OpenVLA 模型时简单地过滤掉每条示教的第一个转移,这在大多数情况下足以缓解卡住的行为。

However, the RT-2-X model was trained without such data preprocessing, so it often suffers the aforementioned freezing behavior if deployed out of the box without modifying the model querying procedure – which severely deteriorates rollout performance. Since this is a proprietary model that is infeasible for us to re-train (e.g., with our preprocessed version of the BridgeData V2 dataset), we mitigated this issue by simply querying the second-most-likely action from the model, since the first-most-likely action was often all zeros while the second-most-likely action was not. (Note that this is the same workaround that was applied by the developers of the RT-2-X model for BridgeData V2 evaluations reported in the Open X-Embodiment experiments [1].) This workaround led to much stronger RT-2-X performance on BridgeData V2 evaluations – though we believe that it is still suboptimal compared to re-training the model on the preprocessed version of the dataset.

然而,RT-2-X 模型训练时没有做这种数据预处理,所以如果不修改模型的查询方式就开箱部署,它常常出现上述卡住的行为——这会严重损害试验表现。由于这是一个专有模型,我们无法重新训练它(例如用我们预处理过的 BridgeData V2 数据集),我们缓解这一问题的办法是直接从模型中查询概率第二高的动作,因为概率最高的动作常常是全零,而概率第二高的动作不是。(注意,这与 RT-2-X 模型的开发者在 Open X-Embodiment 实验 [1] 中报告 BridgeData V2 评测时所用的变通办法相同。)这一变通办法使 RT-2-X 在 BridgeData V2 评测上的表现强了很多——不过我们认为,与在预处理过的数据集上重新训练模型相比,这仍不是最优的。

We also tried to dynamically query RT-2-X, i.e., by first sampling the first-most-likely action and then sampling the second-most-likely action if the first one was all zeros. However, we empirically found that dynamic querying led to worse performance than simply querying the second-most-likely action at all times. We hypothesize that this is due to a change in the robot’s dynamics that arises from dynamic querying: pausing in the middle of a trajectory to re-query the model leads to slight interruptions in the robot’s movement due to non-neglible latency in the querying pipeline, and this leads to subtle performance degradation. Therefore, we report the performance of RT-2-X when always querying the second-most-likely action, as done in the Open X-Embodiment project [1].

我们也尝试过动态地查询 RT-2-X,即先采样概率最高的动作,如果它是全零,再采样概率第二高的动作。然而我们在实验中发现,动态查询的性能比始终查询概率第二高的动作更差。我们推测这是因为动态查询改变了机器人的动力学:由于查询流水线存在不可忽略的延迟,在轨迹中途停下来重新查询模型会让机器人的运动出现轻微中断,从而导致细微的性能下降。因此,我们报告的是 RT-2-X 始终查询概率第二高动作时的性能,与 Open X-Embodiment 项目 [1] 的做法一致。

Appendix D Additional Experiments and Ablations

附录 D 补充实验与消融

In this section, we conduct several additional experiments to analyze the effects of individual components of the OpenVLA model architecture and training scheme, as well as provide quantitative evidence for claims made in earlier sections of this work. We aim to answer the following questions:

本节我们进行若干补充实验,分析 OpenVLA 模型架构和训练方案中各个组件的作用,并为本文前面各节的论断提供定量证据。我们希望回答以下问题:

  1. How important is OpenX training and how does it impact OpenVLA’s performance (Section D.1)?
  2. What effect does using a fused SigLIP-DinoV2 vision encoder have on OpenVLA’s performance, compared to using a SigLIP-only vision encoder (Section D.2)?
  3. Is it better to fine-tune or freeze the vision encoder in OpenVLA (Section D.3)?
  4. How do the quantized inference results discussed in Section 5.3 change when policy performance is disentangled from model inference speed (Section D.4)?
  1. OpenX 训练有多重要?它如何影响 OpenVLA 的性能(D.1 节)?
  2. 与仅用 SigLIP 的视觉编码器相比,使用融合的 SigLIP-DinoV2 视觉编码器对 OpenVLA 的性能有什么影响(D.2 节)?
  3. 在 OpenVLA 中,微调视觉编码器好还是冻结它好(D.3 节)?
  4. 当把策略性能与模型推理速度分离开时,第 5.3 节讨论的量化推理结果会如何变化(D.4 节)?

译注:量化推理实验实际在第 5.4 节,此处及 D.4 节开头的“第 5.3 节”原文如此。

We discuss the experimental setup and results addressing each of the above questions sequentially in the following sections.

下面各节依次讨论回答上述问题的实验设置和结果。

D.1 OpenX Training Data Ablation Experiments

D.1 OpenX 训练数据消融实验

As discussed in Section 3.3, OpenVLA is trained on a large dataset of robot embodiments, scenes, and tasks from the Open X-Embodiment dataset [1] (OpenX). In this section, we ablate the OpenX mixture and train a VLA policy solely on one robot dataset, to assess the impact of OpenX training on policy performance. Note that we have already observed the negative effect of ablating OpenX training in the fine-tuning regime, as discussed in Section 5.2 (see OpenVLA (Scratch)), but we discuss additional experiments on another robot embodiment in this section to provide more supporting evidence.

如第 3.3 节所述,OpenVLA 在来自 Open X-Embodiment 数据集 [1](OpenX)、涵盖多种机器人本体、场景和任务的大型数据集上训练。本节我们消融 OpenX 配比,只在一个机器人数据集上训练 VLA 策略,以评估 OpenX 训练对策略性能的影响。注意,我们已经在微调场景中观察到去掉 OpenX 训练的负面影响,见第 5.2 节的讨论(见 OpenVLA (Scratch)),但本节讨论在另一种机器人本体上的补充实验,以提供更多支持证据。

Experimental setup and tasks. We compare the original OpenVLA model with OpenVLA-Bridge, which is produced by taking the same pretrained VLM as OpenVLA (Prismatic VLM [44]) and fine-tuning it solely on BridgeData V2 [6] rather than the entire OpenX training mixture discussed in Appendix A. We evaluate OpenVLA and OpenVLA-Bridge on a subset of 8 representative tasks from the BridgeData V2 WidowX robot evaluation suite discussed in Section B.1.1. The tasks are listed in Table 9.

实验设置与任务。我们把原始 OpenVLA 模型与 OpenVLA-Bridge 比较:后者取与 OpenVLA 相同的预训练 VLM(Prismatic VLM [44]),只在 BridgeData V2 [6] 上微调,而不是在附录 A 所述的完整 OpenX 训练配比上微调。我们在 B.1.1 节所述 BridgeData V2 WidowX 机器人评测套件中选出 8 个有代表性的任务组成的子集上评测 OpenVLA 和 OpenVLA-Bridge。任务列于表 9。

Results. Results for the OpenX training mixture ablation are shown in Table 9. By comparing OpenVLA with OpenVLA-Bridge, we see that performance drops drastically (reduction of 30 percent in absolute success rate), which demonstrates the importance of OpenX pretraining on final policy performance. Although the language grounding performance is not impacted, we observe performance reduction across all generalization categories. This result suggests that the large diversity of scenes, objects, and tasks in the OpenX training mixture is essential for unlocking improved generalization capabilities in the OpenVLA model.

结果。OpenX 训练配比的消融结果见表 9。比较 OpenVLA 和 OpenVLA-Bridge 可以看到性能急剧下降(绝对成功率降低 30 个百分点),这表明 OpenX 预训练对最终策略性能的重要性。虽然语言落地性能没有受到影响,但我们观察到所有泛化类别上的性能都有所下降。这一结果表明,OpenX 训练配比中场景、物体和任务的高度多样性,对于释放 OpenVLA 模型更强的泛化能力至关重要。

Category Task # Trials OpenVLA # Successes OpenVLA-Bridge # Successes OpenVLA-Bridge-SigLIP # Successes
Visual gen Put Eggplant into Pot (Easy Version) 10 10 8 8
Visual gen Put Eggplant into Pot 10 10 2 3
Visual gen Put Cup from Counter into Sink 10 7 4 2
Motion gen Lift Eggplant 10 7.5 5.5 6.5
Physical gen Put Carrot on Plate 10 8 4 1
Physical gen Lift AAA Battery 10 7 2 2
Semantic gen Take Purple Grapes out of Pot 10 4 3 3
Language grounding Put {Eggplant, Red Bottle} into Pot 10 7.5 8 7
Mean Success Rate 76.3 ±\pm± 4.8% 45.6 ±\pm± 5.6% 40.6 ±\pm± 5.5%

Table 9: BridgeData V2 WidowX ablation experiment results. We evaluate various methods on a subset of 8 representative tasks to assess the importance of different components of the OpenVLA model architecture and training scheme. OpenVLA-Bridge is a version of OpenVLA without OpenX training (it is trained only on BridgeData V2), and OpenVLA-Bridge-SigLIP additionally ablates the fused vision backbone by removing the DinoV2 encoder (its vision backbone only consists of the SigLIP encoder). We observe that both OpenX training and the fused vision encoder improve policy performance, though the former has a much greater effect than the latter.

类别 任务 试验次数 OpenVLA 成功次数 OpenVLA-Bridge 成功次数 OpenVLA-Bridge-SigLIP 成功次数
视觉泛化 把茄子放进锅里(简单版) 10 10 8 8
视觉泛化 把茄子放进锅里 10 10 2 3
视觉泛化 把杯子从台面放进水槽 10 7 4 2
运动泛化 举起茄子 10 7.5 5.5 6.5
物理泛化 把胡萝卜放到盘子上 10 8 4 1
物理泛化 举起 AAA 电池 10 7 2 2
语义泛化 把紫葡萄从锅里拿出来 10 4 3 3
语言落地 把 {茄子, 红瓶子} 放进锅里 10 7.5 8 7
平均成功率 76.3 ±\pm± 4.8% 45.6 ±\pm± 5.6% 40.6 ±\pm± 5.5%

表 9:BridgeData V2 WidowX 消融实验结果。我们在 8 个有代表性的任务组成的子集上评测多种方法,以评估 OpenVLA 模型架构和训练方案中不同组件的重要性。OpenVLA-Bridge 是没有经过 OpenX 训练的 OpenVLA 版本(只在 BridgeData V2 上训练),OpenVLA-Bridge-SigLIP 则进一步消融融合视觉骨干,去掉了 DinoV2 编码器(其视觉骨干只由 SigLIP 编码器构成)。我们观察到 OpenX 训练和融合视觉编码器都能提升策略性能,但前者的作用比后者大得多。

D.2 Dual vs. Single Vision Encoder Experiments

D.2 双视觉编码器与单视觉编码器实验

The OpenVLA model architecture consists of a fused vision backbone that combines the SigLIP [9] and DinoV2 [25] encoders. In this section, we ablate the DinoV2 component to assess the importance of using a dual vision encoder.

OpenVLA 的模型架构包含一个融合 SigLIP [9] 和 DinoV2 [25] 编码器的视觉骨干。本节我们消融 DinoV2 组件,以评估使用双视觉编码器的重要性。

Experimental setup and tasks. We instantiate a model, OpenVLA-Bridge-SigLIP, which is a version of OpenVLA that is trained only on BridgeData V2 and consists of only the SigLIP encoder as the vision backbone. We compare this model with the OpenVLA-Bridge model discussed in the previous section (Section D.1), which shares the same model architecture as the original OpenVLA model and is only trained on Bridge robot data. Therefore, the only difference between OpenVLA-Bridge-SigLIP and OpenVLA-Bridge is that the former omits the DinoV2 encoder in the vision backbone. We evaluate these models on the same subset of 8 Bridge tasks described in the previous section.

实验设置与任务。我们实例化一个模型 OpenVLA-Bridge-SigLIP,它是只在 BridgeData V2 上训练、视觉骨干只有 SigLIP 编码器的 OpenVLA 版本。我们把它与上一节(D.1 节)讨论的 OpenVLA-Bridge 模型比较,后者与原始 OpenVLA 模型架构相同,只在 Bridge 机器人数据上训练。因此,OpenVLA-Bridge-SigLIP 与 OpenVLA-Bridge 的唯一区别是前者在视觉骨干中省略了 DinoV2 编码器。我们在上一节所述的同一组 8 个 Bridge 任务上评测这些模型。

Results. Results for the dual vision encoder ablation are shown in Table 9. The drop in performance from OpenVLA-Bridge to OpenVLA-Bridge-SigLIP implies that additionally including the DinoV2 encoder in the vision backbone improves policy performance. However, the 5 percent reduction in performance here is not as significant as the 30 percent drop in performance observed from ablating OpenX training. The low-level spatial features represented in DinoV2 appear to aid generalization in only some cases.

结果。双视觉编码器的消融结果见表 9。从 OpenVLA-Bridge 到 OpenVLA-Bridge-SigLIP 的性能下降说明,在视觉骨干中额外加入 DinoV2 编码器能提升策略性能。不过,这里 5 个百分点的性能下降,远不如去掉 OpenX 训练时观察到的 30 个百分点那样显著。DinoV2 所表示的低层空间特征似乎只在部分情况下有助于泛化。

D.3 Fine-Tuned vs. Frozen Vision Encoder Experiments

D.3 微调视觉编码器与冻结视觉编码器实验

As discussed in Section 3.4, prior work on VLMs observed higher performance from freezing the vision encoder than fine-tuning its parameters [44]. However, when training OpenVLA, we fine-tuned all 7B parameters in the model, including the SigLIP-DinoV2 vision backbone, as we discovered early on during development that fine-tuning the vision encoder led to higher-performing VLAs — a finding which held across various pretrained VLMs and model architectures. We discuss details of such findings below.

如第 3.4 节所述,先前关于 VLM 的工作观察到冻结视觉编码器比微调其参数性能更高 [44]。然而在训练 OpenVLA 时,我们微调了模型的全部 70 亿参数,包括 SigLIP-DinoV2 视觉骨干,因为我们在开发早期就发现,微调视觉编码器能得到性能更高的 VLA——这一发现在多种预训练 VLM 和模型架构上都成立。下面讨论这些发现的细节。

Experimental setup and tasks. In this section, we report the performance of two VLA policies produced by fine-tuning two different pretrained models from the Prismatic VLMs [44] repository on BridgeData V2. The two pretrained models are named SigLIP ViT-SO 224px and LLaVa v1.5 7B (Reproduction); see Karamcheti et al. [44] for details on their architectures and training mixtures. We evaluate both policies on various Bridge tasks shown in Table 10. Note that the evaluation configurations here differ from previously discussed Bridge evaluations, so the results are not directly comparable to results in other similar experiments.

实验设置与任务。本节报告两个 VLA 策略的性能,它们分别由 Prismatic VLMs [44] 仓库中的两个不同预训练模型在 BridgeData V2 上微调得到。这两个预训练模型名为 SigLIP ViT-SO 224px 和 LLaVa v1.5 7B (Reproduction);它们的架构和训练配比详见 Karamcheti 等人 [44]。我们在表 10 所示的多个 Bridge 任务上评测这两个策略。注意,这里的评测配置与前面讨论的 Bridge 评测不同,因此结果不能与其他类似实验的结果直接比较。

Results. Results for the fine-tuned vs. frozen vision encoder experiments are shown in Table 10. We find that for both VLAs tested, fine-tuning the vision encoder leads to significantly higher success rates across various tasks. Qualitatively, in some cases, deploying the frozen vision encoder policies leads to unstable robot behaviors that are clearly suboptimal. Consequently, we decided early on during development to not conduct further experimentation with frozen vision encoders.

结果。微调与冻结视觉编码器的实验结果见表 10。我们发现,对两个被测 VLA 来说,微调视觉编码器都能在多个任务上带来显著更高的成功率。定性地看,在某些情况下,部署冻结视觉编码器的策略会导致机器人行为不稳定,明显不理想。因此,我们在开发早期就决定不再对冻结视觉编码器做进一步实验。

Task # Trials SigLIP ViT-SO 224px: Frozen Vision # Successes SigLIP ViT-SO 224px: Fine-Tuned # Successes LLaVa v1.5 7B (Reproduction): Frozen Vision # Successes LLaVa v1.5 7B (Reproduction): Fine-Tuned # Successes
Put Eggplant into Pot 10 7 10 5 9
Put Corn on Plate 10 10 9 0 9
Mean Success Rate 85 95 25 90
Put { Eggplant, Red Bottle } into Pot 4 2 4 – 3
Put { Blue Cup, Pink Cup } on Plate 4 0 0 – 0
Lift { Cheese, Red Chili Pepper } 4 0 3 – 2
Put { Strawberry, Lime } into Pot 4 1 0 – 3
Move { Sushi, Grapes } 4 3 4 – 3
Mean Success Rate 30 55 – 55

Table 10: Fine-tuned vs. frozen vision encoder experiment results. We evaluate the performance of fine-tuning (“Fine-Tuned”) vs. freezing the vision encoder (“Frozen Vision”) in two VLA policies built on top of two different pretrained VLMs from the Prismatic VLMs [44] repository. BridgeData V2 WidowX tasks shown here are performed in the same sink environment used for other Bridge experiments in this work (however, the initial environment configurations here differ, as these evaluations were conducted at an earlier stage in the project). We find that fine-tuning the vision encoder is crucial to obtain good policy performance. Certain frozen vision encoder evaluations were discontinued due to very poor (near-zero) performance and unstable robot behaviors. Among the evaluations where both frozen vision and fine-tuned approaches are tested, fine-tuning the vision encoder leads to 80.0% average success versus 46.7% average success from leaving it frozen.

任务 试验次数 SigLIP ViT-SO 224px:冻结视觉编码器 成功次数 SigLIP ViT-SO 224px:微调 成功次数 LLaVa v1.5 7B (Reproduction):冻结视觉编码器 成功次数 LLaVa v1.5 7B (Reproduction):微调 成功次数
把茄子放进锅里 10 7 10 5 9
把玉米放到盘子上 10 10 9 0 9
平均成功率 85 95 25 90
把 { 茄子, 红瓶子 } 放进锅里 4 2 4 – 3
把 { 蓝杯子, 粉杯子 } 放到盘子上 4 0 0 – 0
举起 { 奶酪, 红辣椒 } 4 0 3 – 2
把 { 草莓, 青柠 } 放进锅里 4 1 0 – 3
移动 { 寿司, 葡萄 } 4 3 4 – 3
平均成功率 30 55 – 55

表 10:微调与冻结视觉编码器的实验结果。我们在两个 VLA 策略中评测微调视觉编码器(“微调”)与冻结视觉编码器(“冻结视觉编码器”)的性能,这两个策略分别建立在 Prismatic VLMs [44] 仓库中两个不同的预训练 VLM 之上。这里的 BridgeData V2 WidowX 任务在本文其他 Bridge 实验所用的同一个水槽环境中进行(不过初始环境配置不同,因为这些评测是在项目较早阶段进行的)。我们发现微调视觉编码器对获得良好的策略性能至关重要。部分冻结视觉编码器的评测因性能极差(接近零)且机器人行为不稳定而中止。在同时测试了冻结和微调两种方式的评测中,微调视觉编码器的平均成功率为 80.0%,而保持冻结的平均成功率为 46.7%。

译注:原表为两级表头(SigLIP ViT-SO 224px 与 LLaVa v1.5 7B (Reproduction) 各占两列),此处合并为一级表头;原文两组任务各有一行“平均成功率”。

D.4 Additional Quantized Inference Experiments: Disentangling Policy Performance and Model Inference Speed

D.4 补充量化推理实验:把策略性能与模型推理速度分离开

In Section 5.3, we evaluated OpenVLA with different levels of precision at inference time: half precision (bfloat16), 8-bit quantization, and 4-bit quantization. 8-bit quantization led to lower BridgeData V2 performance relative to the other two approaches, and we hypothesized that the reduction in performance was caused by lower model inference speed from the operations used in 8-bit quantization. In this section, we conduct experiments to assess the veracity of this claim.

在第 5.3 节中,我们用不同的推理精度评测了 OpenVLA:半精度(bfloat16)、8 比特量化和 4 比特量化。8 比特量化在 BridgeData V2 上的性能低于另外两种方式,我们推测性能下降是由 8 比特量化所用的运算导致模型推理速度降低造成的。本节我们通过实验来检验这一说法是否属实。

Specifically, we evaluate OpenVLA again with the three different levels of precision listed above, but now with blocking control. In other words, each action is fully executed on the robot before the next one is predicted by the policy and executed by the controller. This scheme controls system dynamics across methods with varying amounts of latency and thus allows us to test the quality of a policy’s action predictions, independent of its prediction speed. Effectively, the precision levels that have higher throughput – bfloat16 and 4-bit quantization – are forced to run slower to match the dynamics observed when deploying OpenVLA with 8-bit precision. Therefore, we expect OpenVLA’s performance with 8-bit precision to match the performance of bfloat16 and 4-bit precision under blocking control.

具体来说,我们再次用上述三种精度评测 OpenVLA,但这次使用阻塞控制。也就是说,每个动作都在机器人上完全执行完毕,策略才预测下一个动作并交给控制器执行。这种方式在延迟各不相同的方法之间控制住了系统动力学,因此可以独立于预测速度来检验策略动作预测的质量。实际上,吞吐更高的精度——bfloat16 和 4 比特量化——被迫放慢运行,以匹配以 8 比特精度部署 OpenVLA 时观察到的动力学。因此我们预期,在阻塞控制下,OpenVLA 以 8 比特精度运行的性能会与 bfloat16 和 4 比特精度相当。

Experimental setup and tasks. We report the performance of OpenVLA with blocking control and quantized inference on the same subset of 8 BridgeData V2 tasks used in Section D.1 and Section D.2.

实验设置与任务。我们在 D.1 节和 D.2 节所用的同一组 8 个 BridgeData V2 任务上,报告 OpenVLA 在阻塞控制和量化推理下的性能。

Results. Quantized inference experiment results with blocking control are shown in Table 11. Unlike in Table 2, where 8-bit quantization led to the worst rollout performance due to low inference speed, here we observe that 8-bit quantization performs comparably to bfloat16 precision and 4-bit quantization given that we evaluate with blocking control to remove the influence of varying inference speeds on task performance. This confirms our hypothesis about the effect of inference speed on 8-bit quantization performance in previous experiments (when using non-blocking control). We also see no substantial performance degradation when using the lowest precision, 4-bit, as also observed in Section 5.3.

结果。阻塞控制下的量化推理实验结果见表 11。在表 2 中,8 比特量化因推理速度低而试验表现最差;而这里由于我们用阻塞控制评测,消除了推理速度差异对任务性能的影响,我们观察到 8 比特量化的表现与 bfloat16 精度和 4 比特量化相当。这证实了我们关于先前实验中(使用非阻塞控制时)推理速度影响 8 比特量化性能的假设。我们还看到,使用最低的 4 比特精度时没有明显的性能下降,这与第 5.3 节的观察一致。

Category Task # Trials bfloat16 # Successes int8 # Successes int4 # Successes
Visual gen Put Eggplant into Pot (Easy Version) 10 10 10 10
Visual gen Put Eggplant into Pot 10 9 10 10
Visual gen Put Cup from Counter into Sink 10 5 5 3
Motion gen Lift Eggplant 10 8 7 7.5
Physical gen Put Carrot on Plate 10 10 10 10
Physical gen Lift AAA Battery 10 3 6 4
Semantic gen Take Purple Grapes out of Pot 10 2 2 2
Language grounding Put {Eggplant, Red Bottle} into Pot 10 9 9.5 8.5
Mean Success Rate 70.0 ±\pm± 5.1% 74.4 ±\pm± 4.9% 68.8 ±\pm± 5.2%

Table 11: Quantized inference experiment results with blocking control. We report the success rate and standard error of OpenVLA on various BridgeData V2 WidowX tasks with bfloat16 precision (the default approach), 8-bit quantization (int8), and 4-bit quantization (int4) at inference time. All average success rates have overlapping error bars, which suggests that all methods perform comparably.

类别 任务 试验次数 bfloat16 成功次数 int8 成功次数 int4 成功次数
视觉泛化 把茄子放进锅里(简单版) 10 10 10 10
视觉泛化 把茄子放进锅里 10 9 10 10
视觉泛化 把杯子从台面放进水槽 10 5 5 3
运动泛化 举起茄子 10 8 7 7.5
物理泛化 把胡萝卜放到盘子上 10 10 10 10
物理泛化 举起 AAA 电池 10 3 6 4
语义泛化 把紫葡萄从锅里拿出来 10 2 2 2
语言落地 把 {茄子, 红瓶子} 放进锅里 10 9 9.5 8.5
平均成功率 70.0 ±\pm± 5.1% 74.4 ±\pm± 4.9% 68.8 ±\pm± 5.2%

表 11:阻塞控制下的量化推理实验结果。我们报告 OpenVLA 在多个 BridgeData V2 WidowX 任务上,推理时分别使用 bfloat16 精度(默认方式)、8 比特量化(int8)和 4 比特量化(int4)的成功率和标准误。所有平均成功率的误差棒都相互重叠,表明各方法的表现相当。

Appendix E LIBERO Simulation Experiments

附录 E LIBERO 仿真实验

Our previous discussions in Section 5.2 and Section 5.3 focused on adapting OpenVLA to novel real-world robot setups and tasks. This section explores adapting OpenVLA to simulated robot setups and tasks, specifically utilizing the LIBERO benchmark [116]. Our experimentation in simulation offers two key advantages:

前面第 5.2 节和第 5.3 节的讨论集中于把 OpenVLA 适配到新的真实世界机器人配置和任务。本节探索把 OpenVLA 适配到仿真机器人配置和任务,具体使用 LIBERO 基准 [116]。仿真实验有两个关键优势:

  1. Demonstration of versatility: We show that OpenVLA, despite having been pretrained exclusively on real-world robot data, can effectively adapt to simulated domains, overcoming potential disparities between real-world and simulated environments and dynamics.
  2. Enhanced accessibility and reproducibility: Integration of OpenVLA into a publicly available simulation platform makes our model more accessible to other researchers, especially those who may not have access to robotic hardware. Additionally, simulated experiments are more easily reproduced than their real-world counterparts.
  1. 展示通用性:我们表明,OpenVLA 尽管完全在真实世界机器人数据上预训练,仍能有效适配仿真领域,克服真实环境与仿真环境及其动力学之间可能存在的差异。
  2. 提高可及性和可复现性:把 OpenVLA 集成到公开可用的仿真平台中,使其他研究者更容易使用我们的模型,尤其是那些没有机器人硬件的研究者。此外,仿真实验比真实世界实验更容易复现。

We discuss the experimental setup in Section E.1 and the results in Section E.2. We release the materials required to reproduce the experiments along with the OpenVLA codebase.

我们在 E.1 节讨论实验设置,在 E.2 节讨论结果。复现这些实验所需的材料随 OpenVLA 代码库一起发布。

E.1 LIBERO Simulation Experimental Setup

E.1 LIBERO 仿真实验设置

Simulation setup and tasks. The LIBERO benchmark [116] consists of four task suites designed for studying lifelong learning in robotic manipulation, and the original paper therefore investigates both forward and backward transfer to a variety of tasks. In our experiments, we focus solely on supervised fine-tuning on the target task suite, measuring the performance of various policies trained via behavioral cloning on successful demonstrations of the tasks.

仿真设置与任务。LIBERO 基准 [116] 由四个任务套件组成,用于研究机器人操作中的终身学习,因此原论文同时研究了对多种任务的前向和后向迁移。在我们的实验中,我们只关注在目标任务套件上的监督微调,衡量通过行为克隆在任务成功示教上训练的各种策略的性能。

We perform experiments with the following four task suites, which each contain 10 tasks with 50 human-teleoperated demonstrations each:

我们用以下四个任务套件做实验,每个套件包含 10 个任务,每个任务有 50 条人类遥操作示教:

  • LIBERO-Spatial consists of the same set of objects but different layouts, and tests the model’s understanding of spatial relationships.
  • LIBERO-Object consists of the same scene layouts but different objects, and tests the model’s understanding of object types.
  • LIBERO-Goal consists of the same objects and layouts but different task goals, and tests the model’s knowledge of different task-oriented behaviors.
  • LIBERO-Long (also called LIBERO-10) consists of long-horizon tasks with diverse objects, layouts, and tasks.
  • LIBERO-Spatial 由同一组物体、不同布局构成,测试模型对空间关系的理解。
  • LIBERO-Object 由相同的场景布局、不同物体构成,测试模型对物体类型的理解。
  • LIBERO-Goal 由相同的物体和布局、不同的任务目标构成,测试模型对不同任务导向行为的掌握。
  • LIBERO-Long(也称 LIBERO-10)由包含多样物体、布局和任务的长时程任务构成。

We make the following modifications to each of the training datasets above:

我们对上述每个训练数据集做如下修改:

  1. To accommodate methods requiring higher-resolution images (such as 256×256256\times 256256×256px or 224×224224\times 224224×224px), we regenerate all demonstrations at an increased resolution of 256×256256\times 256256×256px. Originally, the dataset provided by the benchmark consists of 128×128128\times 128128×128px images. We find that simply upscaling these images to 256×256256\times 256256×256px results in poor image quality. Therefore, we choose to begin with higher-resolution images, which can be downscaled as necessary, ensuring higher image quality across various resolution requirements. These higher-resolution images were obtained by stepping through the simulation environments with the actions stored in the provided human-collected demonstrations and saving the images rendered by the simulator.
  2. We filter out all “no-op” actions from the dataset, i.e., actions that have near-zero magnitude in the translation and rotation components and do not change the state of the robot’s gripper. We find that this simple data cleaning step is crucial for highly expressive single-step policies such as OpenVLA, which otherwise learn to imitate these no-op actions and consequently freeze indefinitely at certain states during evaluation.
  3. We rotate all third-person images at both train and test time by 180 degrees because we observe that the LIBERO environments return images that are upside down on our hardware.
  4. Since we train policies via imitation learning, which expects demonstrations to be successful, we replay all demonstrations in the corresponding simulation environments and filter out the demonstrations that fail to complete the task (as determined by the environments’ success criteria). As a result, we remove 68 of 500 LIBERO-Spatial demonstrations, 46 of 500 LIBERO-Object demonstrations, 72 of 500 LIBERO-Goal demonstrations, and 121 of 500 LIBERO-Long demonstratinos.
  5. For all methods in our comparisons, we only utilize the static third-person camera images; we do not use the wrist camera images that are additionally provided in the original datasets. This is for sake of having fair comparisons, as OpenVLA’s visual inputs only consist of third-person camera images.
  1. 为了适应需要更高分辨率图像的方法(例如 256×256256\times 256256×256px 或 224×224224\times 224224×224px),我们以更高的 256×256256\times 256256×256px 分辨率重新生成所有示教。基准原本提供的数据集由 128×128128\times 128128×128px 的图像组成。我们发现,简单地把这些图像放大到 256×256256\times 256256×256px 会导致图像质量很差。因此我们选择从更高分辨率的图像开始,需要时再缩小,从而在各种分辨率需求下都保证较高的图像质量。这些高分辨率图像的获取方式是:用所提供的人类采集示教中存储的动作逐步运行仿真环境,并保存仿真器渲染出的图像。
  2. 我们从数据集中过滤掉所有“无操作”动作,即平移和旋转分量的幅度接近零、且不改变机器人夹爪状态的动作。我们发现,这一简单的数据清洗步骤对 OpenVLA 这种表达力很强的单步策略至关重要;否则它们会学着模仿这些无操作动作,在评测中于某些状态无限期地卡住。
  3. 我们在训练和测试时都把所有第三人称图像旋转 180 度,因为我们观察到在我们的硬件上,LIBERO 环境返回的图像是上下颠倒的。
  4. 由于我们通过模仿学习训练策略,而模仿学习要求示教是成功的,我们在对应的仿真环境中回放所有示教,过滤掉没能完成任务的示教(以环境的成功判据为准)。结果,我们去掉了 LIBERO-Spatial 500 条示教中的 68 条、LIBERO-Object 500 条中的 46 条、LIBERO-Goal 500 条中的 72 条,以及 LIBERO-Long 500 条中的 121 条。
  5. 对比较中的所有方法,我们只使用固定的第三人称相机图像;不使用原始数据集中额外提供的腕部相机图像。这是为了比较公平,因为 OpenVLA 的视觉输入只有第三人称相机图像。

Comparisons. The methods that we compare include Diffusion Policy⁸ [3] trained from scratch, Octo [5] fine-tuned on the target dataset, and OpenVLA fine-tuned on the target dataset via LoRA (r=32r=32r=32) as described in Section 5.3. Each policy is trained independently on each of the task suites above (rather than training a single policy on all four suites combined). All policies are trained with the same set of demonstrations, so all methods benefit from the data cleaning steps described above.

对比方法。我们比较的方法包括从零训练的 Diffusion Policy⁸ [3]、在目标数据集上微调的 Octo [5],以及如第 5.3 节所述、通过 LoRA(r=32r=32r=32)在目标数据集上微调的 OpenVLA。每个策略在上述每个任务套件上独立训练(而不是在四个套件合并的数据上训练单一策略)。所有策略都用同一组示教训练,因此所有方法都受益于上述数据清洗步骤。

⁸ We use the implementation of Diffusion Policy that is described in the DROID dataset paper [11], which conditions action generation on DistilBERT [117] language embeddings of the task label.

⁸ 我们使用 DROID 数据集论文 [11] 中描述的 Diffusion Policy 实现,它以任务标签的 DistilBERT [117] 语言嵌入作为动作生成的条件。

Evaluation details. To ensure lower variance in the experimental results, all methods are evaluated across 500 trials for each task suite, and the reported performance is the average success rate over three random seeds (resulting in 1500 total trials per statistic). Although we modify the training datasets, as described earlier, we do not change the test environments but rather use the same initial environment configurations provided by the original LIBERO benchmark.

评测细节。为了降低实验结果的方差,所有方法在每个任务套件上都评测 500 次,报告的性能是三个随机种子上的平均成功率(因此每个统计量共 1500 次试验)。虽然我们如前所述修改了训练数据集,但我们没有改动测试环境,而是使用原始 LIBERO 基准提供的同一套初始环境配置。

E.2 LIBERO Simulation Experimental Results

E.2 LIBERO 仿真实验结果

We present the LIBERO experimental results in Table 12. Importantly, we observe that OpenVLA can be effectively adapted to tasks in the LIBERO simulation environments, as it obtains highest average success rate and rank among the tested methods. However, we find that the overall margin between OpenVLA and the other methods are tighter here than in the real-world fine-tuning experiments discussed in Section 5.2. We attribute this to the fact that OpenVLA was pretrained with purely real-world robot data and no simulation data, which suggests that fine-tuning the model on simulated robot tasks may not be as effective as fine-tuning it on real-world tasks due to the domain gap between simulated and real-world environments and dynamics. We see evidence for this notion in the results obtained by Octo – another policy pretrained on large amounts of real-world robot data – which also only achieves a small boost in overall performance relative to a simple, strong baseline such as Diffusion Policy trained from scratch. We expect increased gains in performance for the pretrained and fine-tuned methods if simulation data is added to the pretraining data mixture.

LIBERO 实验结果见表 12。重要的是,我们观察到 OpenVLA 能有效适配 LIBERO 仿真环境中的任务,在被测方法中取得最高的平均成功率和平均排名。不过我们发现,这里 OpenVLA 与其他方法之间的总体差距比第 5.2 节讨论的真实世界微调实验中更小。我们把这归因于 OpenVLA 只用真实世界机器人数据预训练、没有使用仿真数据,这表明由于仿真与真实环境及其动力学之间存在领域差异,在仿真机器人任务上微调该模型可能不如在真实世界任务上微调有效。Octo 的结果为这一看法提供了证据:它是另一个在大量真实世界机器人数据上预训练的策略,相对于从零训练的 Diffusion Policy 这种简单而强的基线,总体性能也只有小幅提升。我们预期,如果在预训练数据配比中加入仿真数据,预训练后再微调的方法会获得更大的性能提升。

Method Spatial SR (↑\uparrow↑) Spatial Rank (↓\downarrow↓) Object SR (↑\uparrow↑) Object Rank (↓\downarrow↓) Goal SR (↑\uparrow↑) Goal Rank (↓\downarrow↓) Long SR (↑\uparrow↑) Long Rank (↓\downarrow↓) Average SR (↑\uparrow↑) Average Rank (↓\downarrow↓)
Diffusion Policy from scratch 78.3 ±\pm± 1.1% 3 92.5 ±\pm± 0.7% 1 68.3 ±\pm± 1.2% 3 50.5 ±\pm± 1.3% 3 72.4 ±\pm± 0.7% 2.5
Octo fine-tuned 78.9 ±\pm± 1.0% 2 85.7 ±\pm± 0.9% 3 84.6 ±\pm± 0.9% 1 51.1 ±\pm± 1.3% 2 75.1 ±\pm± 0.6% 2
OpenVLA fine-tuned (ours) 84.7 ±\pm± 0.9% 1 88.4 ±\pm± 0.8% 2 79.2 ±\pm± 1.0% 2 53.7 ±\pm± 1.3% 1 76.5 ±\pm± 0.6% 1.5

Table 12: LIBERO simulation benchmark results. We report the success rate (SR) and standard error of each method for the four task suites in the LIBERO benchmark, averaged over three random seeds with 500 trials each. In addition, we show the ranking of each method within each task suite, where a rank of 1 indicates the strongest method in the suite and a rank of 3 indicates the weakest method. (The average ranking is important to note since it informs which method may be most suitable to use as a default for a variety of tasks; it is more informative than the average success rate, which is not normalized by individual task suite difficulty.) Overall, we find that fine-tuned OpenVLA achieves highest average success rate and rank, followed by fine-tuned Octo and then Diffusion Policy trained from scratch.

方法 Spatial 成功率(↑\uparrow↑) Spatial 排名(↓\downarrow↓) Object 成功率(↑\uparrow↑) Object 排名(↓\downarrow↓) Goal 成功率(↑\uparrow↑) Goal 排名(↓\downarrow↓) Long 成功率(↑\uparrow↑) Long 排名(↓\downarrow↓) 平均成功率(↑\uparrow↑) 平均排名(↓\downarrow↓)
从零训练的 Diffusion Policy 78.3 ±\pm± 1.1% 3 92.5 ±\pm± 0.7% 1 68.3 ±\pm± 1.2% 3 50.5 ±\pm± 1.3% 3 72.4 ±\pm± 0.7% 2.5
微调后的 Octo 78.9 ±\pm± 1.0% 2 85.7 ±\pm± 0.9% 3 84.6 ±\pm± 0.9% 1 51.1 ±\pm± 1.3% 2 75.1 ±\pm± 0.6% 2
微调后的 OpenVLA(本文) 84.7 ±\pm± 0.9% 1 88.4 ±\pm± 0.8% 2 79.2 ±\pm± 1.0% 2 53.7 ±\pm± 1.3% 1 76.5 ±\pm± 0.6% 1.5

表 12:LIBERO 仿真基准结果。我们报告各方法在 LIBERO 基准四个任务套件上的成功率(SR)和标准误,取三个随机种子的平均,每个种子 500 次试验。此外,我们给出每种方法在各任务套件内的排名,排名 1 表示该套件中最强的方法,排名 3 表示最弱的方法。(平均排名值得重视,因为它说明哪种方法最适合作为各类任务的默认选择;它比平均成功率更有参考价值,因为平均成功率没有按各任务套件的难度做归一化。)总体而言,我们发现微调后的 OpenVLA 取得最高的平均成功率和最好的平均排名,其次是微调后的 Octo,再次是从零训练的 Diffusion Policy。

译注:原表为两级表头(每个任务套件下分 SR 与 Rank 两列),此处合并为一级表头,套件名省略了“LIBERO-”前缀。