Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Unbiased Teacher for Semi-Supervised Object Detection

About

Semi-supervised learning, i.e., training networks with both labeled and unlabeled data, has made significant progress recently. However, existing works have primarily focused on image classification tasks and neglected object detection which requires more annotation effort. In this work, we revisit the Semi-Supervised Object Detection (SS-OD) and identify the pseudo-labeling bias issue in SS-OD. To address this, we introduce Unbiased Teacher, a simple yet effective approach that jointly trains a student and a gradually progressing teacher in a mutually-beneficial manner. Together with a class-balance loss to downweight overly confident pseudo-labels, Unbiased Teacher consistently improved state-of-the-art methods by significant margins on COCO-standard, COCO-additional, and VOC datasets. Specifically, Unbiased Teacher achieves 6.8 absolute mAP improvements against state-of-the-art method when using 1% of labeled data on MS-COCO, achieves around 10 mAP improvements against the supervised baseline when using only 0.5, 1, 2% of labeled data on MS-COCO.

Yen-Cheng Liu, Chih-Yao Ma, Zijian He, Chia-Wen Kuo, Kan Chen, Peizhao Zhang, Bichen Wu, Zsolt Kira, Peter Vajda• 2021

Related benchmarks

TaskDatasetResultRank
Object DetectionCOCO 2017 (val)
AP41.3
2643
Object DetectionPASCAL VOC 2007 (test)
mAP50.34
844
Object DetectionMS-COCO 2017 (val)
mAP41.3
237
Object DetectionCOCO (minival)
mAP41.3
184
Object DetectionPascal VOC
mAP77.37
88
Object DetectionCOCO standard (5% labeled)
mAP28.35
70
Semantic segmentationWHU dataset (test)
F1 Score78.6
61
Object DetectionCOCO standard (10%)
mAP31.5
54
Instance SegmentationNWPU VHR-10 (test)
mIoU65.34
48
Ship Instance SegmentationHRSID-Inshore (test)
mIoU47.44
48
Showing 10 of 67 rows

Other info

Code

Follow for update