通过预训练动作专家提升视觉语言动作模型的指令泛化能力
APT: Action Expert Pretraining Improves Instruction Generalization of Vision-Language-Action Policies

- 先用视觉-动作数据预训练动作专家,再引入语言信息融合
- 在未见过的指令和组合任务上显著提升泛化性能
- 适合需要强指令泛化的机器人操作研究者
将预训练视觉-语言模型(VLM)与连续动作专家结合的视觉-语言-动作(VLA)模型虽在操控任务中表现优异,但在分布外(OOD)语言指令上的泛化能力仍较差。根源在于数据结构失衡:语言内容远不如视觉和动作丰富,导致策略易依赖视觉捷径。现有离散动作方法可通过视觉-语言协同训练缓解此问题,但连续动作专家从随机初始化开始,完全依赖失衡数据学习,产生噪声梯度,损害VLM语言能力。本文从贝叶斯视角出发,将策略分解为语言无关的视觉-动作(VA)先验与语言条件的VLA似然,并提出APT两阶段训练方法,强调动作专家预训练。第一阶段,在冻结VLM的基础上,仅用视觉-动作对预训练动作专家作为VA先验;第二阶段,通过门控融合机制注入语言令牌,整合VLM特征的同时保留学习到的视觉运动先验。APT适用于主流VLA架构,包括$π$和GR00T风格结构。大量实验证明,APT在未见指令和组合任务上均实现一致提升。
原文摘要 · Abstract (English)
Vision-Language-Action (VLA) models that couple pretrained Vision-Language Models (VLMs) with continuous action experts have achieved strong manipulation performance, yet generalization to out-of-distribution (OOD) language instructions remains poor. A known challenge is the structural imbalance in VLA data, where language is far less diverse than visual and action content, making policies prone to visual shortcuts. While discrete-action methods mitigate this through vision-language co-training, continuous action experts lack such protection: they start from random initialization and learn entirely from imbalanced data, producing noisy gradients that corrupt the VLM and fail to exploit its language capability. We address this from a Bayesian perspective, factorizing the policy into a language-agnostic Vision-Action (VA) prior and a language-conditioned VLA likelihood, and propose APT, a two-stage training method emphasizing Action expert PreTraining. In Stage 1, the action expert is pretrained as a VA prior on vision-action pairs from a frozen VLM, bypassing the language imbalance. In Stage 2, language tokens are injected through a gated fusion mechanism that integrates VLM features while preserving the learned visuomotor prior. APT applies to mainstream VLA architectures, including the $π$ and GR00T-style architectures. Comprehensive experiments validate that APT achieves consistent gains on unseen instructions and compositional tasks. Project Page: https://xukechun.github.io/papers/APT/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。