用扩散模型修复恶劣天气下的多模态3D检测数据
Diffusion-Based Restoration for Multi-Modal 3D Object Detection in Adverse Weather
- 用扩散模型恢复被天气损坏的图像和点云数据
- 在三个公开数据集上实现当前最优的恶劣天气鲁棒性
- 适合自动驾驶与机器人感知领域研究者参考
多模态3D目标检测对机器人和自动驾驶的可靠感知至关重要,但在恶劣天气下因天气引起的畸变及不同模态间配准错位,性能受限。本文提出DiffFusion框架,通过基于扩散模型的修复与自适应跨模态融合提升挑战性天气下的鲁棒性。核心思路是利用扩散模型强大的去噪与生成能力适应多种天气条件。DiffFusion引入图像修复(Diffusion-IR)恢复受天气影响的图像,以及点云修复(PCR)利用图像目标线索补偿受损的LiDAR数据。为解决模态间错位问题,设计双向自适应融合与对齐模块(BAFAM),实现动态多模态融合与双向鸟瞰图(BEV)对齐,保持空间一致性。在三个公开数据集上的大量实验表明,DiffFusion在恶劣天气下达到当前最优鲁棒性,同时保持良好的清洁数据表现。零样本测试在真实世界DENSE数据集上进一步验证了其泛化能力。代码将开源。
原文摘要 · Abstract (English)
Multi-modal 3D object detection is important for reliable perception in robotics and autonomous driving. However, its effectiveness remains limited under adverse weather conditions due to weather-induced distortions and misalignment between different data modalities. In this work, we propose DiffFusion, a novel framework designed to enhance robustness in challenging weather through diffusion-based restoration and adaptive cross-modal fusion. Our key insight is that diffusion models possess strong capabilities for denoising and generating data that can adapt to various weather conditions. Building on this, DiffFusion introduces Diffusion-IR restoring images degraded by weather effects and Point Cloud Restoration (PCR) compensating for corrupted LiDAR data using image object cues. To tackle misalignments between two modalities, we develop Bidirectional Adaptive Fusion and Alignment Module (BAFAM). It enables dynamic multi-modal fusion and bidirectional bird's-eye view (BEV) alignment to maintain consistent spatial correspondence. Extensive experiments on three public datasets show that DiffFusion achieves state-of-the-art robustness under adverse weather while preserving strong clean-data performance. Zero-shot results on the real-world DENSE dataset further validate its generalization. The implementation of our DiffFusion will be released as open-source.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。