Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

CAT: Exploiting Inter-Class Dynamics for Domain Adaptive Object Detection

About

Domain adaptive object detection aims to adapt detection models to domains where annotated data is unavailable. Existing methods have been proposed to address the domain gap using the semi-supervised student-teacher framework. However, a fundamental issue arises from the class imbalance in the labelled training set, which can result in inaccurate pseudo-labels. The relationship between classes, especially where one class is a majority and the other minority, has a large impact on class bias. We propose Class-Aware Teacher (CAT) to address the class bias issue in the domain adaptation setting. In our work, we approximate the class relationships with our Inter-Class Relation module (ICRm) and exploit it to reduce the bias within the model. In this way, we are able to apply augmentations to highly related classes, both inter- and intra-domain, to boost the performance of minority classes while having minimal impact on majority classes. We further reduce the bias by implementing a class-relation weight to our classification loss. Experiments conducted on various datasets and ablation studies show that our method is able to address the class bias in the domain adaptation setting. On the Cityscapes to Foggy Cityscapes dataset, we attained a 52.5 mAP, a substantial improvement over the 51.2 mAP achieved by the state-of-the-art method.

Mikhail Kennerley, Jian-Gang Wang, Bharadwaj Veeravalli, Robby T. Tan• 2024

Related benchmarks

TaskDatasetResultRank
Object DetectionBDD100K (val)
mAP38.5
60
Object DetectionFoggy Cityscapes
mAP52.5
47
Object DetectionClipart, Comic, and Watercolor
mAP (Clipart)49.1
22
Object DetectionCityscapes to Foggy Cityscapes severity 0.02 1.0 (val)
AP (Person)44.6
22
Object DetectionPASCAL VOC → Clipart1k (test)
AP (aero)40.5
20
Object DetectionBDD100K
mAP38.5
19
Object DetectionFoggy Cityscapes 0.02 (test)
AP (person)44.6
18
Object DetectionFoggy Cityscapes full (val)
AP (Person)44.6
15
Lung Nodule DetectionIF-LUNA16
mAP10.4
8
Object DetectionIF-CCT 1.0 (test)
Rabbit AP21.2
8
Showing 10 of 12 rows

Other info

Code

Follow for update