Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1811001(2022)

Hyperspectral Image Classification Based on Convolution Neural Network with Attention Mechanism

Wenhao Chen1, Jing He1、*, and Gang Liu1,2
Author Affiliations
  • 1College of Earth Sciences, Chengdu University of Technology, Chengdu 610059, Sichuan , China
  • 2State Key Laboratory of Geohazard Prevention and Geoenvironmental Protection, Chengdu University of Technology, Chengdu 610059, Sichuan , China
  • show less

    In recent years, convolutional neural network (CNN), as a representative of the deep learning method, has gradually become a research hotspot in the field of hyperspectral image (HSI) classification because it does not require complex data preprocessing and feature design. In this paper, a deep CNN model with an attention mechanism is proposed based on an existing neural network model combined with the HSI data characteristics. The model used a residual structure to construct a deep CNN to extract spatial-spectral features and introduced a channel attention mechanism to recalibrate the extracted features. According to different importance levels of features, the attention mechanism assigned different weights to features on different channels, highlighted important features, and controlled unimportant features. Experiments were conducted at Indian Pines and Pavia University to validate the proposed technique. When the spatial size of the dataset was 19 × 19, the Indian Pines and Pavia University datasets were divided into 3∶1∶6 and 1∶1∶8, respectively. Additionally, these datasets have the best classification accuracy. The average overall accuracy, average accuracy, and average Kappa coefficient obtained are 99.55%, 99.31%, and 99.45%, respectively. The experimental results show that deep CNN with residual structure can extract high spatial-spectral features of the HSI. Additionally, the attention mechanism recalibrates the features to strengthen the important features, thereby effectively enhancing the HSI’s classification accuracy.

    Tools

    Get Citation

    Copy Citation Text

    Wenhao Chen, Jing He, Gang Liu. Hyperspectral Image Classification Based on Convolution Neural Network with Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1811001

    Download Citation

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

    Category: Imaging Systems

    Received: May. 31, 2021

    Accepted: Jul. 20, 2021

    Published Online: Aug. 22, 2022

    The Author Email: He Jing (xiao00yao@163.com)

    DOI:10.3788/LOP202259.1811001

    Topics