arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.16818cs.CV

SimDistill:用于BEV 3D目标检测的模拟多模态蒸馏

SimDistill: Simulated Multi-modal Distillation for BEV 3D Object Detection

The University of Sydney(悉尼大学);La Trobe University(拉筹伯大学)

Haimei Zhao, Qiming Zhang, Shanshan Zhao, Zhe Chen, Jing Zhang, Dacheng Tao

AI总结:

提出SimDistill多模态蒸馏方法,通过设计相同的师生多模态架构与几何补偿模块,在仅用相机输入下模拟多模态特征,并在BEV空间进行多级蒸馏,显著提升了3D目标检测性能。

摘要:

基于多视角相机的3D目标检测因其低成本而变得流行,但仅从相机数据准确推断3D几何结构仍然具有挑战性,并可能导致性能下降。尽管从LiDAR数据中蒸馏精确的3D几何知识有助于应对这一挑战,但不同传感模态之间显著的模态鸿沟可能会极大地阻碍LiDAR信息的收益。为了解决这个问题,我们提出了一种模拟多模态蒸馏(SimDistill)方法,通过精心设计模型架构和蒸馏策略。具体而言,我们为教师和学生模型设计了多模态架构,包括基于LiDAR-相机融合的教师模型和基于模拟融合的学生模型。得益于这种“相同”的架构设计,学生模型可以模仿教师模型,在仅输入多视角图像的情况下生成多模态特征,其中引入了几何补偿模块来弥合模态鸿沟。此外,我们提出了一种全面的多模态蒸馏方案,在鸟瞰图(Bird's-eye-view)空间中同时支持模态内、跨模态和多模态融合蒸馏。将它们结合在一起,我们的SimDistill能够为3D目标检测学习更好的特征表示,同时保持仅使用相机的低成本部署。大量实验验证了SimDistill相对于最先进方法的有效性和优越性,在基线检测器上实现了4.8% mAP和4.1% NDS的提升。源代码将发布在https://github.com/ViTAE-Transformer/SimDistill。

英文摘要:

Multi-view camera-based 3D object detection has become popular due to its low cost, but accurately inferring 3D geometry solely from camera data remains challenging and may lead to inferior performance. Although distilling precise 3D geometry knowledge from LiDAR data could help tackle this challenge, the benefits of LiDAR information could be greatly hindered by the significant modality gap between different sensory modalities. To address this issue, we propose a Simulated multi-modal Distillation (SimDistill) method by carefully crafting the model architecture and distillation strategy. Specifically, we devise multi-modal architectures for both teacher and student models, including a LiDAR-camera fusion-based teacher and a simulated fusion-based student. Owing to the ``identical'' architecture design, the student can mimic the teacher to generate multi-modal features with merely multi-view images as input, where a geometry compensation module is introduced to bridge the modality gap. Furthermore, we propose a comprehensive multi-modal distillation scheme that supports intra-modal, cross-modal, and multi-modal fusion distillation simultaneously in the Bird's-eye-view space. Incorporating them together, our SimDistill can learn better feature representations for 3D object detection while maintaining a cost-effective camera-only deployment. Extensive experiments validate the effectiveness and superiority of SimDistill over state-of-the-art methods, achieving an improvement of 4.8\% mAP and 4.1\% NDS over the baseline detector. The source code will be released at https://github.com/ViTAE-Transformer/SimDistill.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑