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...
Guardado en:
Autores principales: | , , , |
---|---|
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!
|
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. |
---|