• Laser & Optoelectronics Progress
  • Vol. 59, Issue 18, 1811001 (2022)
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
    DOI: 10.3788/LOP202259.1811001 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

    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.
    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