arXiv:2509.23880cs.CV2025-09中稿 · ICRA

自动挑选高质量伪标签,提升3D目标检测的半监督学习效果

Learning Adaptive Pseudo-Label Selection for Semi-Supervised 3D Object Detection

  • 用可学习模块融合教师输出,自适应确定伪标签阈值
  • 在KITTI和Waymo上实现更高精度与召回率,伪标签更可靠
  • 适合追求高效标注的3D检测研究者和工业应用

半监督3D目标检测(SS3DOD)旨在减少昂贵的3D标注,利用未标注数据。现有方法采用基于伪标签的师生框架,但其主要挑战在于如何从教师预测中选择高质量伪标签。以往方法多通过手动设定置信度阈值进行筛选,最新工作虽引入动态阈值或伪标签质量优化,但仍忽略物体距离、类别、学习状态等上下文信息,且仅依赖网络部分输出评估伪标签质量。本文提出一种新型SS3DOD框架,包含可学习的伪标签选择模块,能自动适应性地挑选高质伪标签。该方法在教师输出层引入两个网络,通过分数融合评估伪标签质量,并由伪标签与真实框对齐情况监督,生成上下文自适应阈值。此外,引入软监督策略,使学生网络在伪标签噪声下仍能稳健学习,优先关注清晰标签。在KITTI和Waymo数据集上的大量实验表明,所提方法在保持更广上下文覆盖的同时,显著提升伪标签精度与召回率,有效改进现有SS3DOD方法。

原文摘要 · Abstract (English)

Semi-supervised 3D object detection (SS3DOD) aims to reduce costly 3D annotations utilizing unlabeled data. Recent studies adopt pseudo-label-based teacher-student frameworks and demonstrate impressive performance. The main challenge of these frameworks is in selecting high-quality pseudo-labels from the teacher's predictions. Most previous methods, however, select pseudo-labels by comparing confidence scores over thresholds manually set. The latest works tackle the challenge either by dynamic thresholding or refining the quality of pseudo-labels. Such methods still overlook contextual information e.g. object distances, classes, and learning states, and inadequately assess the pseudo-label quality using partial information available from the networks. In this work, we propose a novel SS3DOD framework featuring a learnable pseudo-labeling module designed to automatically and adaptively select high-quality pseudo-labels. Our approach introduces two networks at the teacher output level. These networks reliably assess the quality of pseudo-labels by the score fusion and determine context-adaptive thresholds, which are supervised by the alignment of pseudo-labels over GT bounding boxes. Additionally, we introduce a soft supervision strategy that can learn robustly under pseudo-label noises. This helps the student network prioritize cleaner labels over noisy ones in semi-supervised learning. Extensive experiments on the KITTI and Waymo datasets demonstrate the effectiveness of our method. The proposed method selects high-precision pseudo-labels while maintaining a wider coverage of contexts and a higher recall rate, significantly improving relevant SS3DOD methods.

3D目标检测半监督学习伪标签自适应选择

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。