用双机制让大模型更懂自动驾驶决策与场景预测
ReasonPlan: Unified Scene Prediction and Decision Reasoning for Closed-loop Autonomous Driving
- 通过场景预测与决策链协同训练,提升视觉与驾驶意图对齐
- 在Bench2Drive上比主流方法高19% L2误差和16.1分驾驶得分
- 零样本泛化能力强,适合处理未见过的复杂交通场景
由于多模态大语言模型(MLLM)具备强大的视觉-语言推理与泛化能力,其在端到端(E2E)自动驾驶领域受到广泛关注。然而,其在闭环系统中的应用仍不充分,现有基于MLLM的方法尚未显著优于主流的E2E模仿学习方法。本文提出ReasonPlan,一种面向闭环驾驶的新型MLLM微调框架,通过自监督的下一场景预测任务与有监督的决策思维链过程,实现整体性推理。该双重机制促使模型将视觉表征与可操作的驾驶上下文对齐,同时推动可解释且因果合理的决策。我们构建了以规划为导向的决策推理数据集PDR,包含21万条多样且高质量样本。在Bench2Drive基准上,本方法相较主流E2E模仿学习方法在L2误差上领先19%,驾驶得分提升16.1。此外,ReasonPlan在未见的DOS基准上展现出强零样本泛化能力,表明其在处理零样本边缘情况时具有优异适应性。代码与数据集详见https://github.com/Liuxueyi/ReasonPlan。
原文摘要 · Abstract (English)
Due to the powerful vision-language reasoning and generalization abilities, multimodal large language models (MLLMs) have garnered significant attention in the field of end-to-end (E2E) autonomous driving. However, their application to closed-loop systems remains underexplored, and current MLLM-based methods have not shown clear superiority to mainstream E2E imitation learning approaches. In this work, we propose ReasonPlan, a novel MLLM fine-tuning framework designed for closed-loop driving through holistic reasoning with a self-supervised Next Scene Prediction task and supervised Decision Chain-of-Thought process. This dual mechanism encourages the model to align visual representations with actionable driving context, while promoting interpretable and causally grounded decision making. We curate a planning-oriented decision reasoning dataset, namely PDR, comprising 210k diverse and high-quality samples. Our method outperforms the mainstream E2E imitation learning method by a large margin of 19% L2 and 16.1 driving score on Bench2Drive benchmark. Furthermore, ReasonPlan demonstrates strong zero-shot generalization on unseen DOS benchmark, highlighting its adaptability in handling zero-shot corner cases. Code and dataset will be found in https://github.com/Liuxueyi/ReasonPlan.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。