Teaching Vision-Language-Action Models What to See and Where to Look
Summary
DriveTeach-VLA is a novel framework designed to enhance Vision-Language-Action (VLA) models for end-to-end autonomous driving by explicitly teaching them visual focus and spatial awareness. Existing VLA models often rely on text-centric visual question answering and chain-of-thought reasoning, leading to representations that capture semantic knowledge but lack crucial spatial dependencies for reliable trajectory prediction. DriveTeach-VLA addresses this through two main components: Driving-aware Vision Distillation (DVD), which injects driving-specific perceptual priors into the vision encoder, and 2D Trajectory-Guided Prompts (2D-TGP), which provide spatial conditioning aligned with feasible driving trajectories. This vision-guided learning pipeline, encompassing DVD pretraining, TGP-guided SFT, and TGP-guided GRPO, achieved top performance on both NAVSIM and nuScenes benchmarks.
Key takeaway
For Machine Learning Engineers developing autonomous driving systems, this research highlights the need to move beyond purely text-centric VLA training. You should consider integrating explicit spatial conditioning and driving-specific perceptual priors, as demonstrated by DriveTeach-VLA's DVD and 2D-TGP components. This approach can significantly improve trajectory prediction reliability and overall VLA model performance on benchmarks like NAVSIM and nuScenes.
Key insights
DriveTeach-VLA improves VLA models for autonomous driving by integrating driving-specific visual priors and trajectory-guided spatial conditioning.
Principles
- VLA models require explicit spatial dependency training.
- Text-centric VQA data alone is insufficient for action-grounded planning.
- Integrating perceptual priors enhances vision encoders.
Method
DriveTeach-VLA employs Driving-aware Vision Distillation (DVD) for perceptual priors, followed by 2D Trajectory-Guided Prompts (2D-TGP) for spatial conditioning. This pipeline includes DVD pretraining, TGP-guided SFT, and TGP-guided GRPO.
In practice
- Apply DVD for driving-specific vision pretraining.
- Use 2D-TGP for spatial conditioning in VLA models.
- Evaluate VLA performance on NAVSIM and nuScenes.
Topics
- Vision-Language-Action Models
- Autonomous Driving
- Trajectory Prediction
- Vision Distillation
- Spatial Conditioning
- nuScenes Benchmark
Code references
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.