VGAS: Value-Guided Action-Chunk Selection for Few-Shot Vision-Language-Action Adaptation
Changhua Xu, Jie Lu, Junyu Xuan, En Yu
TL;DR
The paper tackles the brittleness of few-shot Vision-Language-Action (VLA) adaptation under scarce demonstrations, where near-miss geometric errors cause failures despite semantically plausible actions. It reframes adaptation as a generate-then-select problem, proposing VGAS, which uses a high-recall base VLA to propose action chunks and a geometry-aware Transformer critic (Q-Chunk-Former) to perform Best-of-$N$ selection. VGAS introduces Explicit Geometric Regularization (EGR) to prevent value landscape collapse in offline RL, and proves convergence properties of the chunked Expected-Max backup operator, while ensuring temporal consistency and fine-grained spatial ranking. Experiments on LIBERO show that VGAS substantially improves success rates and robustness over SFT and standard offline-RL baselines, with EGR providing the largest gains and the Transformer critic enabling precise geometry-grounded valuations. The approach offers a practical path for robust, data-efficient VLA adaptation, albeit with increased inference latency and evaluation limited to simulation so far.
Abstract
Vision--Language--Action (VLA) models bridge multimodal reasoning with physical control, but adapting them to new tasks with scarce demonstrations remains unreliable. While fine-tuned VLA policies often produce semantically plausible trajectories, failures often arise from unresolved geometric ambiguities, where near-miss action candidates lead to divergent execution outcomes under limited supervision. We study few-shot VLA adaptation from a \emph{generation--selection} perspective and propose a novel framework \textbf{VGAS} (\textbf{V}alue-\textbf{G}uided \textbf{A}ction-chunk \textbf{S}election). It performs inference-time best-of-$N$ selection to identify action chunks that are both semantically faithful and geometrically precise. Specifically, \textbf{VGAS} employs a finetuned VLA as a high-recall proposal generator and introduces the \textrm{Q-Chunk-Former}, a geometrically grounded Transformer critic to resolve fine-grained geometric ambiguities. In addition, we propose \textit{Explicit Geometric Regularization} (\texttt{EGR}), which explicitly shapes a discriminative value landscape to preserve action ranking resolution among near-miss candidates while mitigating value instability under scarce supervision. Experiments and theoretical analysis demonstrate that \textbf{VGAS} consistently improves success rates and robustness under limited demonstrations and distribution shifts. Our code is available at https://github.com/Jyugo-15/VGAS.
