GitHub

Overview

This paper studies VLA initialization as a controlled representation-design problem along three axes: capability-level embodied VQA supervision, parameter-update strategy, and robot-data pretraining. [arXiv]

Key findings:

  • The original pretrained VLM representation is a major source of action performance (>20% drop when training from scratch).
  • Embodied VQA adaptation is conditionally useful: its benefit depends on the downstream bottleneck.
  • LoRA provides a more reliable initialization than Full Finetune — overly reshaping the pretrained representation weakens VLA initialization.
  • The best initialization follows a staged route: adapt with {Grounding + Egocentric Understanding}, then continue with LoRA-based robot-data pretraining.

Main Results

Table 1: Single-Domain VQA Adaptation

Effect of adapting the VLM with a single embodied VQA domain before VLA training.

Domain MLP Head (OFT) Diffusion Expert (PI)
Libero-10 SimplerBridge RoboCasa Libero-10 SimplerBridge RoboCasa
Train from scratch66.619.420.368.628.930.1
Baseline (pretrained VLM)92.445.849.591.850.551.7
Spatial93.041.449.292.449.950.0
Grounding95.644.850.494.250.852.7
Plan & Reasoning95.237.847.592.848.151.1
Camera Prediction93.243.047.992.646.650.7
Ego Understanding95.343.249.993.049.852.0
Temporal96.438.247.993.149.150.7
Action-NTP95.444.050.093.449.652.3

Table 2: Domain Composition

Combining multiple VQA domains.

Configuration MLP Head (OFT) Diffusion Expert (PI)
Libero-10 RoboCasa Libero-10 RoboCasa
Grounding95.650.494.252.7
Ego Understanding95.349.993.052.0
Action-NTP95.450.093.452.3
Grounding + Ego95.751.595.853.5
Grounding + Action-NTP95.250.694.551.9
Ego + Action-NTP95.050.294.851.7
Grounding + Ego + Action-NTP95.049.594.151.2
Grounding + Ego + Spatial94.649.793.650.7
Grounding + Ego + Action-NTP + Spatial94.548.493.050.4
Uniform 7-domain94.249.193.950.4

Table 3: Robot-Data Pretraining

Init. VLM Pretrain Data Update Strategy RoboCasa GR1 SR (%)
Base49.5
G+E adapted51.5
BaseAgiBotFull FT52.0
BaseAgiBot + VQAFull FT53.2
BaseAgiBotLoRA r6454.0
BaseAgiBot + VQALoRA r6452.4
BaseAgiBot + VQALoRA r1651.5
G+E adaptedAgiBotLoRA r6455.2
BaseAgiBot + G+E VQALoRA r6452.6

Quick Start

The experiments follow a two-stage pipeline. Stage 1 adapts the VLM with embodied VQA supervision. Stage 2 trains the VLA policy from the adapted checkpoint.

Stage 1: VLM Adaptation with Qwen3-VL

We use Qwen3-VL with LoRA to adapt the base VLM on embodied VQA data. The high-level workflow:

  1. Clone the Qwen3-VL repo and install its environment following the qwen-vl-finetune instructions.
  2. Prepare your embodied VQA data in the Qwen-VL conversation format (JSON/JSONL with image + conversations fields), and register it in data/__init__.py within the finetune module. See Data Sources below for what datasets to download per domain.
  3. Launch LoRA training via torchrun with --lora_enable True. Our paper uses rank 16, alpha 32, lr 5e-5, 1 epoch, 800K samples per domain, and unfreezes the last 25% of vision encoder layers.

Stage 2: VLA Training with StarVLA

We use StarVLA to train the VLA policy from the Stage-1 adapted checkpoint.

  1. Clone the StarVLA repo (use the stable starVLA branch) and install its environment following the Quick Start Guide.
  2. Point framework.qwenvl.base_vlm in the training config to your Stage-1 LoRA-merged checkpoint directory (HuggingFace format).
  3. Launch VLA training with either the QwenOFT (MLP action head) or QwenPI (flow-matching diffusion action expert, pi0-style) framework.

For training commands, dataset preparation (LeRobot format), evaluation (client-server architecture), and full configuration details, refer to the documentation.


Embodied VQA Data Sources

We organize Stage-1 data into seven capability-oriented domains.

Click to expand full data source table
Domain Dataset Source
Spatial SpaceLLaVA HuggingFace
SpaceThinker HuggingFace
OpenSpaces HuggingFace
STVQA-7K HugginFace
SpatialRGPT GitHub
SpatialQA HugginFace
VST-500K GitHub
VSI-590K HugginFace
Spatial-SSRL HugginFace
ScanQA HugginFace
SpaceR HugginFace
RoboSpatial GitHub
Grounding PixMo Points HuggingFace
RoboPoint GitHub
RoboRefer GitHub
RoboAfford HuggingFace
EO-Data1.5M HuggingFace
RoboRefIt GitHub
ShareRobot-affordance HuggingFace
Plan & Reasoning RoboRefer GitHub
VLM-3R GitHub
Camera Prediction Puffin-4M GitHub
VSI-590K HuggingFace
Egocentric Understanding Robo2VLM GitHub
EgoThinker GitHub
EgoTaskQA GitHub
EO-Data1.5M HuggingFace
ShareRobot HuggingFace
Temporal Understanding VSI-590K video HuggingFace
VICA-332K HuggingFace
VLM-3R-video GitHub
SpaceR HuggingFace
Action-NTP OpenX-Embodiment GitHub
AgiBot-World-Beta GitHub

Citation

If you find this work useful, please cite:

@article{lin2026rethinking,
  title={Rethinking VLM Representation for VLA Initialization},
  author={Lin, Weifeng and Huang, Siyuan and Li, Hao and Chen, Tingwei and An, Ruichuan and Wei, Xinyu and Liu, Jianbo and Li, Hongsheng},
  journal={arXiv preprint arXiv:2605.25802},
  year={2026}
}

Read the original on github.com ↗