Laser & Optoelectronics Progress, Volume. 61, Issue 4, 0437010(2024)

Multi-Scale Feature Extraction Method of Hyperspectral Image with Attention Mechanism

Zhangchi Xu, Baofeng Guo*, Wenhao Wu, Jingyun You, and Xiaotong Su
Author Affiliations
  • School of Automation, Hangzhou Dianzi University, Hangzhou 310018, Zhejiang, China
  • show less

    In recent years, with the development of deep learning, feature extraction methods based on deep learning have shown promising results in hyperspectral data processing. We propose a multi-scale hyperspectral image feature extraction method with an attention mechanism, including two parts that are respectively used to extract spectral features and spatial features. We use a score fusion strategy to combine these features. In the spectral feature extraction network, the attention mechanism is used to alleviate the vanishing gradient problem caused by spectral high-dimension and multi-scale spectral features are extracted. In the spatial feature extraction network, the attention mechanism helps branch networks extract important information by making the network backbone focus on important parts in the neighborhood. Five spectral feature extraction methods, three spatial feature extraction methods and three spatial-spectral joint feature extraction methods are used to perform comparative experiments on three datasets. The experimental results show that the proposed method can steadily and effectively improve the classification accuracy of hyperspectral images.

    Tools

    Get Citation

    Copy Citation Text

    Zhangchi Xu, Baofeng Guo, Wenhao Wu, Jingyun You, Xiaotong Su. Multi-Scale Feature Extraction Method of Hyperspectral Image with Attention Mechanism[J]. Laser & Optoelectronics Progress, 2024, 61(4): 0437010

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Digital Image Processing

    Received: Mar. 28, 2023

    Accepted: May. 4, 2023

    Published Online: Feb. 26, 2024

    The Author Email: Guo Baofeng (gbf@hdu.edu.cn)

    DOI:10.3788/LOP230974

    Topics