冻结视觉编码器提升自动驾驶泛化能力
FROST-Drive: Scalable and Efficient End-to-End Driving with a Frozen Vision Encoder
- 用冻结的视觉语言模型编码器保留通用知识
- 在Waymo数据集上RFS评分显著高于全微调模型
- 适合追求高泛化性的自动驾驶系统研发
端到端自动驾驶模型直接从传感器输入生成控制指令,但其在新场景下的泛化能力仍是关键挑战。传统方法对视觉编码器进行全量微调,可能导致模型过度适应训练数据。本文提出FROST-Drive,通过冻结预训练视觉语言模型的编码器,直接迁移其丰富的通用世界知识。模型结合基于Transformer的适配器实现多模态融合,以及基于GRU的解码器生成平滑路径点。同时设计了针对评分员反馈分数(RFS)的定制损失函数,直接优化轨迹规划鲁棒性。在专为捕捉长尾场景而构建的大规模Waymo Open E2E数据集上,该方法显著优于全微调模型,证明保持强大VLM知识比深度领域适配更有效,为视觉模型应对真实复杂场景提供了新路径。
原文摘要 · Abstract (English)
End-to-end (E2E) models in autonomous driving aim to directly map sensor inputs to control commands, but their ability to generalize to novel and complex scenarios remains a key challenge. The common practice of fully fine-tuning the vision encoder on driving datasets potentially limits its generalization by causing the model to specialize too heavily in the training data. This work challenges the necessity of this training paradigm. We propose FROST-Drive, a novel E2E architecture designed to preserve and leverage the powerful generalization capabilities of a pretrained vision encoder from a Vision-Language Model (VLM). By keeping the encoder's weights frozen, our approach directly transfers the rich, generalized world knowledge from the VLM to the driving task. Our model architecture combines this frozen encoder with a transformer-based adapter for multimodal fusion and a GRU-based decoder for smooth waypoint generation. Furthermore, we introduce a custom loss function designed to directly optimize for Rater Feedback Score (RFS), a metric that prioritizes robust trajectory planning. We conduct extensive experiments on Waymo Open E2E Dataset, a large-scale datasets deliberately curated to capture the long-tail scenarios, demonstrating that our frozen-encoder approach significantly outperforms models that employ full fine-tuning. Our results provide substantial evidence that preserving the broad knowledge of a capable VLM is a more effective strategy for achieving robust, generalizable driving performance than intensive domain-specific adaptation. This offers a new pathway for developing vision-based models that can better handle the complexities of real-world application domains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。