Hyperspectral image classification based on three branch network with grouped spatial-spectral attention
Hyperspectral images have abundant spatial and spectral information.Numerous hyperspectral classification algorithms focus on the extraction and maximization of spatial and spectral information.Deep feature extraction networks generally extract spectral-spatial features using single-branch serial or double-branch parallel structures.However,single-branch structures may lead to mutual interference between features of spectral and spatial dimensions,and double-branch parallel structures tend to ignore the correlation between spatial and spectral features.This paper proposes a three-branch grouped spatial-spectral attention network(TGSSAN)to consider the differences and correlations between spatial and spectral features.TGSSAN can extract independent spectral-spatial features while preserving their correlation.This paper proposes the TGSSAN,which has three parallel branches(i.e.,spectral,spatial,and spectral-spatial branches).These branches can separately extract spectral,spatial,and spatial-spectral features.Different attention blocks are designed in three branches to enhance the discriminative capability of features.In particular,a grouped spatial-spectral attention mechanism is proposed in the spectral-spatial branch to obtain spatial and spectral attention simultaneously.Finally,three branch features are fused for classification.In the experiment,the proposed TGSSAN algorithm is compared with some advanced deep learning algorithms,such as SSRN,FDSSC,DBMA,DBDA,HResNetAM,and A2S2KResNet.The performance of different algorithms is evaluated on five hyperspectral data sets.Experimental results show that the proposed algorithm achieved superior classification performance on IP,PU,SA,HU,and HHK datasets.In particular,the proposed algorithm achieves higher classification accuracy despite limited training samples compared with the existing advanced algorithms.The TGSSAN method proposed in this paper improves the shortcomings of the single-branch serial and double-branch parallel structures for continuous extraction of spectral-spatial features,which can effectively extract image spectral-spatial feature information.The three attention blocks designed in this paper namely,spectral,grouped spatial-spectral,and spatial attention modules,can effectively enhance the feature discrimination capability and further improve the classification performance.
万方数据
维普
