Lightweight Spectral–Spatial Attention Network for Hyperspectral Image Classification
Ying Cui, Jinbiao Xia, Zhiteng Wang, Shan Gao, Liguo Wang
Abstract
Convolutional neural networks (CNNs) have exhibited extraordinary achievements in hyperspectral image (HSI) classification due to their detailed representation of features. However, the improvement of classification accuracy often leads to an evident increase in the complexity of the model, which makes it challenging for the model with the state-of-the-art performance to be applied in the actual scene. Considering MobileNetV3 as a lightweight feature extractor, this article proposes a model suitable for HSI classification based on MobileNetV3. To decrease the problem of massive redundant calculations in the existing spatial attention module, this article proposes a more concise and efficient spatial attention module based on the visual feature maps experiment. Besides, multiclass focal-loss is applied to solve the problem that the difficulty of classification varies for each sample. The experimental results demonstrate that in the case of using very few training sets, the proposed model can tremendously reduce the number of calculations and parameters while maintaining high accuracy.