Acta Optica Sinica, Volume. 41, Issue 3, 0310001(2021)

Hyperspectral Image Classification Based on Dilated Convolutional Attention Neural Network

Xiangdong Zhang*, Tengjun Wang, Shaojun Zhu, and Yun Yang
Author Affiliations
  • School of Geology Engineering and Geomatics, Chang′an University, Xi′an, Shaanxi 710054, China
  • show less

    In order to solve the problem of low classification accuracy of hyperspectral images in the case of limited training samples, we proposed a tandem three-dimensional(3D)-two-dimensional(2D) convolutional neural network model combining dilated convolution with attention mechanism. First, with the tandem 3D-2D convolutional neural network as the basic structure, the model used 3D convolution to simultaneously extract the spatial-spectral features of hyperspectral images and 2D convolution to further extract high-level spatial semantic information. Then, by introducing dilated convolution to enlarge the receptive field of the convolution kernel, we constructed a multi-scale feature extraction structure for the fusion of multi-scale features. Finally, the attention mechanism was applied to make the network pay attention to important spatial-spectral features and suppress noise and redundant information. Furthermore, we performed a comparative experiment between the proposed algorithm and four deep-learning-based algorithms on two common data sets. The results show that the proposed model achieves the most accurate classification results and effectively improves the classification accuracy under the condition of limited training samples.

    Tools

    Get Citation

    Copy Citation Text

    Xiangdong Zhang, Tengjun Wang, Shaojun Zhu, Yun Yang. Hyperspectral Image Classification Based on Dilated Convolutional Attention Neural Network[J]. Acta Optica Sinica, 2021, 41(3): 0310001

    Download Citation

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

    Category: Image Processing

    Received: Jul. 6, 2020

    Accepted: Sep. 8, 2020

    Published Online: Feb. 28, 2021

    The Author Email: Zhang Xiangdong (18755150056@163.com)

    DOI:10.3788/AOS202141.0310001

    Topics