Attention-based Domain Adaptation for Single Stage Detectors
This work addresses domain adaptation for resource-constrained single-stage detectors, offering a generic solution that outperforms existing methods, though it is incremental in adapting techniques from two-stage detectors.
The paper tackles the problem of domain adaptation for single-stage object detectors, which lack region proposals, by introducing an attention mechanism to focus adaptation on important regions, achieving state-of-the-art performance on benchmark datasets with SSD and YOLOv5.
While domain adaptation has been used to improve the performance of object detectors when the training and test data follow different distributions, previous work has mostly focused on two-stage detectors. This is because their use of region proposals makes it possible to perform local adaptation, which has been shown to significantly improve the adaptation effectiveness. Here, by contrast, we target single-stage architectures, which are better suited to resource-constrained detection than two-stage ones but do not provide region proposals. To nonetheless benefit from the strength of local adaptation, we introduce an attention mechanism that lets us identify the important regions on which adaptation should focus. Our method gradually adapts the features from global, image-level to local, instance-level. Our approach is generic and can be integrated into any single-stage detector. We demonstrate this on standard benchmark datasets by applying it to both SSD and YOLOv5. Furthermore, for equivalent single-stage architectures, our method outperforms the state-of-the-art domain adaptation techniques even though they were designed for specific detectors.