用知识图谱增强视觉模型,提升自动驾驶安全判断能力
SafeDriveRAG: Towards Safe Autonomous Driving with Knowledge Graph-based Retrieval-Augmented Generation
- 基于知识图谱的检索增强生成,提升视觉模型对交通场景的理解
- 在22.8万个数据上验证,安全类任务准确率最高提升14.57%
- 适合研究自动驾驶安全、多模态推理与知识融合的学者
本文研究如何利用视觉语言模型(VLM)提升自动驾驶系统的安全性,涵盖感知、情境理解与路径规划。现有研究普遍忽视了在交通安全隐患场景下的评估。为此,我们构建了首个大规模多模态问答基准SafeDrive228K,包含228,000个样本和18个子任务,覆盖交通事故、边缘案例及通用安全常识等多样查询,可全面评估模型的理解与推理能力。同时提出SafeDriveRAG,一种基于多尺度子图检索的插件式知识图谱增强生成方法,通过整合互联网收集的交通规则知识,显著提升模型应对安全关键场景的能力。在五种主流VLM上进行评估显示,引入RAG后,在交通事故任务中提升4.73%,边缘案例任务提升8.79%,交通安全常识任务提升14.57%,证明该基准与方法对推动交通安全研究具有重要价值。代码与数据已开源。
原文摘要 · Abstract (English)
In this work, we study how vision-language models (VLMs) can be utilized to enhance the safety for the autonomous driving system, including perception, situational understanding, and path planning. However, existing research has largely overlooked the evaluation of these models in traffic safety-critical driving scenarios. To bridge this gap, we create the benchmark (SafeDrive228K) and propose a new baseline based on VLM with knowledge graph-based retrieval-augmented generation (SafeDriveRAG) for visual question answering (VQA). Specifically, we introduce SafeDrive228K, the first large-scale multimodal question-answering benchmark comprising 228K examples across 18 sub-tasks. This benchmark encompasses a diverse range of traffic safety queries, from traffic accidents and corner cases to common safety knowledge, enabling a thorough assessment of the comprehension and reasoning abilities of the models. Furthermore, we propose a plug-and-play multimodal knowledge graph-based retrieval-augmented generation approach that employs a novel multi-scale subgraph retrieval algorithm for efficient information retrieval. By incorporating traffic safety guidelines collected from the Internet, this framework further enhances the model's capacity to handle safety-critical situations. Finally, we conduct comprehensive evaluations on five mainstream VLMs to assess their reliability in safety-sensitive driving tasks. Experimental results demonstrate that integrating RAG significantly improves performance, achieving a +4.73% gain in Traffic Accidents tasks, +8.79% in Corner Cases tasks and +14.57% in Traffic Safety Commonsense across five mainstream VLMs, underscoring the potential of our proposed benchmark and methodology for advancing research in traffic safety. Our source code and data are available at https://github.com/Lumos0507/SafeDriveRAG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。