用可控文生图生成新驾驶场景,提升3D检测模型鲁棒性
DriveGEN: Generalized and Robust 3D Detection in Driving via Controllable Text-to-Image Diffusion Generation
- 基于自注意力特征提取物体几何原型,实现精准3D对象保留
- 无需训练新模型,生成多样化分布外场景下检测性能提升12.7%
- 适合做自动驾驶感知系统鲁棒性增强的研究者和工程师
在自动驾驶中,视觉主导的3D检测旨在从图像中识别3D物体。然而,高昂的数据采集成本和多变的真实场景限制了训练数据规模。当训练与测试数据分布发生偏移时,现有方法常出现性能下降,即分布外(OOD)问题。为此,可控文生图(T2I)扩散模型可作为数据增强手段,生成具有精确3D几何结构的多样化OOD场景。但现有方法受限于训练数据规模或难以保持所有标注3D对象。本文提出DriveGEN,一种无需训练的可控文生图生成方法,用于提升驾驶场景中3D检测器的泛化能力。其包含两个阶段:1)自原型提取:发现自注意力特征具语义感知性,通过布局精确定位3D物体,提取称为自原型的精确物体特征;2)原型引导扩散:在去噪过程中进行语义感知特征对齐与浅层特征对齐,以保持不同OOD场景下的物体完整性。大量实验表明,DriveGEN能有效提升3D检测性能,在多个OOD设置下平均提升12.7%。代码已开源。
原文摘要 · Abstract (English)
In autonomous driving, vision-centric 3D detection aims to identify 3D objects from images. However, high data collection costs and diverse real-world scenarios limit the scale of training data. Once distribution shifts occur between training and test data, existing methods often suffer from performance degradation, known as Out-of-Distribution (OOD) problems. To address this, controllable Text-to-Image (T2I) diffusion offers a potential solution for training data enhancement, which is required to generate diverse OOD scenarios with precise 3D object geometry. Nevertheless, existing controllable T2I approaches are restricted by the limited scale of training data or struggle to preserve all annotated 3D objects. In this paper, we present DriveGEN, a method designed to improve the robustness of 3D detectors in Driving via Training-Free Controllable Text-to-Image Diffusion Generation. Without extra diffusion model training, DriveGEN consistently preserves objects with precise 3D geometry across diverse OOD generations, consisting of 2 stages: 1) Self-Prototype Extraction: We empirically find that self-attention features are semantic-aware but require accurate region selection for 3D objects. Thus, we extract precise object features via layouts to capture 3D object geometry, termed self-prototypes. 2) Prototype-Guided Diffusion: To preserve objects across various OOD scenarios, we perform semantic-aware feature alignment and shallow feature alignment during denoising. Extensive experiments demonstrate the effectiveness of DriveGEN in improving 3D detection. The code is available at https://github.com/Hongbin98/DriveGEN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。