arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2603.07464cs.CV

跨模态蒸馏的 selective transfer learning 用于单目3D物体检测

Selective Transfer Learning of Cross-Modality Distillation for Monocular 3D Object Detection

Institute of Artificial Intelligence and Robotics, Xi’an Jiaotong University(人工智能与机器人研究所,西安交通大学)

Rui Ding, Meng Yang, Nanning Zheng

AI总结:

本文提出MonoSTL方法,通过解决跨模态蒸馏中的模态差距问题,提升单目3D物体检测的准确性。

摘要:

单目3D物体检测是自动驾驶领域一个有前景但问题重重的任务,由于缺乏准确的深度信息。跨模态知识蒸馏可以有效将深度信息从激光雷达转移到图像网络。然而,图像与激光雷达之间的模态差距严重限制了其准确性。在本文中,我们首次系统地研究了跨模态蒸馏中由模态差距引起的负转移问题,不仅包括架构不一致的问题,更重要的是特征过拟合的问题。我们提出了一种名为MonoSTL的选学方法,以克服这些问题,鼓励从激光雷达到图像网络的正向转移。一方面,我们利用相似的架构来确保图像网络和激光雷达网络之间特征的空间对齐。另一方面,我们开发了两个新的蒸馏模块,即Depth-Aware Selective Feature Distillation (DASFD) 和 Depth-Aware Selective Relation Distillation (DASRD),通过将深度不确定性整合到特征和关系蒸馏中,分别选择性地学习对象的正向特征和关系。我们的方法可以无缝集成到各种CNN和DETR模型中,其中我们选取了三个最近的KITTI模型和一个最近的NuScenes模型进行验证。广泛的实验表明,我们的方法显著提高了基础模型的准确性,从而在所有最近发布的SOTA模型中实现了最佳准确性。

英文摘要:

Monocular 3D object detection is a promising yet ill-posed task for autonomous vehicles due to the lack of accurate depth information. Cross-modality knowledge distillation could effectively transfer depth information from LiDAR to image-based network. However, modality gap between image and LiDAR seriously limits its accuracy. In this paper, we systematically investigate the negative transfer problem induced by modality gap in cross-modality distillation for the first time, including not only the architecture inconsistency issue but more importantly the feature overfitting issue. We propose a selective learning approach named MonoSTL to overcome these issues, which encourages positive transfer of depth information from LiDAR while alleviates the negative transfer on image-based network. On the one hand, we utilize similar architectures to ensure spatial alignment of features between image-based and LiDAR-based networks. On the other hand, we develop two novel distillation modules, namely Depth-Aware Selective Feature Distillation (DASFD) and Depth-Aware Selective Relation Distillation (DASRD), which selectively learn positive features and relationships of objects by integrating depth uncertainty into feature and relation distillations, respectively. Our approach can be seamlessly integrated into various CNN-based and DETR-based models, where we take three recent models on KITTI and a recent model on NuScenes for validation. Extensive experiments show that our approach considerably improves the accuracy of the base models and thereby achieves the best accuracy compared with all recently released SOTA models.

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

↑