基于卷积残差注意力网络的高光谱图像分类算法

Hyperspectral Image Classification Based on Convolutional Residual Attention Network

  • 摘要: 基于卷积神经网络(CNN: Convolutional Neural Network)的高光谱图像(HSI: Hyperspectral Image)分类方法显著提高了分类性能,然而在获取深层语义特征方面存在瓶颈,且随着网络层数增加,计算成本明显上升。注意力机制允许网络集中注意力于特定区域或通道,提高对关键信息的感知,且有助于处理图像中的长距离依赖关系,促使网络同时获取局部特征和全局特征。因此,本文提出一种基于卷积残差注意力网络的高光谱图像分类算法。首先,设计了光谱-空间特征提取模块,使网络能够更全面地捕捉高光谱图像中的信息;其次,设计注意力增强骨干,通过注意力机制和残差网络的交互,更加关注重要的通道和特定区域,提高特征的判别能力并增强对空间信息的感知能力;最后引入注意力对比学习,增强样本类内间的关联度与类间的区分度。在三个公开的高光谱数据集上的实验结果表明,相较于已有代表性高光谱图像分类算法,所提方法的分类性能更优异。

     

    Abstract: ‍ ‍Hyperspectral images contain rich spatial and spectral information, and several methods exist for the accurate classification of hyperspectral images. To address the problems in hyperspectral remote sensing images, such as the difficulty of deep semantic feature extraction and the increase of network computational cost with increasing network layers, this paper proposes a hyperspectral image classification algorithm based on convolutional residual attention network. First, the spectral-spatial feature extraction module was designed to enable the network to capture the information in the hyperspectral image more comprehensively. Second, the attention enhancement backbone was designed to address important channels and specific areas through the interaction between the attention mechanism and the residual network, improving the discrimination ability of features and enhancing the perception of spatial information. Finally, attention contrastive learning was introduced to enhance the intra-class correlation and inter-class discrimination of samples. A fixed proportion of training and testing samples were randomly selected on three public hyperspectral datasets for experimental analysis. Indicators such as overall accuracy (OA), average accuracy (AA), and Kappa coefficient were used to evaluate the classification results. The dataset selected in this paper includes both high- and low-resolution hyperspectral data, which can fully test the generalization of the proposed method. Experimental results on three datasets show that compared with some representative hyperspectral image-classification algorithms, the proposed method has better classification performance. The proposed method plays an important role in promoting research in this field. However, the proposed method has limitations including the use of single data source and difficulty in determining the input scale. In future research, we will address the exploration from the perspective of multi-scale and multi-source data and consider irregular and regular inputs to improve the classification performance of hyperspectral images.

     

/

返回文章
返回