基于空谱注意力机制及预激活残差网络的高光谱图像分类算法

Hyperspectral Image Classification Based on Spatial Spectral Attention and Pre-Activation Residual Networks

  • 摘要: 面向高光谱图像分类的许多深度学习算法中,由于提取的空谱特征表示鉴别性不足,其模型的分类性能有待提高。针对该问题,本文提出了一种基于空谱注意力机制及预激活残差网络的高光谱图像分类算法。首先,设计了基于空谱注意力机制的空谱特征提取模块,对空谱特征进行重校准,为空谱特征在后续联合学习时能专注于更具辨别力的通道和空间位置提供保证;其次,设计了基于预激活残差网络的空谱特征联合学习模块,其中预激活残差网络改进了原始残差构建块的网络结构,从而能在利用注意力机制重校准的空谱特征的联合学习时捕获更具鉴别性的深层空谱特征,以提高分类器的分类性能。实验结果表明,和已有的一些高光谱图像分类算法相比,所提出的算法的分类准确率更高,表明该算法能有效地获得判别能力更强的空谱特征表示。

     

    Abstract: ‍ ‍In many deep learning algorithms for hyperspectral image classification, the classification performance of their models needed to be improved due to the poor discriminant representation of extracted spatial spectral features. To solve this problem, this paper proposed a hyperspectral image classification algorithm based on spatial spectral attention and pre-activation residual network. Firstly, a spatial spectral feature extraction module based on the spatial spectral attention mechanism was designed to recalibrate the spatial spectral features, so as to ensure that the spatial spectral features can focus on more discriminative channels and spatial positions during subsequent joint learning; secondly, a joint learning module of spatial spectral features based on the pre-activation residual network was designed, in which the pre-activated residual network improved the network structure of the original residual building block, so that it could capture more discriminative deep spatial spectral features during joint learning of spatial spectral features recalibrated by attention mechanism, and therefore the classification performance of the classifier could be improved. The experimental results showed that compared with some existing hyperspectral image classification algorithms, the proposed algorithm could attain higher classification accuracy. It indicated that the algorithm can effectively obtain spatial spectral features representation with stronger discriminative ability.

     

/

返回文章
返回