LiteSCANet: An Efficient Lightweight Network Based on Spectral and Channel-Wise Attention for Hyperspectral Image Classification
Qiao Su, Xuemei Dong, Jiangtao Peng, Weiwei Sun
Abstract
Deep learning (DL) algorithms have been demonstrated to have great potential in hyperspectral image (HSI) classification. However, most DL methods mainly focus on improving classification performance, neglecting the computational cost. In order to broaden the application scenarios of DL-based HSI classification methods, it is necessary to develop lightweight and fast models to fit the deployment needs of computationalresource-limited platforms. Taking into account the efficency and accuracy, this paper designs a lightweight network architecture based on spectral and channel-wise attention modules, namely LiteSCANet, for HSI classification. It contains a residual double branch structure, which makes the model effectively extract spectral-spatial features and achieve good performance with fast inference speed and low computational consumption (i.e., floating-point operations, number of parameters and graphics processing unit memory usage). The experiment results on four benchmark data sets show that our proposed model achieves an excellent trade-off between efficiency and accuracy compared with the other six existing networks.