Laser & Optoelectronics Progress, Volume. 61, Issue 24, 2428002(2024)

Hyperspectral-Image Classification Combining Spatial-Spectral Self-Attention and Multigranularity Feature Extraction

Lin Wei1,2, Zhe Chen1、*, and Yuping Yin3
Author Affiliations
  • 1School of Electronics and Information Engineering, Liaoning University of Engineering and Technology, Huludao 125105, Liaoning , China
  • 2Department of Basic Teaching, Liaoning University of Engineering and Technology, Huludao 125105, Liaoning , China
  • 3Faculty of Electrical and Control Engineering, Liaoning University of Engineering and Technology, Huludao 125105, Liaoning , China
  • show less

    For hyperspectral image (HSI) classification, although convolutional neural network (CNN)-based feature extraction methods have been widely applied and have achieved notable results, they still have limitations such as fixed receptive-field sizes and a tendency to overlook spatial-spectral correlations when extracting local features. In this regard, a Transformer network architecture that integrates multigranularity CNN and spatial-spectral self-attention (SSSA) is proposed herein. This architecture optimizes traditional CNN using multigranularity CNN by employing three-dimensional and two-dimensional convolutions to extract spatial-spectral and deep spatial features. Meanwhile, heterogeneous convolution is employed to finely extract multigranularity features, thereby overcoming the limitation of fixed kernel size in traditional CNN. In addition, to solve the problem of the neglect of local features in the self-attention mechanism in traditional Transformers, the mechanism is improved to enable the involved model to simultaneously construct global correlations for spatial and spectral information. Moreover, by introducing dual-channel depth-separable convolution for spatial-spectral-feature embedding, an effective connection between multigranularity CNN and SSSA is achieved. Further, experimental results show that owing to the successful extraction of local and global features, the involved model outperforms other mainstream HSI classification models on various datasets.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Lin Wei, Zhe Chen, Yuping Yin. Hyperspectral-Image Classification Combining Spatial-Spectral Self-Attention and Multigranularity Feature Extraction[J]. Laser & Optoelectronics Progress, 2024, 61(24): 2428002

    Download Citation

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

    Category: Remote Sensing and Sensors

    Received: Mar. 6, 2024

    Accepted: Apr. 18, 2024

    Published Online: Dec. 17, 2024

    The Author Email: Zhe Chen (415899149@qq.com)

    DOI:10.3788/LOP240832

    CSTR:32186.14.LOP240832

    Topics