Cross-Domain Few-Shot Contrastive Learning for Hyperspectral Images Classification
Suhua Zhang, Zhikui Chen, Dan Wang, Z. Jane Wang
Abstract
Deep learning has achieved impressive results on Hyperspectral image (HSI) classification, which generally requires sufficient training samples and a huge number of parameters. However, it is challenging to label HSIs, and likely only a few samples are available in practice. Learning a large number of parameters by the model is also resource-intensive. This paper proposes an HSI classification model that achieves promising classification performance with fewer parameters in few-shot settings. The proposed model adopts the residual 3D-CNN as feature extraction network, and contrastive learning is introduced to learn more discriminative representations for HSIs which can conquer the obstacles from HSIs’ high inter-class similarity and large intra-class variance. The proposed few-shot contrastive learning HSI classification model is tested on five popular HSI datasets and outperforms the state-of-the-art models.