Mixed Pseudo Labels for Semi-Supervised Object Detection

Zeming Chen, Wenwei Zhang, Xinjiang Wang, Kai Chen, Zhi Wang
12/12/2023
cs.CVcs.AI

No official code found. Best implementation: Czm369/MixPL (70 stars, Apache-2.0, last commit 2024-03-07).

Abstract

While the pseudo-label method has demonstrated considerable success in semi-supervised object detection tasks, this paper uncovers notable limitations within this approach. Specifically, the pseudo-label method tends to amplify the inherent strengths of the detector while accentuating its weaknesses, which is manifested in the missed detection of pseudo-labels, particularly for small and tail category objects. To overcome these challenges, this paper proposes Mixed Pseudo Labels (MixPL), consisting of Mixup and Mosaic for pseudo-labeled data, to mitigate the negative impact of missed detections and balance the model's learning across different object scales. Additionally, the model's detection performance on tail categories is improved by resampling labeled data with relevant instances. Notably, MixPL consistently improves the performance of various detectors and obtains new state-of-the-art results with Faster R-CNN, FCOS, and DINO on COCO-Standard and COCO-Full benchmarks. Furthermore, MixPL also exhibits good scalability on large models, improving DINO Swin-L by 2.5% mAP and achieving nontrivial new records (60.2% mAP) on the COCO val2017 benchmark without extra annotations.

View on arXivView PDF

Code Implementations(1)

Czm369/MixPLCommunity100%
Apache-2.0

Mixed Pseudo Labels for Semi-Supervised Object Detection

704Shell, PythonNov 30, 20232 years agoApache-2.0

Similar Papers

Cite this paper

@article{chen2023mixed,
  title  = {Mixed Pseudo Labels for Semi-Supervised Object Detection},
  author = {Zeming Chen and Wenwei Zhang and Xinjiang Wang and Kai Chen and Zhi Wang},
  year   = {2023},
  eprint = {2312.07006},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2312.07006v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2312.07006.svg)](https://codeofpaper.com/paper/2312.07006)

Discussion