Dynamic Retraining-Updating Mean Teacher for Source-Free Object Detection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Khanh, Trinh Le Ba, Nguyen, Huy-Hung, Pham, Long Hoang, Tran, Duong Nguyen-Ngoc, Jeon, Jae Wook
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917731020832768
author Khanh, Trinh Le Ba
Nguyen, Huy-Hung
Pham, Long Hoang
Tran, Duong Nguyen-Ngoc
Jeon, Jae Wook
author_facet Khanh, Trinh Le Ba
Nguyen, Huy-Hung
Pham, Long Hoang
Tran, Duong Nguyen-Ngoc
Jeon, Jae Wook
contents In object detection, unsupervised domain adaptation (UDA) aims to transfer knowledge from a labeled source domain to an unlabeled target domain. However, UDA's reliance on labeled source data restricts its adaptability in privacy-related scenarios. This study focuses on source-free object detection (SFOD), which adapts a source-trained detector to an unlabeled target domain without using labeled source data. Recent advancements in self-training, particularly with the Mean Teacher (MT) framework, show promise for SFOD deployment. However, the absence of source supervision significantly compromises the stability of these approaches. We identify two primary issues, (1) uncontrollable degradation of the teacher model due to inopportune updates from the student model, and (2) the student model's tendency to replicate errors from incorrect pseudo labels, leading to it being trapped in a local optimum. Both factors contribute to a detrimental circular dependency, resulting in rapid performance degradation in recent self-training frameworks. To tackle these challenges, we propose the Dynamic Retraining-Updating (DRU) mechanism, which actively manages the student training and teacher updating processes to achieve co-evolutionary training. Additionally, we introduce Historical Student Loss to mitigate the influence of incorrect pseudo labels. Our method achieves state-of-the-art performance in the SFOD setting on multiple domain adaptation benchmarks, comparable to or even surpassing advanced UDA methods. The code will be released at https://github.com/lbktrinh/DRU
format Preprint
id arxiv_https___arxiv_org_abs_2407_16497
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Dynamic Retraining-Updating Mean Teacher for Source-Free Object Detection
Khanh, Trinh Le Ba
Nguyen, Huy-Hung
Pham, Long Hoang
Tran, Duong Nguyen-Ngoc
Jeon, Jae Wook
Computer Vision and Pattern Recognition
In object detection, unsupervised domain adaptation (UDA) aims to transfer knowledge from a labeled source domain to an unlabeled target domain. However, UDA's reliance on labeled source data restricts its adaptability in privacy-related scenarios. This study focuses on source-free object detection (SFOD), which adapts a source-trained detector to an unlabeled target domain without using labeled source data. Recent advancements in self-training, particularly with the Mean Teacher (MT) framework, show promise for SFOD deployment. However, the absence of source supervision significantly compromises the stability of these approaches. We identify two primary issues, (1) uncontrollable degradation of the teacher model due to inopportune updates from the student model, and (2) the student model's tendency to replicate errors from incorrect pseudo labels, leading to it being trapped in a local optimum. Both factors contribute to a detrimental circular dependency, resulting in rapid performance degradation in recent self-training frameworks. To tackle these challenges, we propose the Dynamic Retraining-Updating (DRU) mechanism, which actively manages the student training and teacher updating processes to achieve co-evolutionary training. Additionally, we introduce Historical Student Loss to mitigate the influence of incorrect pseudo labels. Our method achieves state-of-the-art performance in the SFOD setting on multiple domain adaptation benchmarks, comparable to or even surpassing advanced UDA methods. The code will be released at https://github.com/lbktrinh/DRU
title Dynamic Retraining-Updating Mean Teacher for Source-Free Object Detection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2407.16497