Journal of Optoelectronics · Laser, Volume. 35, Issue 2, 143(2024)

Multi-scale feature and dual-attention mechanism for hyperspectral image classification

LV Huanhuan1,2, ZHANG Juntong1, and ZHANG Hui2、*
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    Hyperspectral image classification methods based on the classical convolutional neural network (CNN) have some problems,such as insufficient expression of key detail features and a large number of samples for training.Aiming at these problems,this paper proposes a hyperspectral image classification model with multi-scale features and dual-attention mechanism.Firstly,using 3D convolution,the spatial-spectral features of images can be directly extracted,and transposed convolution is adopted to get more details of the feature map.Then,a feature extraction module is built through convolution kernels of different sizes to achieve multi-scale feature fusion under different receptive fields.Finally,the dual-attention mechanism is designed to suppress the confused regional features and highlight the distinguishing features.The experimental results on two hyperspectral images show that when 10% and 0.5% samples are randomly selected as training samples for each class of ground object,the overall classification accuracy of the proposed model is improved to 99.44% and 98.86%,respectively.This model can obtain higher classification accuracy than some mainstream deep-learning classification models.Since the model can focus on more important detailed features during feature extraction,the classification effect is improved.

    Tools

    Get Citation

    Copy Citation Text

    LV Huanhuan, ZHANG Juntong, ZHANG Hui. Multi-scale feature and dual-attention mechanism for hyperspectral image classification[J]. Journal of Optoelectronics · Laser, 2024, 35(2): 143

    Download Citation

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

    Received: Sep. 5, 2022

    Accepted: --

    Published Online: Sep. 25, 2024

    The Author Email: ZHANG Hui (03013@zjhu.edu.cn)

    DOI:10.16136/j.joel.2024.02.0618

    Topics