Do We Need Learnable Classifiers? A Hyperspectral Image Classification Algorithm Based on Attention-Enhanced ResBlock-in-ResBlock and ETF Classifier

Chuan Fu,Bo Du,Liangpei Zhang
DOI: https://doi.org/10.1109/tgrs.2024.3353383
IF: 8.2
2024-02-02
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Hyperspectral image (HSI) classification plays an important role in the field of remote sensing. Even though we can easily acquire hyperspectral remote sensing images, obtaining a large number of labeled hyperspectral samples remains challenging, especially in high-altitude or uninhabited areas. In this article, we propose a hyperspectral classification scheme for scenarios with insufficient labeled samples. This scheme is based on a variant of the ResBlock and a nonlearned classifier. First, we introduce a new and simplified backbone network for feature extraction. This network primarily consists of an attention-enhanced ResBlock-in-ResBlock module, which utilizes nested residual modules to enhance nonlinear expression and further optimizes the network using channel attention. Building upon this foundation, we address the challenge of achieving optimal classification with limited labeled training samples, a scenario described by the neural collapse theory. To address this, we introduce the equiangular tight frame (ETF) classifier and the dot-regression loss into hyperspectral classification. We conducted extensive comparative experiments using three HSI datasets. The experimental results demonstrate that our algorithm achieves superior classification accuracy, especially when the training sample size is small, outperforming other state-of-the-art algorithms. Furthermore, our algorithm maintains a low number of parameters and an overall complexity level.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?