Contrastive Mean Teacher for Domain Adaptive Object Detectors

Shengcao Cao, D. Joshi, Liangyan Gui, Yu-Xiong Wang
5/1/2023

No official code found. Best implementation: Shengcao-Cao/CMT (50 stars, Apache-2.0, last commit 2023-09-11).

Abstract

Object detectors often suffer from the domain gap between training (source domain) and real-world applications (target domain). Mean-teacher self-training is a powerful paradigm in unsupervised domain adaptation for object detection, but it struggles with low-quality pseudo-labels. In this work, we identify the intriguing alignment and synergy between mean-teacher self-training and contrastive learning. Motivated by this, we propose Contrastive Mean Teacher (CMT) - a unified, general-purpose framework with the two paradigms naturally integrated to maximize beneficial learning signals. Instead of using pseudo-labels solely for final predictions, our strategy extracts object-level features using pseudo-labels and optimizes them via contrastive learning, without requiring labels in the target domain. When combined with recent mean-teacher self-training methods, CMT leads to new state-of-the-art target-domain performance: 51.9% mAP on Foggy Cityscapes, outperforming the previously best by 2.1% mAP. Notably, CMT can stabilize performance and provide more significant gains as pseudo-label noise increases.

View on arXivView PDF

Code Implementations(1)

Shengcao-Cao/CMTCommunity100%
Apache-2.0

[CVPR 2023] CMT: Contrastive Mean Teacher for Domain Adaptive Object Detectors

509Shell, PythonMar 21, 20233 years agoApache-2.0

Similar Papers

Cite this paper

@article{cao2023contrastive,
  title  = {Contrastive Mean Teacher for Domain Adaptive Object Detectors},
  author = {Shengcao Cao and D. Joshi and Liangyan Gui and Yu-Xiong Wang},
  year   = {2023},
  eprint = {2305.03034},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/pdf/2305.03034},
  journal = {CVPR 2023 2023}
}

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/2305.03034.svg)](https://codeofpaper.com/paper/2305.03034)

Discussion