Hyperspectral Image Classification via Deep Network with Attention Mechanism and Multigroup Strategy

Jun Wang,Jinyue Sun,Erlei Zhang,Tian Zhang,Kai Yu,Jinye Peng
DOI: https://doi.org/10.1016/j.eswa.2023.119904
IF: 8.5
2023-03-29
Expert Systems with Applications
Abstract:In order to address the correlation between feature maps and the context information of spectral sequences that are not considered in the classi- fication of hyperspectral image (HSI) classification, we proposed a novel deep network with the Attention Mechanism and Multigroup Strategy (AMM-Net), which combines a spatial feature extraction module with attention mechanism and a spectral feature extraction module with multi-grouping strategy for im- proving classification accuracy. The spatial feature extraction module with at- tention mechanism is based on the channel spatial attention module (CSAM), which can deal with the output features of the convolution layer differently, fo- cus on more useful classification features and enhance the expression ability of the model. The spectral feature extraction module with multi-grouping strat- egy includes a long and short time memory (LSTM) network to explore more spectral information and features. Finally, the deep spatial-spectral features are inputs of the softmax classifier. In the experiment, three HSIs benchmarks are utilized to evaluate the performance of the proposed method. The experi- mental results show that the AMM-Net has excellent performance.
computer science, artificial intelligence,engineering, electrical & electronic,operations research & management science
What problem does this paper attempt to address?