Discriminative Prototype Learning for Few-Shot Object Detection in Remote-Sensing Images
M. Guo, Yanan You, Fang Liu
Abstract
Few-shot object detection (FSOD) in remote sensing images, which aims to detect never-seen objects with few training samples, has attracted wide attention. Some recent works leverage meta-learning to tackle this challenging task and achieve promising performance. However, information attenuation during feature extraction and simple prototype representation hamper further improvement in detecting novel classes. In this article, we propose a novel meta-learning-based FSOD approach named DPL-Net. Specially, within the meta-learning-based framework, performing role-specific feature extraction in query and support branches, DPL-Net adopts a Fine-grained Information Fusion (FIF) module to capture scale-aware information within query regions of interest (RoIs) and a Multi-frequency Information Enhancement (MIE) module to retain the spectral information of support samples, respectively. Moreover, considering the variability of remote sensing objects, a Discriminative Prototype Learning (DPL) strategy is developed to rectify the ambiguous distribution of support samples for more representative class-aware prototypes. Experiments on two benchmark datasets (NWPU VHR-10 and DIOR) demonstrate that our method effectively improves the performance of meta-learning in detecting remote sensing images with limited training data.