用中间训练提升视觉语言模型在机器人任务中的表现
EmbodiedMidtrain: Bridging the Gap between Vision-Language Models and Vision-Language-Action Models via Mid-training

- 通过轻量级接近度估计器筛选与机器人任务对齐的视觉语言数据
- 在三个机器人操作基准上显著提升性能,媲美专业训练模型
- 适合研究视觉语言动作模型与数据对齐的学者使用
视觉-语言-动作模型(VLAs)继承自视觉-语言模型(VLMs),但多数VLAs采用未经具身域适配的现成VLM,限制了下游性能。本文提出EmbodiedMidtrain,通过分析发现VLA数据分布紧凑且远离VLM整体分布,且对齐程度在不同数据源间差异显著。为此构建中段训练数据引擎,利用轻量可学习的接近度估计器从大规模VLM池中挑选最匹配VLA的数据,对VLM进行中段训练后再微调为下游VLA。在三个机器人操纵基准测试中,该方法在不同VLM主干网络上均持续提升性能,达到与专家级VLAs及更大规模、更长训练周期的现成VLM相当的效果。进一步分析显示,中段训练提供更强初始化,增益从早期即出现并随训练过程扩大。数据引擎同时捕捉数据集级和样本级对齐信号,更偏好空间推理而非文本中心任务,同时保持VLM数据多样性。代码、数据与模型将公开。
原文摘要 · Abstract (English)
Vision-Language-Action Models (VLAs) inherit their visual and linguistic capabilities from Vision-Language Models (VLMs), yet most VLAs are built from off-the-shelf VLMs that are not adapted to the embodied domain, limiting their downstream performance. In this work, we propose EmbodiedMidtrain to bridge the gap between VLMs and VLAs. We first characterize the data distribution gap between them, showing that VLA data occupy compact regions that are largely separated from the broader VLM distribution, while the degree of alignment varies substantially both across and within VLM data sources. Then, we build a mid-training data engine that leverages a lightweight learnable proximity estimator to select the most VLA-aligned candidates from a large VLM pool, and mid-trains the VLM on this curated mixture before downstream VLA fine-tuning. Experiments on three robot manipulation benchmarks show that mid-training consistently improves performance across different VLM backbones, achieving results competitive with expert VLAs and off-the-shelf VLMs trained with larger model scale and training budgets. Further analysis reveals that mid-training provides a stronger initialization for VLA fine-tuning, with gains emerging from the earliest steps and widening throughout training. Moreover, the data engine captures both dataset-level and sample-level alignment signals, favoring spatial reasoning over text-centric tasks while preserving the diversity of the VLM data. We will release all code, data and models for future research.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。