Mutual Attention Feature Alignment in Cross-domain Detection.

Jian Hu,Zhongshu Chen,Maolin Luo,Jiahang Li,Xianlong Tian,Lin Zuo
DOI: https://doi.org/10.1109/smartworld-uic-atc-scalcom-digitaltwin-pricomp-metaverse56740.2022.00171
2022-01-01
Abstract:Domain shift exists among the data from different scenes, that is, the object detection model trained in a well-labeled domain has a poor performance in an unlabeled domain. Many works attempt to address this issue by incorporating domain adaptation (DA) techniques into object detection. However, these cross-domain detection works have not considered the different contributions from various features to domain alignment. In addition, most recent cross-domain detection methods use two-stage detector as the baseline model, which has low object detection efficiency. To solve these problems, this paper proposes the mutual attention feature alignment network (MAFA), which utilizes the mutual attention scores to highlight the features with high transferability and suppress the features with low transferability for domain alignment. After that, we successfully build a one-stage cross-domain detection framework. We use the MAFA to align the image-level feature and instance-level feature distributions on the YOLOv5s, which can significantly improve the model detection. At last, we experimentally validate our model with extensive experiments.
What problem does this paper attempt to address?