机器人 · Robotics  |  arXiv 2024

What Matters in Building Vision-Language-Action Models for Generalist Robots

大规模系统研究:构建通用机器人 VLA 模型的三大关键要素
Xinghang Li, Peiyan Li, Long Qian, Minghuan Liu, Dong Wang, Jirong Liu, Bingyi Kang, Xiao Ma, Xinlong Wang, Di Guo, Tao Kong, Hanbo Zhang, Huaping Liu

本文通过超过 600 组精心设计的实验,系统研究了构建 Vision-Language-Action (VLA) 模型的三大核心决策:VLM backbone 的选择、策略架构的制定方式,以及跨机体数据的整合策略,并基于实证结果提炼出最优设计准则,发布了 RoboVLMs 框架与配套数据集。

arXiv 2412.14058 600+ 实验 8+ VLM Backbones 📄 arXiv:2412.14058 🌐 Project Page
VLA Vision-Language-Action 机器人操作 VLM backbone policy architecture cross-embodiment RoboVLMs 策略学习

01 Motivation

大型视觉语言模型(VLM)为通用机器人策略带来了新的可能,但如何将其转化为高效的 VLA 模型,业界缺乏系统性的实证研究——现有工作各自选择不同的 backbone、架构和数据策略,难以进行公平比较,也无法给出可推广的设计原则。

"We observe a significant gap between the performance of VLAs and expected performance of generalist robots, while the community lacks a systematic study covering all key design factors."
teaser
图 1:学习型机器人策略的分类与比较。左侧展示了从 imitation learning 到 VLA 的演进路径;右侧展示了本文研究覆盖的三大设计维度——VLM backbone、架构制定方式(formulation)、跨机体数据(cross-embodiment data),并标注各 design choice 对最终性能的影响程度。
600+精心设计的对照实验
8+VLM backbone 横向对比
4策略架构类型
+30.3%5-task 连续执行绝对提升(vs. GR-1)

研究聚焦三个核心问题:

ingredients
图 2:构建 VLA 的三大关键要素。研究系统性地覆盖 backbone 选择、架构制定与跨机体数据整合三个维度,每个维度均设有多个对比变量。

02 Method

RoboVLMs 是一个高度模块化的实验框架,支持自由组合各类 VLM backbone 与策略架构。研究在统一的数据集和评测基准(CALVIN 仿真 + SimplerEnv + Kinova Gen3 真机)下进行所有对比实验,确保结论可靠。

architecture
图 3:VLA 架构的四种制定方式。(a) One-step Continuous:单步预测连续动作序列;(b) One-step Discrete:将动作离散化为 token,以文本生成方式预测;(c) Interleaved:在 observation token 与动作 token 之间交替编码历史信息;(d) Policy-head:VLM 提供单步特征表示,额外的 policy head(RNN/Transformer/Diffusion)建模历史动作依赖。其中 MoE(Mixture of Experts)结构可集成到各类架构中。

VLM Backbone 选择

实验涵盖 LLaVA、Flamingo、KosMos(8B)、PaliGemma(3B)等 8 种以上 backbone。研究发现,在大规模视觉-语言数据上充分预训练的模型(KosMos、PaliGemma)在机器人任务中表现"distinctively better",而参数规模并不是决定性因素——3B 的 PaliGemma 与 8B 的 KosMos 表现相当。

策略架构制定

动作空间使用 7 维向量(6-DoF gripper pose + open/close 状态),连续动作归一化至 [-1,1],离散动作均匀划分为 256 bins。四种架构中,policy-head + continuous action 在 CALVIN 上达到最高平均完成长度(Avg. Len. 4.49),同时在 zero-shot 泛化中展现出最强鲁棒性。训练目标方面,Flow Matching "slightly outperforms MSE+BCE in all experiments",但差距不显著。MoE 结构在 zero-shot 设置下有助于泛化,但在已见场景中无额外增益。

跨机体数据整合策略

研究区分了 co-training(同时使用跨机体数据和领域内数据)与 post-training(先在跨机体数据上预训练,再用领域内数据微调)两种策略。实验表明,co-training 单独使用收益有限,而 post-training 在少样本(few-shot)场景中带来显著提升:单任务成功率提高 17.2%,平均完成任务数多 0.25。

03 Experiments

实验在 CALVIN 仿真基准(ABC→D zero-shot 泛化,最多 5 步连续执行)、SimplerEnv(WidowX+Bridge、Google Robot)以及真实 Kinova Gen3 机械臂(105 个操作任务,74K 轨迹)上全面评测。

CALVIN 基准——Backbone 对比(Policy-head + Continuous Action)

Backbone1-task2-task3-task4-task5-taskAvg. Len.
LLaVA0.8730.6780.5060.3760.2752.71
Flamingo0.9640.8960.8240.7400.6624.09
PaliGemma (3B)0.9840.9330.8880.8350.7794.42
KosMos (8B)0.9670.9300.8990.8650.8264.49

SimplerEnv 对比——Google Robot

方法Pick CokeMove NearOpen/CloseOverall
RT-1 (Converged)0.9600.9000.7300.630
OpenVLA-7b0.2700.0300.3560.219
RoboVLMs (Ours)1.0000.9100.5440.818

训练目标与 MoE 消融(PaliGemma,CALVIN Split ABC→D)

训练目标执行范式MoE1-task5-taskAvg. Len.
Flow MatchingChunk0.9400.5973.84
Flow MatchingChunk0.9100.5733.68
MSE+BCEChunk0.9330.6884.04
Flow MatchingFirst0.8980.5443.56
results
图 4:RoboVLMs 在仿真与真实场景的整体实验结果,涵盖 CALVIN 多步执行、SimplerEnv 多任务以及真实 Kinova 机械臂的五类评测设置。最优 VLA 配置在所有真实世界评测中均达到最强表现,并展现了基线模型所不具备的自我纠错能力(self-correction)。

消融:数据规模效应

ablation cross-embodiment
图 5:跨机体训练配置的消融。不同的 co-training / post-training 组合对 Bridge 环境少样本学习的影响:post-training 使单任务成功率从 44% 提升至 50%(Bridge),少样本成功率提升 17.2%。

数据规模消融(KosMos Policy-head)显示:从 0.1× 扩展至 1× CALVIN 数据,5-task 成功率从 17.6% 跃升至 82.6%;进一步扩展至 5× 仅带来边际增益(82.6% → 83.0%),说明更大的 VLM 具备更强的数据效率。

04 Limitations

Note: 本文没有专门的 Limitations 章节。以下各点均为从设计与实验中推断(inferred from the design),非作者明确陈述。
Interleaved 架构仅适用于 decoder-only 模型

由于 attention masking 机制的限制,interleaved 历史建模只能应用于 decoder-only 结构的 VLM,不适用于 encoder-decoder 类模型,限制了该架构与更多 backbone 的组合可能。

跨机体数据收益在 full fine-tune 场景有限

实验显示,cross-embodiment pre-training 的增益主要体现在 few-shot 场景;在充足的领域内数据下,co-training 单独使用"does not consistently yield significant improvements in final performance",说明跨机体数据的价值主要体现在数据稀缺时。

真实机器人测试局限于单一平台

真实世界验证仅在 7-DoF Kinova Gen3(配 Robotiq 夹爪)上进行,结论能否迁移至其他机械臂、移动底盘或双臂系统尚未验证。

计算成本随历史长度增长

Policy-head 与 interleaved 架构在处理长历史序列时计算开销显著增加,实际部署中的实时性约束未作充分讨论。