A comparative attention framework for better few-shot object detection on aerial images

被引:1
作者
Le Jeune, Pierre [1 ,2 ]
Bahaduri, Bissmella [1 ]
Mokraoui, Anissa [1 ]
机构
[1] Univ Sorbonne Paris Nord, L2TI, 99 Ave Jean Baptiste Clement, F-93430 Villetaneuse, France
[2] COSE, 5 bis route St Leu, F-95360 Montmagny, France
关键词
Few-shot learning; Object detection; Aerial image; Few-shot object detection; Attention mechanisms;
D O I
10.1016/j.patcog.2024.111243
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-Shot Object Detection (FSOD) methods are mainly designed and evaluated on natural image datasets such as Pascal VOC and MS COCO. However, it is not clear whether the best methods for natural images are also the best for aerial images. Furthermore, a direct comparison of performance between FSOD methods difficult due to the wide variety of detection frameworks and training strategies. To this end, our contributions are twofold. First, we propose a benchmarking framework that provides a flexible environment to implement and compare attention-based FSOD methods. The proposed framework focuses on attention mechanisms and is divided into three modules: spatial alignment, global attention, and fusion layer. To remain competitive with existing methods, which often leverage complex training, we propose new augmentation techniques designed specifically for object detection. Using this framework, several FSOD methods are reimplemented and compared. This comparison highlights two distinct performance regimes on aerial and natural images: FSOD performs worse on aerial images. Our experiments confirm that small objects account for the poor performance. Small objects are difficult to detect, however in the few-shot regime, this challenge is largely reinforced. While the small object detection issue is well-known, to our knowledge this few-shot complication has never been reported in the literature. Second, always within the proposed framework, we develop a novel alignment method called Cross-Scales Query-Support Alignment (XQSA) for FSOD, to improve the detection of small objects. XQSA significantly outperforms the state-of-the-art on DOTA and DIOR, two aerial image datasets.
引用
收藏
页数:15
相关论文
共 40 条
  • [1] Bertinetto L, 2016, ADV NEUR IN, V29
  • [2] Chen H, 2018, AAAI CONF ARTIF INTE, P2836
  • [3] Dual-Awareness Attention for Few-Shot Object Detection
    Chen, Tung-, I
    Liu, Yueh-Cheng
    Su, Hung-Ting
    Chang, Yu-Cheng
    Lin, Yu-Hsiang
    Yeh, Jia-Fong
    Chen, Wen-Chin
    Hsu, Winston H.
    [J]. IEEE TRANSACTIONS ON MULTIMEDIA, 2023, 25 : 291 - 301
  • [4] Self-regularized prototypical network for few-shot semantic segmentation
    Ding, Henghui
    Zhang, Hui
    Jiang, Xudong
    [J]. PATTERN RECOGNITION, 2023, 133
  • [5] The Pascal Visual Object Classes (VOC) Challenge
    Everingham, Mark
    Van Gool, Luc
    Williams, Christopher K. I.
    Winn, John
    Zisserman, Andrew
    [J]. INTERNATIONAL JOURNAL OF COMPUTER VISION, 2010, 88 (02) : 303 - 338
  • [6] Few-Shot Object Detection with Attention-RPN and Multi-Relation Detector
    Fan, Qi
    Zhuo, Wei
    Tang, Chi-Keung
    Tai, Yu-Wing
    [J]. 2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, : 4012 - 4021
  • [7] Generalized Few-Shot Object Detection without Forgetting
    Fan, Zhibo
    Ma, Yuchen
    Li, Zeming
    Sun, Jian
    [J]. 2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, : 4525 - 4534
  • [8] Finn C, 2017, PR MACH LEARN RES, V70
  • [9] Few-Shot Object Detection with Fully Cross-Transformer
    Han, Guangxing
    Ma, Jiawei
    Huang, Shiyuan
    Chen, Long
    Chang, Shih-Fu
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, : 5311 - 5320
  • [10] Han GX, 2022, AAAI CONF ARTIF INTE, P780