Back to Search Start Over

Hyperspectral Image Classification Based on Two-Branch Spectral–Spatial-Feature Attention Network.

Authors :
Wu, Hanjie
Li, Dan
Wang, Yujian
Li, Xiaojun
Kong, Fanqiang
Wang, Qiang
Source :
Remote Sensing. Nov2021, Vol. 13 Issue 21, p4262. 1p.
Publication Year :
2021

Abstract

Although most of deep-learning-based hyperspectral image (HSI) classification methods achieve great performance, there still remains a challenge to utilize small-size training samples to remarkably enhance the classification accuracy. To tackle this challenge, a novel two-branch spectral–spatial-feature attention network (TSSFAN) for HSI classification is proposed in this paper. Firstly, two inputs with different spectral dimensions and spatial sizes are constructed, which can not only reduce the redundancy of the original dataset but also accurately explore the spectral and spatial features. Then, we design two parallel 3DCNN branches with attention modules, in which one focuses on extracting spectral features and adaptively learning the more discriminative spectral channels, and the other focuses on exploring spatial features and adaptively learning the more discriminative spatial structures. Next, the feature attention module is constructed to automatically adjust the weights of different features based on their contributions for classification to remarkably improve the classification performance. Finally, we design the hybrid architecture of 3D–2DCNN to acquire the final classification result, which can significantly decrease the sophistication of the network. Experimental results on three HSI datasets indicate that our presented TSSFAN method outperforms several of the most advanced classification methods. [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
20724292
Volume :
13
Issue :
21
Database :
Academic Search Index
Journal :
Remote Sensing
Publication Type :
Academic Journal
Accession number :
153593874
Full Text :
https://doi.org/10.3390/rs13214262