利用图像和文本信息提升点云3D目标检测的无监督域适应性能
Multi-Modal Assistance for Unsupervised Domain Adaptation on Point Cloud 3D Object Detection
- 通过图像和文本特征桥接源域与目标域的3D特征
- 结合2D检测器生成更准确的伪标签,提升3D检测效果
- 适合做点云3D检测跨域迁移的研究者参考
基于教师-学生架构和伪标签的激光雷达3D目标检测无监督域适应近年取得显著进展。尽管点云与图像常同时采集,但很少关注图像数据在3D UDA中的作用。本文提出MMAssist方法,通过多模态辅助提升3D UDA性能。将真实或伪标签投影到图像,生成2D边界框,从预训练视觉主干中提取图像特征,利用大视觉语言模型(LVLM)生成框的文本描述,并用预训练文本编码器获取文本特征。在源域训练和目标域学生模型训练中,对齐预测框的3D特征与其对应的图像和文本特征,并通过可学习权重融合,用于最终预测。同时对齐目标域学生分支与教师分支的特征。为增强伪标签,使用现成2D检测器从图像生成2D框,借助点云估计对应3D框,再与教师模型生成的伪标签融合。实验表明,在三个主流3D目标检测数据集上的三个域适应任务中,该方法优于现有最先进方法。
原文摘要 · Abstract (English)
Unsupervised domain adaptation for LiDAR-based 3D object detection (3D UDA) based on the teacher-student architecture with pseudo labels has achieved notable improvements in recent years. Although it is quite popular to collect point clouds and images simultaneously, little attention has been paid to the usefulness of image data in 3D UDA when training the models. In this paper, we propose an approach named MMAssist that improves the performance of 3D UDA with multi-modal assistance. A method is designed to align 3D features between the source domain and the target domain by using image and text features as bridges. More specifically, we project the ground truth labels or pseudo labels to the images to get a set of 2D bounding boxes. For each 2D box, we extract its image feature from a pre-trained vision backbone. A large vision-language model (LVLM) is adopted to extract the box's text description, and a pre-trained text encoder is used to obtain its text feature. During the training of the model in the source domain and the student model in the target domain, we align the 3D features of the predicted boxes with their corresponding image and text features, and the 3D features and the aligned features are fused with learned weights for the final prediction. The features between the student branch and the teacher branch in the target domain are aligned as well. To enhance the pseudo labels, we use an off-the-shelf 2D object detector to generate 2D bounding boxes from images and estimate their corresponding 3D boxes with the aid of point cloud, and these 3D boxes are combined with the pseudo labels generated by the teacher model. Experimental results show that our approach achieves promising performance compared with state-of-the-art methods in three domain adaptation tasks on three popular 3D object detection datasets. The code is available at https://github.com/liangp/MMAssist.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。