Class-Balanced Contrastive Learning for Fine-Grained Airplane Detection

Yan Li,Qixiong Wang,Xiaoyan Luo,Jihao Yin
DOI: https://doi.org/10.1109/lgrs.2022.3220197
IF: 5.343
2022-01-01
IEEE Geoscience and Remote Sensing Letters
Abstract:Airplane detection and fine-grained recognition in remote sensing images are challenging due to class imbalance and high interclass indistinction. To alleviate these issues, we propose a class-balanced contrastive learning (CBCL) approach for airplane detection to exploit the correlation between samples in different images, which is rarely explored in previous research. Specifically, we first dynamically build class-balanced memory queues during training, which mitigates class imbalance by memorizing training samples. Upon class-balanced memory queues, hard triplet contrastive learning is introduced to increase the interclass discriminability, which enforces the maximum distance of the positive sample pair to be smaller than the minimum distance of the negative sample pair. We integrate the proposed CBCL strategy into oriented object detection frameworks for fine-grained airplane detection. The experimental results on the FAIR1M dataset reveal that several state-of-the-art algorithms with CBCL achieve significantly improvements.
What problem does this paper attempt to address?