• Acta Optica Sinica
  • Vol. 41, Issue 3, 0310001 (2021)
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
    DOI: 10.3788/AOS202141.0310001 Cite this Article Set citation alerts
    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 Copy Citation Text show less

    Abstract

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