LiteSCANet: An Efficient Lightweight Network Based on Spectral and Channel-Wise Attention for Hyperspectral Image Classification

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 cl...

Descripción completa

Guardado en:
Detalles Bibliográficos
Autores principales: Su Qiao, Xue-Mei Dong, Jiangtao Peng, Weiwei Sun
Formato: article
Lenguaje:EN
Publicado: IEEE 2021
Materias:
Acceso en línea:https://doaj.org/article/71468806f9ef491ab9ebcf4065cd60e0
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
Descripción
Sumario: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 computational-resource-limited platforms. Taking into account the efficiency and accuracy, this article 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 datasets show that our proposed model achieves an excellent trade-off between efficiency and accuracy compared with the other six existing networks.