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

FSD-BEV: 面向多视角3D目标检测的前景自蒸馏方法

FSD-BEV: Foreground Self-Distillation for Multi-view 3D Object Detection

Hangzhou Innovation Institute, Beihang University(北京航空航天大学杭州创新研究院);School of Software, Beihang University(北京航空航天大学软件学院);State Key Laboratory of Virtual Reality Technology and Systems, Beihang University(北京航空航天大学虚拟现实技术与系统国家重点实验室);Shanghai ZEEKR Blue New Energy Technology Co., Ltd.(上海极氪蓝新能源科技有限公司);Zhongguancun Laboratory, Beijing, China(中关村实验室)

Zheng Jiang, Jinqing Zhang, Yanan Zhang, Qingjie Liu, Zhenghui Hu, Baohui Wang, Yunhong Wang

AI总结:

针对多视角BEV 3D目标检测性能不及激光雷达方法、跨模态蒸馏存在分布差异的问题,提出前景自蒸馏方案、点云增强策略及多尺度前景增强模块,构建FSD-BEV框架,在nuScenes数据集上达到SOTA性能。

摘要:

尽管基于鸟瞰图(Bird's-Eye-View, BEV)范式的多视角3D目标检测作为一种经济且易于部署的自动驾驶感知解决方案已获得广泛关注,但与基于激光雷达的方法相比仍存在性能差距。近年来,研究者提出了多种跨模态蒸馏方法,旨在将教师模型中的有益信息迁移到学生模型中以提升性能。然而,由于不同数据模态和网络结构导致的特征分布差异,这些方法面临挑战,使得知识迁移极具难度。本文提出了一种前景自蒸馏(Foreground Self-Distillation, FSD)方案,该方案有效避免了分布差异问题,无需预训练教师模型或复杂的蒸馏策略即可保持显著的蒸馏效果。此外,我们设计了两种点云增强(Point Cloud Intensification, PCI)策略,通过帧组合和伪点赋值来弥补点云的稀疏性。最后,我们开发了多尺度前景增强(Multi-Scale Foreground Enhancement, MSFE)模块,通过预测的椭圆高斯热图提取并融合多尺度前景特征,进一步提升模型性能。我们将上述所有创新整合到一个名为FSD-BEV的统一框架中。在nuScenes数据集上的大量实验表明,FSD-BEV取得了当前最优性能,证明了其有效性。代码和模型可在以下链接获取:https://github.com/CocoBoom/fsd-bev。

英文摘要:

Although multi-view 3D object detection based on the Bird's-Eye-View (BEV) paradigm has garnered widespread attention as an economical and deployment-friendly perception solution for autonomous driving, there is still a performance gap compared to LiDAR-based methods. In recent years, several cross-modal distillation methods have been proposed to transfer beneficial information from teacher models to student models, with the aim of enhancing performance. However, these methods face challenges due to discrepancies in feature distribution originating from different data modalities and network structures, making knowledge transfer exceptionally challenging. In this paper, we propose a Foreground Self-Distillation (FSD) scheme that effectively avoids the issue of distribution discrepancies, maintaining remarkable distillation effects without the need for pre-trained teacher models or cumbersome distillation strategies. Additionally, we design two Point Cloud Intensification (PCI) strategies to compensate for the sparsity of point clouds by frame combination and pseudo point assignment. Finally, we develop a Multi-Scale Foreground Enhancement (MSFE) module to extract and fuse multi-scale foreground features by predicted elliptical Gaussian heatmap, further improving the model's performance. We integrate all the above innovations into a unified framework named FSD-BEV. Extensive experiments on the nuScenes dataset exhibit that FSD-BEV achieves state-of-the-art performance, highlighting its effectiveness. The code and models are available at: https://github.com/CocoBoom/fsd-bev.

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

↑