Transformer-enhanced convolutional neural network with self-supervised learning for hyperspectral image classification

Yifan Sun,Zhixiang Xue,Mofan Dai,Xibing Zuo,Kuiliang Gao,Bing Liu
DOI: https://doi.org/10.1117/12.2667329
2023-02-10
Abstract:In recent years, deep-learning-based hyperspectral image (HSI) processing and analysis have made significant progress. However, models with high performance require sufficient training samples because scarce labeled samples limit their generalization ability. To solve this problem, we adopt a self-supervised learning strategy and conduct self-training for a neural network model by obtaining different views of the same sample (positive pairs). As a result, the network can learn representative features for classification from unlabeled samples. In addition, to increase the spatial receptive field compared with the use of conventional convolutions, we use the transformer to capture long-distance dependencies for feature enhancement and adequately combine their advantages. Experimental results on two publicly available HSI datasets demonstrate that the proposed method can extract robust features through self-training on unlabeled samples and can be adapted to HSI classification tasks under the small sample conditions.
Engineering,Environmental Science,Computer Science
What problem does this paper attempt to address?