Self-Supervised Learning With Prediction of Image Scale and Spectral Order for Hyperspectral Image Classification

Xiaofei Yang,Weijia Cao,Yao Lu,Yicong Zhou
DOI: https://doi.org/10.1109/tgrs.2022.3225663
IF: 8.2
2022-12-28
IEEE Transactions on Geoscience and Remote Sensing
Abstract:In recent years, convolutional neural networks (CNNs) have achieved great success in hyperspectral image (HSI) classification attributed to their unparalleled capacity to extract the local information. However, to successfully learn the high-level semantic image features, they always require massive amounts of manually labeled data during the training process, which is expensive, scarce, and impractical, and severely hinders the improvement of supervised deep learning methods. To alleviate these burdens, we present self-supervised learning (SSL) methods for HSI classification by a pretraining model using extensive unlabeled data and fine-tuning the HSI target classification. In this article, we propose a new method for learning image characteristics by training a CNN to recognize the image scale (IS) that is applied to the HSIs. In addition, we propose a multipretext task (MT) method to learn stable and good feature representations combing two different pretext task methods and contrastive loss function. We evaluate the proposed methods in SSL benchmarks on four benchmark HSIs datasets. The experiment results demonstrate that the proposed methods outperform the traditional supervised deep learning methods when large amounts of unlabeled HSIs data are used. Moreover, it demonstrates that the SSL method is promising to alleviate dependence on manually labeled data of HSI classification. Finally, our research contributes to the creation and refinement of SSL methods for pretextual tasks within the HSIs community.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?