A Novel Semi-Supervised Long-Tailed Learning Framework With Spatial Neighborhood Information for Hyperspectral Image Classification

Abstract

Deep learning technologies have been successfully applied to hyperspectral (HS) image classification with remarkable performance. However, compared with traditional machine learning methods, neural networks usually need more data. In remote sensing (RS) research, obtaining a large number of labeled HS data is very difficult and expensive work. Simultaneously, the distribution of feature information is bound to be unbalanced, and tends to conform to the long tail. At present, the neighborhood information of unlabeled samples is usually ignored in HS image classification tasks based on semi-supervised learning. In this letter, we propose a new semi-supervised long-tail learning framework based on spatial neighborhood information (SLN-SNI), which can complete the HS image classification task under unbalanced small sample data. Specifically, a new semi-supervised learning strategy is proposed. On this basis, a new method to determine the label of unlabeled samples based on spatial neighborhood information (SNI) is proposed. The coarse classification results divided into three situations are judged again, and the accuracy of pseudo labels is improved. The performance of the proposed method is tested on three public HS image datasets. Compared with the current advanced methods have achieved a certain improvement.