Search papers, labs, and topics across Lattice.
This paper introduces APT, a two-stage training method designed to enhance the instruction generalization of Vision-Language-Action (VLA) models by addressing the imbalance in training data. By pretraining the action expert as a language-agnostic Vision-Action prior and subsequently integrating language tokens through a gated fusion mechanism, APT effectively mitigates the noise introduced by imbalanced data. Experimental results demonstrate that APT significantly improves performance on out-of-distribution language instructions and compositional tasks across various VLA architectures.
APT achieves substantial improvements in instruction generalization for VLA models by effectively decoupling language and action learning, addressing a critical data imbalance issue.
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/
