GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization

GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization

Xiaosong Jia, Bowen Yang, Zuhao Ge, Xian Nie, Yuchen Zhou, Cunxin Fan, Yufeng Li, Yilin Chai, Chao Jing, Zijian Liang, Qingwen Bu, Haidong Cao, Chao Wu, Qifeng Li, Zhenjie Yang, Chenhe Zhang, Hongyang Li, Zuxuan Wu, Junchi Yan, Yu-Gang Jiang · · 2026

Framework

N/A

License

N/A

Stars

0

Summary

Vision-Language-Action (VLA) models aim for general robot learning by aligning action as a modality within powerful Vision-Language Models (VLMs). Existing VLAs rely on end-to-end supervision to implicitly enable the action decoding process to learn task-relevant features.

Abstract Summary

Vision-Language-Action (VLA) models aim for general robot learning by aligning action as a modality within powerful Vision-Language Models (VLMs). Existing VLAs rely on end-to-end supervision to implicitly enable the action decoding process to learn task-relevant features. However, without explicit guidance, these models often overfit to spurious correlations, such as visual shortcuts or environmental noise, limiting their generalization. In this paper, we introduce GuidedVLA, a framework designed to manually guide the action generation to focus on task-relevant factors. Our core insight is to treat the action decoder not as a monolithic learner, but as an assembly of functional components. Individual attention heads are supervised by manually defined auxiliary signals to capture distinct factors. As an initial study, we instantiate this paradigm with three specialized heads: object grounding, spatial geometry, and temporal skill logic. Across simulation and real-robot experiments, GuidedVLA improves success rates in both in-domain and out-of-domain settings compared to strong VLA baselines. Finally, we show that the quality of these specialized factors correlates positively with task performance and that our mechanism yields decoupled, high-quality features. Our results suggest that explicitly guiding action-decoder learning is a promising direction for building more robust and general VLA models.

Key Points

  • Introduces GuidedVLA, advancing vla research.
  • Detailed evaluation demonstrates effectiveness on benchmarks and real-world scenarios.
  • Provides insights into methodology and architectural choices.

Method

Recent representative Vision-Language-Action (VLA) models [ 8 , 45 , 104 ] extend Vision-Language Models (VLMs) by introducing action tokens $\bm{a}$ alongside vision tokens $\bm{v}$ and language tokens $\bm{l}$. The action generation process is trained to regress robot trajectories by denoising $\bm{a}$ conditioned on $(\bm{v},\bm{l})$.

Although VLMs provide rich semantic features in $(\bm{v},\bm{l})$, the action tokens $\bm{a}$ do not inherently learn to extract task-critical information.

Method — GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization
Fig. 3 : Automatic factor annotation pipeline. Object masks are initialized by Qwen3-VL point prompts and propagated by SAM2, skill labels are generated by Qwen3-VL from stage descriptions and a predefined skill list, and depth guidance uses frozen depth features without requiring depth labels. The pipeline substantially reduces human annotation time while preserving a human verification step for supervision quality.

Experiments

LIBERO-Plus [ 28 ] is a robustness-oriented benchmark built upon LIBERO [ 59 ] . It is designed to evaluate generalist manipulation policies under distribution shifts .

It introduces perturbations along seven dimensions: camera viewpoint, robot initial state, language variation, lighting condition, background texture, sensor noise, and object layout to expose failure modes under generalization scenario beyond in-domain evaluation. We compare with state-of-the-art baselines in Table I .

Experiments — GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization
Fig. 5 : Real-world Robot Platforms and Evaluation Tasks. (a) ALOHA AgileX dual-arm mobile manipulator with left/right wrist Orbbec Dabai cameras and a third-person Orbbec Dabai camera; we evaluate three household tasks: pick up fruits and vegetables, stack the bowls, clean the tabletop. (b) PSI-Bot equipped with RealMan RM63 arm(s) and DexHand2 Pro hands, with head/chest RealSense D435 cameras; we evaluate three lab tasks: pick up beaker, stack beakers, and heat beaker.

Analysis

Do VLAs under-utilize vision-language representations in action decoding process, and can explicit factor guidance close this gap? (Section V-B ) Does our proposed GuidedVLA improve baseline performance under both in-distribution and out-of-distribution evaluations? (Section V-A ) Which factors (object, skill, geometry) matter most for which task types?

(Section V-A ) Does our attention head specialization indeed lead to learning decoupled features? (Section V-C ) How different architectural choices for guidance influence performance? (Section...

Analysis — GuidedVLA: Specifying Task-Relevant Factors via Plug-and-Play Action Attention Specialization
Fig. 6 : Higher Factor Quality Leads to Better Task Performance. Top: Quantitative analysis on the LIBERO-Plus layout perturbation track shows that improving the quality of each specialized head consistently boosts success rates. (a) Object Head: as the proportion of attention focused on task-relevant object regions increases, success rises from 61.3% to 77.4%, highlighting the importance of precise object-centric attention. (b) Skill Head: higher skill-recognition accuracy, measured by a linear probe, correlates with improved performance (66.2% to 77.7%), indicating that better temporal understanding enhances control. (c) Depth Head: increasing the ratio of true depth features (versus noise) dramatically improves both qualitative depth estimation and quantitative success (15.0% to 74.2%), confirming that explicit 3D cues are critical for robust manipulation. Bottom: Qualitative visualizations show how changes along the x-axis metrics are reflected in the corresponding feature representations.

Sources and Demonstrations

Figures are reproduced from Jia et al. See the full paper for experimental details and the project page, when the paper links one, for demonstrations and videos.

Share

Related Papers