用多教师蒸馏让小模型高效实现自动驾驶视觉语言能力。
Drive-KD: Multi-Teacher Distillation for VLMs in Autonomous Driving
- 将自动驾驶拆解为感知-推理-规划三阶段,分阶段蒸馏知识。
- 1B小模型仅需42倍少显存,速度提升11.4倍,性能超78B大模型。
- 适合追求高效率、低资源的自动驾驶视觉语言系统研发者。
自动驾驶是重要且关乎安全的任务,近年来大语言模型/视觉语言模型(LLMs/VLMs)的发展为该领域带来了新的推理与规划可能性。然而,大型模型需要大量GPU内存且推理延迟高,而传统监督微调(SFT)难以弥补小型模型的能力差距。为此,我们提出Drive-KD框架,将自动驾驶分解为‘感知-推理-规划’三元结构,并通过知识蒸馏转移这些能力。我们识别出层级特定注意力作为蒸馏信号,构建出表现优于基线的单教师能力专用模型。进一步地,我们将这些单教师设置统一为多教师蒸馏框架,并引入非对称梯度投影以缓解跨能力梯度冲突。大量实验验证了方法在多种模型家族和规模下的泛化能力。结果显示,我们的蒸馏后InternVL3-1B模型,在约42倍更少的GPU内存和约11.4倍更高的吞吐量下,于DriveBench上整体性能优于同家族预训练的78B模型,并在规划维度超越GPT-5.1,为高效自动驾驶VLM提供了新思路。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。