3 个月前

一致教师:面向减少半监督目标检测中不一致伪目标的研究

一致教师:面向减少半监督目标检测中不一致伪目标的研究

摘要

在本研究中,我们深入探讨了半监督目标检测(SSOD)中伪目标(pseudo targets)不一致性的问题。我们的核心观察是:震荡不定的伪目标会严重损害检测器的训练效果,向学生网络的训练过程引入噪声,导致严重的过拟合问题。为此,我们提出了一种系统性的解决方案,命名为 ConsistentTeacher,旨在有效降低伪目标的不一致性。首先,我们引入自适应锚框分配机制(Adaptive Anchor Assignment, ASA),替代传统的基于IoU的静态分配策略,使学生网络具备对噪声伪边界框的鲁棒性。其次,我们设计了一个三维特征对齐模块(3D Feature Alignment Module, FAM-3D),用于校准各子任务的预测结果。该模块使每个分类特征能够自适应地在任意尺度和位置上查询最优的特征向量,以支持回归任务,从而提升特征表达的一致性。最后,我们采用高斯混合模型(Gaussian Mixture Model, GMM)动态调整伪边界框的置信度阈值,有效稳定训练初期的真实目标数量,并缓解训练过程中不可靠的监督信号问题。在多种半监督目标检测评估设置下,ConsistentTeacher均展现出强大的性能。在仅使用10%标注的MS-COCO数据(ResNet-50为主干网络)条件下,模型达到40.0 mAP,相比此前基于伪标签的方法提升了约3 mAP。当在完全标注的MS-COCO数据基础上引入额外的无标签数据进行训练时,性能进一步提升至47.7 mAP。相关代码已开源,地址为:\url{https://github.com/Adamdad/ConsistentTeacher}。

代码仓库

adamdad/consistentteacher
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-object-detection-on-coco-1Consistent-Teacher
mAP: 25.5
semi-supervised-object-detection-on-coco-10Consistent-Teacher
detector: RetinaNet-Res50
mAP: 40.0
semi-supervised-object-detection-on-coco-100Consistent-Teacher
mAP: 48.20
semi-supervised-object-detection-on-coco-2Consistent-Teacher
mAP: 30.7

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
一致教师:面向减少半监督目标检测中不一致伪目标的研究 | 论文 | HyperAI超神经