Few-Shot Scene Classification with Attention Mechanism in Remote Sensing

ZHANG Duona,ZHAO Hongjia,LU Yuanyao,CUI Jian,ZHANG Baochang
DOI: https://doi.org/10.1088/1742-6596/1961/1/012015
2021-01-01
Journal of Physics Conference Series
Abstract:Remote sensing scene classification is a hot research topic in computer vision and it is of great significance to the semantic understanding of remote sensing images. At present, remote sensing scene classification methods based on deep learning occupy a dominant position in this field. However, it suffers from the lack of samples and poor model generalization ability in actual application scenarios. Therefore, this paper proposes a few-shot remote scene classification method based on attention mechanism, and designs a structure of dual-branches similarity measurement. This method is based on the meta-learning training strategy to divide the dataset into tasks. At the meantime, the input images are divided into blocks to preserve the feature distribution in the remote sensing image. Then the lightweight attention module is introduced into the feature extraction network to reduce the risk of overfitting and ensure the acquisition of discriminative features. Finally we add a dual-branches similarity measurement module on the basis of Earth Mover’s Distance to improve the discriminative ability of the classifier. The results show that compared with the classic small-sample learning method, the few-shot remote scene classification method proposed in this paper can significantly improve the classification performance.
What problem does this paper attempt to address?