Teaching Vision-Language-Action Models What to See and Where to Look

· Source: Computer Vision and Pattern Recognition · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Robotics & Autonomous Systems · Depth: Expert, quick

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

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

Topics

Code references

Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Computer Vision and Pattern Recognition.