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

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

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

convolutional neural networkhyperspectral imageattention mechanismresidual network

涂兵、贺燕、胡江红、陈云云

展开 >

南京信息工程大学物理与光电工程学院, 江苏南京 210044

南京信息工程大学江苏省大气海洋光电探测重点实验室, 江苏南京 210044

南京信息工程大学江苏省气象光子学与光电探测国际合作联合实验室,江苏南京 210044

湖南理工学院信息科学与工程学院,湖南岳阳 414002

展开 >

卷积神经网络 高光谱图像 注意力机制 残差网络

国家自然科学基金南京信息工程大学科研启动经费项目

622712002023r091

2024

信号处理
中国电子学会

信号处理

CSTPCD北大核心
影响因子:1.502
ISSN:1003-0530
年,卷(期):2024.40(3)
  • 36