Neural Architecture Search-Based Few-Shot Learning for Hyperspectral Image Classification
Fen Xiao, Han Xiang, Chunhong Cao, Xieping Gao
Abstract
Few-shot learning (FSL) has achieved promising performance in hyperspectral image classification (HSIC) with few labeled samples by designing a proper embedding feature extractor. However, the performance of embedding feature extractors relies on the design of efficient deep convolutional neural network architectures, which heavily depends on the expertise knowledge. Particularly, FSL requires extracting discriminative features effectively across different domains, which makes the construction even more challenging. In this paper, we propose a novel neural architecture search-based FSL model for HSI classification, called HCFSL-NAS. Three novel strategies are proposed in this work. First, a neural architecture search-based embedding feature extractor is developed to the FSL in HSIC, whose search space includes a group of proposed multi-scale convolutions with channel attention. Second, a multi-source learning framework is employed to aggregate abundant heterogeneous and homogeneous source data, which enables the powerful generalization of network to the HSIC with only few labeled samples. Finally, the pointwise-based cross-entropy loss and the pairwise-based adaptive sparse loss are jointly optimized to maximize inter-class distance and minimize the distance within a class simultaneously. Experimental results on four publicly hyperspectral data sets demonstrate that HCFSL-NAS outperforms both the exiting FSL methods and supervised learning methods for HSI classification with only few labeled samples. Code is available at: https://github.com/xh-captain/HCFSL-NAS.