用视觉语言模型引导扩散模型,实现端到端自动驾驶路径生成
VDT-Auto: End-to-end Autonomous Driving with VLM-Guided Diffusion Transformers
- 结合视觉语言模型与扩散Transformer,从环境语义和几何特征生成驾驶路径
- 在nuScenes数据集上平均路径误差0.52米,碰撞率降低至21%
- 适用于复杂场景下的自动驾驶系统,尤其擅长处理罕见路况
在自动驾驶中,动态环境与极端情况对车辆决策的鲁棒性构成重大挑战。针对端到端自动驾驶中的状态-动作映射表示问题,我们提出VDT-Auto新框架。该框架利用视觉语言模型(VLM)对环境的理解能力,结合基于扩散变压器的动作生成机制,从几何与语义双重角度为扩散过程提供条件。几何方面,采用鸟瞰图(BEV)编码器从周围图像提取特征网格;语义方面,对微调后VLM的结构化输出进行文本嵌入与噪声路径处理。扩散过程中,前向过程的噪声源自微调VLM输出的噪声路径,而反向过程则由提取的BEV特征与文本嵌入共同引导。在nuScenes开环规划评估中,平均L2误差达0.52米,平均碰撞率21%。真实世界演示验证了其优异泛化能力。代码与数据集将在论文接收后公开。
原文摘要 · Abstract (English)
In autonomous driving, dynamic environment and corner cases pose significant challenges to the robustness of ego vehicle's decision-making. To address these challenges, commencing with the representation of state-action mapping in the end-to-end autonomous driving paradigm, we introduce a novel pipeline, VDT-Auto. Leveraging the advancement of the state understanding of Visual Language Model (VLM), incorporating with diffusion Transformer-based action generation, our VDT-Auto parses the environment geometrically and contextually for the conditioning of the diffusion process. Geometrically, we use a bird's-eye view (BEV) encoder to extract feature grids from the surrounding images. Contextually, the structured output of our fine-tuned VLM is processed into textual embeddings and noisy paths. During our diffusion process, the added noise for the forward process is sampled from the noisy path output of the fine-tuned VLM, while the extracted BEV feature grids and embedded texts condition the reverse process of our diffusion Transformers. Our VDT-Auto achieved 0.52m on average L2 errors and 21% on average collision rate in the nuScenes open-loop planning evaluation. Moreover, the real-world demonstration exhibited prominent generalizability of our VDT-Auto. The code and dataset will be released after acceptance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。