Drive-KD: Multi-Teacher Distillation for VLMs in Autonomous Driving

Weitong Lian, Zecong Tang, Haoran Li, Tianjian Gao, Yifei Wang +10 more
1/29/2026
cs.AIcs.CV
distillation

No confident code implementation found for this paper.

Abstract

Autonomous driving is an important and safety-critical task, and recent advances in LLMs/VLMs have opened new possibilities for reasoning and planning in this domain. However, large models demand substantial GPU memory and exhibit high inference latency, while conventional supervised fine-tuning (SFT) often struggles to bridge the capability gaps of small models. To address these limitations, we propose Drive-KD, a framework that decomposes autonomous driving into a "perception-reasoning-planning" triad and transfers these capabilities via knowledge distillation. We identify layer-specific attention as the distillation signal to construct capability-specific single-teacher models that outperform baselines. Moreover, we unify these single-teacher settings into a multi-teacher distillation framework and introduce asymmetric gradient projection to mitigate cross-capability gradient conflicts. Extensive evaluations validate the generalization of our method across diverse model families and scales. Experiments show that our distilled InternVL3-1B model, with ~42 times less GPU memory and ~11.4 times higher throughput, achieves better overall performance than the pretrained 78B model from the same family on DriveBench, and surpasses GPT-5.1 on the planning dimension, providing insights toward efficient autonomous driving VLMs.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Similar Papers

Cite this paper

@article{lian2026drivekd,
  title  = {Drive-KD: Multi-Teacher Distillation for VLMs in Autonomous Driving},
  author = {Weitong Lian and Zecong Tang and Haoran Li and Tianjian Gao and Yifei Wang and Zixu Wang and Lingyi Meng and Tengju Ru and Zhejun Cui and Yichen Zhu and Hangshuo Cao and Qi Kang and Tianxing Chen and Kaixuan Wang and Yu Zhang},
  year   = {2026},
  eprint = {2601.21288},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.21288v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.21288.svg)](https://codeofpaper.com/paper/2601.21288)

Discussion