4 个月前

无偏教师在半监督目标检测中的应用

无偏教师在半监督目标检测中的应用

摘要

半监督学习,即使用标记数据和未标记数据共同训练网络,近年来取得了显著进展。然而,现有的研究主要集中在图像分类任务上,而忽视了需要更多标注努力的目标检测任务。在本工作中,我们重新审视了半监督目标检测(SS-OD),并识别出其中存在的伪标签偏差问题。为了解决这一问题,我们引入了一种简单而有效的方法——无偏教师(Unbiased Teacher),该方法以互惠互利的方式联合训练一个学生模型和一个逐步进步的教师模型。结合类别平衡损失来降低过于自信的伪标签权重,无偏教师在COCO标准、COCO附加和VOC数据集上显著提升了现有最先进方法的表现。具体而言,当使用MS-COCO数据集中1%的标记数据时,无偏教师相比最先进方法实现了6.8个绝对mAP的提升;而在仅使用0.5%、1%和2%的标记数据时,无偏教师相对于有监督基线方法在MS-COCO数据集上实现了约10个mAP的提升。

代码仓库

yeonsikch/Unbiased-Teacher
pytorch
GitHub 中提及
facebookresearch/unbiased-teacher
官方
pytorch
GitHub 中提及
hikvision-research/SSOD
pytorch
GitHub 中提及
hik-lab/ssod
pytorch
GitHub 中提及

基准测试

基准方法指标
semi-supervised-object-detection-on-coco-0-5Unbiased Teacher
mAP: 16.94± 0.23
semi-supervised-object-detection-on-coco-1Unbiased Teacher
mAP: 20.75± 0.12
semi-supervised-object-detection-on-coco-10Unbiased Teacher
mAP: 31.5± 0.10
semi-supervised-object-detection-on-coco-100UNBIASED TEACHER
mAP: 41.3
semi-supervised-object-detection-on-coco-2Unbiased Teacher
mAP: 24.324.30 ± 0.07
semi-supervised-object-detection-on-coco-5Unbiased Teacher
mAP: 28.27± 0.11
semi-supervised-person-bounding-box-detectionUnbiased-Teacher
Person Bounding Box AP: 39.18

用 AI 构建 AI

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

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

Hyper Newsletters

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