EN中文

自然资源遥感 >

2025 , Vol. 37 >Issue 5: 101 - 112

DOI: https://doi.org/10.6046/zrzyyg.2024191

技术方法

改进的残差式3D-CNN和近邻注意力的高光谱遥感图像分类

  • 潘增滢 , 1, 2 ,
  • 吴瑞姣 3 ,
  • 林易丰 1, 2 ,
  • 翁谦 , 1, 2, 4 ,
  • 林嘉雯 1, 2, 4
展开
  • 1.福州大学计算机与大数据学院，福州 350116
  • 2.福州大学福建省网络计算与智能信息处理重点实验室，福州 350116
  • 3.福建省地质测绘院，福州 350116
  • 4.福州大学空间数据挖掘与信息共享教育部重点实验室，福州 350116
翁 谦（1983-），男，博士，副教授，主要从事计算机视觉和时空大数据研究。Email：。

潘增滢（2000-），男，硕士研究生，主要从事高光谱遥感信息处理研究。Email：。

Copy editor: 张仙

收稿日期: 2024-05-24

  修回日期: 2024-09-21

  网络出版日期: 2026-06-03

基金资助

福建省自然科学基金项目“人机协同的自然资源要素提取关键技术研究”(2023J01432)

国家自然科学基金项目“基于深度迁移学习网络的高分影像土地利用分类方法研究”(41801324)

收起

Hyperspectral remote sensing image classification using improved residual 3D-CNN and neighborhood attention

  • PAN Zengying , 1, 2 ,
  • WU Ruijiao 3 ,
  • LIN Yifeng 1, 2 ,
  • WENG Qian , 1, 2, 4 ,
  • LIN Jiawen 1, 2, 4
Expand
  • 1. Fuzhou University College of Computer and Data Science，Fuzhou 350116，China
  • 2. Fujian Key Laboratory of Network Computing and Intelligent Information Processing（Fuzhou University），Fuzhou 350116，China
  • 3. Fujian Geologic Surveying and Mapping Institute，Fuzhou 350116，China
  • 4. Key Lab of Spatial Data Mining & Information Sharing，Ministry of Education（Fuzhou University），Fuzhou 350116，China

Received date: 2024-05-24

  Revised date: 2024-09-21

  Online published: 2026-06-03

Fold

摘要

在引起广泛关注的高光谱遥感图像分类中，同物异谱、同谱异物和少样本都大大限制了分类方法的性能。为了充分挖掘高光谱图像的空间-光谱特征，该文提出了一种改进的残差卷积和近邻注意力网络用于高光谱遥感图像分类。该方法包含3个部分：结合了残差连接和3D卷积神经网络（3D convolutional neural network，3D-CNN）的残差式光谱特征提取模块、使用混合卷积的空间-光谱特征融合模块、用于增强模型对同质区域的关注能力的近邻注意力模块。在3个公开的高光谱数据集Indian pines，Pavia University，Houston2013上的实验结果显示，相比近期先进高光谱分类方法，所提方法有更高的分类精度，且在使用10%以下训练样本的前提下总体精度可分别达到99.39%，99.67%和98.64%，实现了少样本下的高精度分类。

关键词： 高光谱图像分类; 卷积神经网络; 残差连接; 近邻注意力

本文引用格式

潘增滢 , 吴瑞姣 , 林易丰 , 翁谦 , 林嘉雯 . 改进的残差式3D-CNN和近邻注意力的高光谱遥感图像分类[J]. 自然资源遥感, 2025 , 37(5) : 101 -112 . DOI: 10.6046/zrzyyg.2024191

Abstract

Hyperspectral remote sensing image classification has attracted widespread attention，yet the performance of classification methods remains greatly limited by challenges such as spectral variability （same object with different spectra），spectral confusion （different objects with similar spectra），and limited availability of training samples. To fully exploit the spatial-spectral features of hyperspectral images，this study proposed an improved network integrating residual convolution and neighborhood attention mechanisms. The proposed method consists of：（1） a residual-based spectral feature extraction module combining residual connections and a 3D convolutional neural network （3D-CNN）；（2） a spatial-spectral feature fusion module using mixed convolutions；and （3） a neighborhood attention module designed to enhance the model's ability to focus on homogeneous regions. Experiments were conducted on three public hyperspectral datasets-Indian Pines，Pavia University，and Houston 2013. The results demonstrate that the proposed method achieves higher classification accuracy compared to recent state-of-the-art approaches. Using less than 10% of the samples for training，it attains overall accuracies of 99.39%，99.67%，and 98.64%，respectively，confirming its capability for high-accuracy classification under small-sample conditions.

Key words： hyperspectral image classification; convolutional neural network; residual connection; neighborhood attention

0 引言

高光谱图像（hyperspectral images，HSI）是一种能够获取大量连续光谱信息的图像数据，相较于多光谱图像，HSI具有更窄的光谱波段信息，且可以覆盖可见光、近红外、中红外到远红外的更大光谱范围，通常拥有数十个甚至数百个连续波段[1]，具有广泛的应用潜力。在高光谱图像处理中，高光谱遥感图像分类是重要的下游任务之一，可实现对地物目标的逐像素识别[2]，在土地利用[3]、植被覆盖[4]、矿产资源勘探[5]等方面具有重要意义。
早期基于机器学习的高光谱图像分类方法普遍依赖于手工特征提取或浅层特征描述，缺乏自动学习深层次特征的能力，导致模型的泛化能力和适用性较差[6]。随着深度学习的发展，越来越多深度学习方法被用于高光谱图像分类，基于自编码器的方法[7]和基于递归神经网络的方法[8-9]被证明其在高光谱分类领域中有不错的表现，但是这些方法注重于将HSI视为序列而忽略了图像本身的空间关系，导致分类精度不高。
卷积神经网络（convolutional neural networks，CNN）因其强大的图像特征提取能力在许多计算机视觉领域，如图像分类[10]、目标检测[11]、语义分割[12]等方向取得了巨大的进展。在高光谱分类领域中，CNN独特的滑动窗口的计算方式使其能够有效提取图像中的纹理、边缘等空间信息。2015年Hu等[13]首次将CNN应用于高光谱分类领域，证明了CNN在高光谱分类中的潜力。随后越来越多基于CNN的方法被提出，如Makantasis等[14]使用随机主成分分析（randomized principal component analysis，R-PCA）结合2D-CNN实现了高光谱分类的深监督学习；Zhu等[15]提出残差光谱空间注意力网络（residual spectral-spatial attention network，RSSAN），通过向网络引入光谱-空间注意层，取得了更高的精度；Kang等[16]通过结合2D-CNN中的ResNet和DenseNet定义了一个双路径网络来学习联合特征。尽管2D-CNN在传统图像处理领域有巨大的优势，却无法有效提取HSI特有的光谱特征，而3D-CNN采用三维卷积核进行运算，可以同时对HSI本身丰富的光谱信息和空间信息进行提取，使得分类的精度进一步得到提升。Hamida等[17]将3D-CNN引入高光谱遥感影像分类中，并结合1D-CNN，设计了一个较为高效的架构；频谱-空间残差网络（spectral-spatial residual network，SSRN）[18]通过在网络中加入残差结构，从而缓解因网络深度过深导致的过拟合、梯度消失等问题；郑宗生等[19]改进了3D-CNN用于提取高频和低频空谱信息，并结合三维注意力机制，提高了网络的表征能力；为解决3D-CNN计算量激增的问题，3D-CNN与2D-CNN混合的方式[20]被提出，有效降低了参数量并提升运行速度；Roy等[21]提出了一个混合3D-CNN与2D-CNN的端到端的基于注意力的自适应频谱空间核改进残差网络（attention-based adaptive spectral-spatial kernel improved residual network，A2S2K），可自适地选择不同大小的3D卷积核从而提高分类性能。
近年来由于ViT[22]在图像领域的成功，Transformer也被应用于高光谱分类领域。如结合了数学形态学加强了Transformer对于局部语义元素的建模从而提出了morphFormer[23]，更关注中心像素贡献的SQSFormer[24]，以及充分利用HSI中的光谱-空间信息的基于高斯注意力的轻量化高光谱分类网络（light self-Gaussian-attention，LSGA）[25]。这些工作有效利用了Transformer对于序列的处理能力。然而由于Transformer本身软归纳[26]的运算方式，自注意力机制（self-attention，SA）通常需要大量的数据进行训练才能得到最好的结果。而对于标签样本不足的高光谱分类问题，基于Transformer的方法性能受到很大限制。与之相比，CNN硬归纳的特点和滑动窗口对图像数据本身的契合，能够更好地提取图像的局部空间特征，使得其在仅有少量标记样本的情况下也能得到较为鲁棒的结果。此外，高光谱图像通常具有局部空间平滑性[27]的特性，这就要求分类网络更关注局部空间内同质区域。因此，本文聚焦在传统的CNN上，以期改进传统的CNN模型获得更佳的分类性能。
针对以上问题，本文提出了一个改进的残差式3D-CNN和近邻注意力网络（improved residual 3D-CNN and neighborhood attention network，IR3NAN）用于HSI分类。具体地，网络主要由残差式光谱特征提取（residual based spectral feature extraction，RSFEM）模块、空间-光谱特征融合（space spectral feature fusion module，SSFFM）模块和近邻注意力（neighborhood attention module，NAM）模块构成。进入网络前首先通过主成分分析（principal component analysis，PCA）对原始数据降维处理；随后RSFEM通过结合3D-CNN和残差结构的构建，使得网络能够学习到图像更深层次的光谱语义特征；紧接着NAM模块通过计算空间块上各个像素点与中心像素点的相似性获得近邻空间注意力权重；最后，使用SSFFM模块实现光谱特征与空间特征的高效融合。

1 IR3NAN网络

1.1 网络结构

本文所提出的IR3NAN网络模型如图1所示，包含RSFEM模块、NAM模块和SSFFM模块3个部分。原始HSI数据首先通过PCA降维和图像块分割处理之后，在2个串联的RSFEM模块中使用残差结构改进的3D-CNN网络对图像中隐含的光谱特征进行深度提取，残差结构的融入能有效保留浅层特征，同时减少过拟合；随后使用NAM模块计算每个像素点与中心像素在欧氏空间上的距离并赋予权重，以强化网络对同质区域的关注程度；之后，通过结合使用3D-CNN和2D-CNN的SSFFM模块将光谱特征和空间特征有效融合，最后经平均池化和多层感知机（multilayer perceptron，MLP）分类得到最终分类结果。
图1 改进的残差3D-CNN和近邻注意力网络

Fig.1 Improved residual 3D-CNN and nearest neighbor attention network

假设给定的原有HSI数据X0∈RH×W×S，其中H，W，S分别为高、宽和波段数。为降低光谱冗余，本文首先使用PCA对原始HSI数据光谱维度降维，在降低维度的同时尽可能保留更多的光谱信息。取PCA计算所得前B个特征值所对应的特征向量组成映射函数F∈RS×B，则XPCA=X0×F，此时将原有HSI的光谱维度从S降为B，降维后的HSI为XPCA∈RH×W×B。
早期高光谱分类任务中通常只考虑光谱信息，网络仅输入单一光谱维度的信息，输入数据为X∈R1×1×S，S为输入的光谱维度，导致缺少空间信息影响分类精度。大量研究表明，分类像素与邻域像素之间的空间相关性起着重要作用[28]，因此本文基于空谱联合的思想，以中心像素邻域内大小为h×w的局部空间块作为输入，并以中心像素点的标签作为该空间块的标签。首先需要对高光谱图像边缘进行像素填充，上下和左右分别填充（h-1）/2行和（w-1）/2列，随后剔除原有数据中标签为背景的无效样本后按照设定大小进行图像块裁剪，得到的图像块为X1∈Rh×w×S，组成的样本集作为网络的输入。基于遥感图像的旋转不变性，本文所用图像块的宽度与高度保持一致。

1.2 RSFEM模块

高维的HSI蕴含着丰富的光谱信息，但也存在不少的冗余的噪声波段，大大影响分类的准确性，因此需要先从光谱信息中提取到有用的浅层和深层特征，并减少无用的噪声信息。本节所提出的RSFEM模块主要通过堆叠3D卷积模块完成对高光谱信息的提取，结构图如图2所示。该模块主要由5个连续的3D卷积模块构成，每个3D卷积模块又由卷积层、批归一化层（batch normalization，BN）和激活层ReLU函数串联而成。该模块输入的数据为经过PCA降维后分割的空间块Xin∈Rh×w×B，模块中第一个3D卷积块的卷积核尺寸为1×1×5，在初步提取局部光谱特征的同时生成更多的特征图像块，输出为X1∈Rh×w×B×24。我们观察到，相邻像素间普遍存在光谱混合的现象，因此堆叠一个卷积核尺寸为3×3×1的3D卷积块以减少光谱混合带来的冗余。随后由2个串联的1×1×5的3D卷积块对光谱信息提取更深层的语义特征。最后用一层1×1×B的3D卷积块融合所有光谱维度得到全新的特征表示，得到的输出为X5∈Rh×w×1×B，并通过重构，输出为Xout∈Rh×w×B。本文使用堆叠方式加深网络以获得更深光谱的语义特征，但也导致了网络层数过深，容易出现网络退化的问题。为此，本文在堆叠的3D-CNN模块的基础上加入残差连接，具体而言，在第一个卷积块和第二个卷积块、第一个卷积块和第四个卷积块、第二个卷积块和第四个卷积块之间都添加了残差连接，一方面使得浅层特征也能够通过残差相加向后传播，另一方面更有效地提取了更丰富的多层次光谱语义信息。
图2 RSFEM模块

Fig.2 Residual based spectral feature extraction module

1.3 NAM模块

根据空间相关性定律，在HSI中地物之间相关性与距离成反比，中心像素的类别通常与邻域空间的类别相同，即为同质区域。基于这一特点，如果网络在分类过程中能更关注同质区域，就可以提取到更具分类价值的特征。但直接使用降维后的光谱值来判断同质区域，易因同物异谱和同谱异物而分类错误。因此，本文在RSFEM的基础上提出了NAM模块，该模块以RSFEM模块得到的深层光谱特征作为相似度计算标准，相比降维后的光谱特征，RSFEM获得的光谱语义特征更能够衡量局部空间内像素的同质性，有利于分类性能的提升。NAM模块通过计算周围像素点和中心像素点之间的欧式距离来生成NAM矩阵。计算公式如下：
Dij= $\sqrt[　]{\sum _{k=1}^{B}（{X}_{ij}^{k}-{X}_{c}^{k}{）}^{2}}$，
式中：Dij为在空间块第i行j列的像素与中心像素之间的欧式距离；B为光谱维的数量； ${X}_{ij}^{k}$为第i行j列的像素第k个光谱维度的特征值； ${X}_{c}^{k}$为空间块中心像素第k个光谱维度的特征值。
假设该模块的输入表示为Xin∈Rh×w×B，经过权重赋值和残差相加后，该模块最终的输出为：
Xout=Xin·W+Xin，Xout∈Rh×w×B ，
Wij=1- $\frac{{D}_{ij}}{max（{D}_{ij}）}$，
式中Wij为给第i行j列像素所赋予的权值。

1.4 SSFFM模块

RSFEM从HSI数据中提取的光谱语义特征缺少空间语义信息，基于空谱联合的思想，本文提出一种SSFFM模块，通过图像的空间信息与光谱语义特征的融合提高最终的分类性能。SSFFM模块由2层3D卷积块和4层2D卷积块构成，并引入了逐点卷积和残差连接，以有效融合RSFEM得到的光谱语义特征和NAM模块输出的注意力空间特征，模块结构图如图3所示。每个3D卷积块和2D卷积块均由卷积层、BN层和ReLU激活函数串联而成。假设输入模块的数据为Xin∈Rh×w×B，首先使用2个卷积核大小为3×3×7和3×3×5的3D卷积块对空间光谱信息进行联合提取，并逐步增加特征图像块的个数，将输出的特征图像块的尺寸扩展为X1∈Rh×w×B×24。随后将光谱维和特征维合并得到特征图X2∈Rh×w×C，其中C=24×B。后经一个1×1的逐点卷积块将维度降到60，最后对所有空间-光谱特征使用3个串联的3×3的2D卷积块进行融合。方法中使用的逐点卷积块在减小计算量的同时也可以减少网络的参数。此外，在每层2D-CNN之间使用残差连接，既有效降低网络训练难度避免网络退化，也更好地融合空间信息和光谱信息，提升特征表征能力。
图3 SSFFM模块

Fig.3 Space spectral feature fusion module

2 实验与分析

为了验证本文方法的有效性和鲁棒性，本文选择了空间分辨率不同的3个数据集进行实验：印第安纳松树（Indian pines，IP）、帕维亚大学（Pavia University，PU）和休斯顿2013（Houston2013，HO）。

2.1 数据集介绍

IP数据集于1992年由机载可见光/红外成像光谱仪从美国印第安纳州采集，共包括16个植被类型，图像尺寸为145像素×145像素，空间分辨率大小为20 m。去除水吸收影响的20个噪声波段，剩余的200个波段被用于训练，光谱成像范围为400~2 500 nm。
PU数据集于2003年由德国的机载反射光学光谱成像仪在意大利的帕维亚城拍摄获得，包含9种城市土地覆盖类型，图像尺寸为610像素×340像素，空间分辨率大小为1.3 m。除去噪声波段剩余的103个波段用于训练，光谱成像范围为430~860 nm。
HO数据集为2013年IEEE GRSS数据融合竞赛[29]中，由美国国家机载激光测绘中心拍摄获得的美国休斯顿大学校园的城市区域，包含15种不同土地覆盖类型，图像尺寸为349像素×1 905像素，空间分辨率大小2.5 m。共计144个波段，波长范围为380~1 050 nm。

2.2 实验设置

本文实验使用固定比例随机分割训练集和测试集，由于各个数据集样本数量不同，为确保几个数据集样本数量分布更加均匀，在IP，PU，HO数据集分别随机选取10%，2%和5%的样本作为训练集（如Zhong 等[18]IP数据使用20%样本和PU数据采用10%的训练样本，Roy等[21]，Chen等[24]，Ma等[25]将训练集设置为全10%），其余样本作为测试集，训练周期为150轮，采用Adam优化器，学习率设置为0.001，权重衰减为10-6。实验采用总体精度（overall accuracy，OA），平均精度（average accuracy，AA）和Kappa系数[30]对所提出的方法的精度进行整体评估，而对于每个单独类别的分类精度则采用精确率P来表示，每个实验均进行10次重复训练，取所有结果的平均值作为最终评估标准。
评价指标中OA表示样本中正确分类总数所占所有样本比重，能较为直观地展示模型分正确分类的比例，计算公式为：
OA=（ $\sum _{i=1}^{D}$Ti）/M ，
式中：Ti为第i个类别被正确分类的样本数；D为类别的个数；M为样本总数。
P表示所有预测为正的样本中实际为正样本的概率，用于展示各个类别的分类精度，计算公式为：
P=TP/（TP+FP） ，
式中：TP为分类预测结果为正样本，实际也为正样本的数量；FP为分类预测结果为正样本，实际为负样本的数量，即误报的负样本数量；TN为分类预测结果实际也为负样本的数量。
AA表示每个类别样本分类精度的平均值，可以展现模型各个类别分类精度的总体情况，计算公式为：
AA=（ $\sum _{i=1}^{D}$Pi）/D ，
式中Pi为第i个类别的精确率。
Kappa系数是一个用于一致性检验的指标，取值在[-1，1]之间。类别不均衡时，模型如果将样本全部预测为多数类别也能获得得到较高的OA，因此需要给数量多的类别进行“惩罚”，Kappa系数的计算基于混淆矩阵，相比OA仅考虑分类正确的数量，Kappa系数还额外考虑了错分和漏分的样本，可以视为OA和AA的综合考虑。计算公式为：
Kappa=（po-pe）/（1-pe） ，
pe=（ $\sum _{i=1}^{C}$ai×bi）/M2 ，
式中：po为总体分类精度，即OA；pe为预期分类精度；ai为类别i真实样本的个数；bi为模型预测为类别i的个数；M为样本的总数。

2.3 实验结果

为验证本文提出的IR3NAN的优越性，本文选取7种深度学习分类方法，1D-CNN，2D-CNN[14]，SSRN[18]，HybridSN[20]，A2S2K[21]，morphFormer[23]和LSGA[25]在3个公开数据集上进行实验对比。实验结果如表1，表2和表3，其中黑色加粗表示精度最高的方法。
表1 不同方法在IP数据集上的分类结果

Tab.1 Classification Results of Different Methods on IP Datasets

类别 1D-CNN 2D-CNN SSRN HybridSN A2S2K morphFormer LSGA IR3NAN
苜蓿 22.93 55.61 72.93 79.76 79.27 89.51 88.29 97.56
未翻耕的玉米地 59.95 78.06 84.92 95.35 98.67 97.11 98.79 98.96
翻耕过的玉米地 51.02 82.53 89.21 93.36 98.33 96.49 95.92 98.89
玉米地 38.78 82.35 91.60 88.69 97.56 95.82 97.32 98.50
牧草区 73.72 95.52 96.74 92.34 98.92 96.46 98.32 98.69
草地与树木 86.51 97.84 99.47 97.69 99.27 99.74 99.89 99.82
已收割的牧草区 45.60 42.80 66.40 64.80 92.40 99.20 100.00 99.60
风干的草料 94.26 99.49 98.09 97.19 99.81 99.79 99.88 99.93
燕麦 18.89 65.56 9.44 94.44 99.44 84.44 97.78 98.33
未翻耕的大豆田 48.00 76.99 86.82 96.49 98.89 98.74 99.22 99.05
翻耕过的大豆田 67.89 84.51 97.24 95.71 99.38 98.45 99.50 99.67
已清理的大豆田 43.73 82.73 98.45 91.46 99.29 95.13 98.65 99.16
小麦 93.35 99.19 99.68 98.11 100.00 99.73 99.78 100.00
树林 92.99 98.13 92.68 99.08 99.93 99.87 99.99 100.00
建筑物、草地、树木、车道 39.39 89.57 89.34 88.1 98.67 98.30 98.79 100.00
石、钢、塔、楼 89.76 98.81 98.57 81.79 98.10 94.76 94.88 96.79
OA/% 66.67 86.77 92.84 95.08 99.03 98.04 98.92 99.39
AA/% 60.42 83.11 85.72 90.9 97.37 96.47 97.94 99.06
Kappa 0.619 8 0.847 5 0.918 8 0.943 9 0.988 9 0.977 6 0.987 7 0.993 0
表2 不同方法在PU数据集上的分类结果

Tab.2 Classification results of different methods on the PU dataset

类别 1D-CNN 2D-CNN SSRN HybridSN A2S2K morphFormer LSGA IR3NAN
沥青路面 87.81 95.72 99.96 96.58 99.87 98.75 99.57 99.99
草地 90.85 97.50 99.60 99.77 99.97 99.91 99.88 99.98
碎石 57.84 69.74 92.47 86.49 94.52 89.45 89.90 97.49
树木 85.84 98.36 98.66 88.23 99.07 96.19 97.81 98.61
涂漆金属板 98.75 99.94 100.00 86.66 99.98 100.00 100.00 99.98
裸土 72.67 84.88 99.69 99.60 100.00 99.95 99.97 99.94
沥青 79.44 89.11 99.93 90.75 99.46 99.86 99.51 100.00
自阻砖 84.66 90.74 98.53 89.17 98.46 93.43 96.43 99.28
阴影 99.52 99.67 99.91 78.88 99.94 93.72 97.60 98.84
OA/% 85.82 93.72 99.18 95.71 99.48 98.26 98.85 99.67
AA/% 84.15 91.74 98.75 90.68 99.03 96.81 97.85 99.35
Kappa 0.812 3 0.916 1 0.989 2 0.943 0 0.993 1 0.976 9 0.984 8 0.995 6
表3 不同方法在HO集上的分类结果

Tab.3 Classification results of different methods on the HO dataset

类别 1D-CNN 2D-CNN SSRN HybridSN A2S2K morphFormer LSGA IR3NAN
健康的草地 93.90 94.65 93.36 91.83 96.37 94.77 94.85 96.83
受压力的草地 94.81 97.63 96.54 95.50 99.13 97.95 98.85 99.21
人造草皮 95.39 95.44 92.63 97.98 99.24 99.31 99.63 99.89
树木 94.03 98.60 99.87 95.53 99.93 97.52 97.66 99.93
土壤 95.86 97.37 99.61 100.00 100.00 100.00 100.00 100.00
水 68.01 67.21 78.16 85.37 84.04 83.75 96.25 95.44
住宅区 80.29 95.12 93.66 94.52 98.24 98.68 98.94 99.05
商业区 78.10 84.39 91.09 91.14 94.93 94.52 94.16 94.94
道路 73.29 89.06 83.47 93.73 98.55 97.80 97.79 98.11
高速公路 71.85 87.05 88.26 98.77 99.35 99.47 99.86 99.97
铁路 63.66 92.69 90.84 99.85 99.78 98.32 99.95 100.00
停车场1 75.47 92.18 88.17 96.51 97.84 96.82 96.77 99.12
停车场2 23.19 88.01 80.87 89.78 91.36 89.21 92.62 92.29
网球场 89.44 98.91 94.84 73.33 99.96 99.89 100.00 100.00
跑道 94.79 100.00 99.34 97.99 100.00 99.89 100.00 100.00
OA/% 81.11 92.91 92.14 95.19 98.21 97.45 97.96 98.64
AA/% 79.47 91.89 91.38 93.46 97.25 96.53 97.82 98.32
Kappa 0.794 7 0.923 0 0.914 5 0.947 7 0.980 6 0.972 3 0.977 8 0.985 2
表1展示的是各个方法在IP数据集上的精度对比。由表1可见本文方法在OA，AA和Kappa系数都取得了最佳的结果，分别为99.39%，99.06%和0.993 0，比A2S2K方法分别多了0.36百分点，1.69百分点和0.004 1。而对比该数据集的16个类别中，本文方法在9个类别上取得了最高的分类精度，其中3个类别的精度达到100%。此外，由于IP数据集类别严重不均衡，在所提供的地物类型中，苜蓿、已收割的牧草区和燕麦这3个类别在10%训练比例以下，本文仅分别选择了5，3，2个的样本作为训练，但是分类精度均能取得较高水平，分别为97.56%，99.6%和98.33%，其中苜蓿类别的精度远超其他方法，相比A2S2K分类精度高出8.05个百分点，说明本文方法对少样本的类别也能很好区分。
表2展示的是各个方法在PU数据集上的精度对比。由于PU数据集为航拍数据，有更高的空间分辨率和更少的噪声，仅取其中2%的样本作为训练。由表2可知，各个对比方法在此数据集上都有较高的精度，而本文所提出方法在OA，AA和Kappa上均取得了最佳的结果，分别为99.67%，99.35%和0.995 6，比第二好的方法A2S2K分别多出0.19百分点，0.32百分点和0.002 5。在数据集的9个类别中，有5个类别达到了最高的精度，其中在沥青路面和沥青这2个相似地物上的分类精度分别达到了99.99%和100%，均高于其他对比方法，说明本文方法即便在地物光谱特征相似的情况下也能很好地区分。
表3展示的是各个方法在HO数据集上的精度对比。由于HO数据集为城市地物数据集，数据集标注较为稀疏，因此具有更大的挑战性，由表3可知本文在3个指标上都取得了最佳的精度，分别为98.64%，98.32%和0.985 2。且在15个类别中的12个类别达到了最高的精度。尤其在树木、铁路、网球场和跑道这4个类别的精度达到了100%，说明了本文方法能够适应多种不同复杂类型的城市地物。
综合表1—3可知，在几个对比方法中，1D-CNN表现最差，在PU数据集和HO数据集中的OA分别为85.82%和81.11%，而对具有更多噪声信息的IP数据集，OA仅为66.67%。这是因为1D-CNN使用像素点的光谱序列作为输入，忽略了作为图像本身的空间信息，造成精度较低。2D-CNN，SSRN和HybridSN方法虽然考虑图像的空间信息，但受限于光谱特征提取能力，表现一般。MorphFormer方法虽然充分应用了Transformer处理序列数据的能力来提取光谱特征，但是仅用浅层的卷积对光谱信息预提取，受噪声信息影响大，性能提升有限。而A2S2K方法和LSGA方法在3个数据集上性能优于前述方法，但OA，AA和Kappa系数都略低于本文方法，说明本文方法提取的空间-光谱特征表征性能更优。而本文方法在卫星高光谱数据和质量更好的航空数据上都取得最佳性能，进一步证明了本文方法的鲁棒性。

2.4 可视化分析

图4—6为本文方法以及其他对比方法的可视化分类结果图。综合图4—6可见，1D-CNN方法的分类图中存在很多噪点，错分严重。2D-CNN，SSRN和HybridSN在各个类别边缘都存在错分的现象，这是由于其对空间-光谱特征提取能力有限导致的。A2S2K，morphFomer和LSGA方法表现较好，但是在IP数据集中，地物面积较小的少样本区域中仍然会出现较多的错分现象，本文方法在3个数据集的分类结果中均有较好的表现，仅在个别类别边缘处有少量的错分，说明了所提出方法的有效性。同时为了证明本文方法对高光谱图像空间-光谱特征的提取能力，在3个数据集对网络的中间特征（线性层前）进行了t-SNE特征可视化分析，如图7所示。由图7可知，3个数据集中不同类别样本在降维后的特征空间上均有明显的的聚类中心，相同类别的样本以簇的形式在该中心周围分布，说明了本文网络得到的中间特征具有更好的类间可分性。
图4 各个方法在IP数据集分类图对比

Fig.4 Comparison of classification maps of various methods on the IP dataset

图5 各个方法在PU数据集分类图对比

Fig.5 Comparison of classification maps of various methods on the PU dataset

图6 各个方法在HO数据集分类图对比

Fig.6 Comparison of classification maps of various methods on the HO dataset

图7 3个数据集的t-SNE特征可视化

Fig.7 Visualization of t-SNE features for 3 datasets

2.5 训练样本数量的影响

为证明本文提出方法的鲁棒性，本节在3个数据集使用了不同比例的训练样本进行对比实验，结果如图8所示，其中1D-CNN和2D-CNN方法精度较低，因此不做展示。在3个数据集中，本文所提出方法都有较好的表现，当训练样本进一步削减时（IP数据集5%，PU数据集1%，HO数据集2%），在IP数据集本文方法略高于A2S2K方法，在PU和HO数据集相比SSRN和LSGA精度有较大提升。当训练样本增多时，在IP数据集，训练样本为15%时本文方法依旧能超过其他对比方法，当训练样本到达20%时，精度达到精度顶峰并趋于稳定。而在PU数据集和HO数据集，几个方法之间的差距随样本增加而变小，但本文方法依旧在精度上优于其他方法，说明本文方法能充分应对少标注样本的情况，且在样本量增多时受冗余特征影响小，也能取得很好的结果，证明了本文方法的鲁棒性。
图8 3个数据集下不同训练样本精度对比

Fig.8 Accuracy comparison of different training samples under 3 data sets

3 讨论

3.1 消融实验

为证明本文所提出的模块的有效性，本节对所提出的RSFEM模块和NAM模块进行了消融实验，结果如表4所示，Baseline为不使用RSFEM和NAM模块的基准网络。可见，同时使用了RSFEM和NAM时，3个数据集上的3个指标都取得最佳值，其OA相比基准网络分别提升了0.35，0.41和0.25百分点。在IP数据集中，使用RSFEM和NAM模块后相比Baseline的OA分别提高了0.34和0.07百分点，RSFEM精度提升更大，这是因为IP数据集为卫星数据，空间分辨率低，冗余光谱和噪声多，此外，由于农田植被地物分布集中，空间关系作用较小，而NA模块是基于空间注意力分配来引导网络学习不同像素点的关系，因此NAM模块的在此数据集的作用较小。而对于PU数据集，相比Baseline 2个模块对OA与AA的提升分别为0.19，0.04百分点和0.44，0.15百分点，而当RSFEM和NAM模块一起使用时，OA和AA则分别提升了0.41和1.15百分点，相比单独使用2个模块互相作用时会有更大的精度提升，并且AA的大幅度提升可以推测模型因此克服了某个易混淆的小类别样本。可见，在PU数据集，当高光谱数据空间分辨率好，空间特征增加时，随着地物分布的复杂程度提升，光谱特征和空间特征都对分类结果有更大影响，因此需要联合考虑。而在HO数据集上，地物类型为空间分布更零散而复杂的城市地物，使用RSFEM与NAM模块相比Baseline OA分别提升0.14和0.11百分点，二者差距不大，对分类结果均有一定的提升。当2个模块结合时能够适应各种不同类型的数据集，充分证明了本文提出模块的有效性。
表4 各个模块的消融对比

Tab.4 Ablation comparison of each module

数据集 精度 Baseline RSFEM NAM RSFEM+NAM
IP OA/% 99.04 99.38 99.11 99.39
AA/% 98.31 99.06 98.38 99.06
Kappa 0.989 1 0.993 0 0.989 9 0.993 0
PU OA/% 99.26 99.45 99.30 99.67
AA/% 98.19 98.63 98.34 99.35
Kappa 0.990 2 0.992 7 0.990 7 0.995 6
HO OA/% 98.39 98.53 98.50 98.64
AA/% 98.17 98.13 98.24 98.32
Kappa 0.982 5 0.984 0 0.983 7 0.985 2

3.2 PCA降维后特征数和空间块大小设置

3.2.1 PCA降维后特征数讨论

本节讨论了各个数据集下不同PCA降维后特征数对分类精度的影响，其影响关系见图9。如图9所示，3个数据集的分类精度随降维后特征数量的增加均呈先上升后下降的趋势。在IP数据集中，当PCA降维获得特征数为80时，OA和AA最高，Kappa系数仅略微低于特征数为90的情况。考虑到降维特征数越高需要更多的模型计算时间，因此在IP数据集降维维度选择为80。而在PU数据集中，当降维后特征数取15时，精度最高。随着降维后特征数的提升，精度迅速下降，这是由于PU数据集影像质量较高、所含的噪声更少，只需少量的主成分分量就能充分表示地物的光谱特征。在质量同样较好的HO数据集中，当降维后特征数为20时也有最高的精度。对比可知，对于存在更多噪声和冗余波段的高光谱数据，需要保留更多的主成分才能得到较高的分类结果。
图9 3个数据集下不同PCA降维后特征数对分类精度影响图

Fig.9 Accuracy line plots of different PCA dimensionality reduction dimensions in 3 datasets

3.2.2 空间块大小选取设置

空谱联合的方法依赖于切分空间块包含的空间信息，空间块过小会导致包含的信息不足，而过大又会引入过多的冗余信息导致精度下降，且空间块大小又与高光谱图像的空间分辨率和地物的尺寸特征相关，如在高光谱高空间分辨率的数据集中通常需要更大的空间块大小才能完整的包含整个地物所需的特征。本节在3个数据集上对图像的空间块大小选取进行了讨论，表5展示了不同空间块大小对分类精度的影响。如表5所示，3个数据集中，精度随着空间块的变大都呈现先上升后下降的趋势。在IP数据集和HO数据集空间块大小为9时最佳，此时能够包含足够的空间信息供模型进行训练，而PU数据集具有较高的空间分辨率，因此所选取的空间块为13。
表5 不同大小空间块分类精度对比

Tab.5 Comparison of classification accuracy for different patch sizes

数据集 精度 7 9 11 13 15
IP OA/% 99.27 99.39 99.22 98.97 98.65
AA/% 98.60 99.06 98.85 95.83 91.74
Kappa 0.991 7 0.993 0 0.991 1 0.988 3 0.984 6
PU OA/% 99.19 99.42 99.58 99.67 99.64
AA/% 98.56 99.04 99.18 99.35 99.33
Kappa 0.989 3 0.992 3 0.994 4 0.995 6 0.995 2
HO OA/% 98.42 98.64 98.52 98.42 98.34
AA/% 97.98 98.32 97.95 97.78 97.68
Kappa 0.982 8 0.985 2 0.983 9 0.982 8 0.981 9

4 结论

本文提出了IR3NAN网络用于高光谱分类，网络主要由RSFEM，NAM和SSFFM 3个部分组成。其中，RSFEM通过3D-CNN和残差结构的构建有效提取浅层和深层光谱语义信息；NAM模块通过计算其他像素与中心像素之间的特征距离分配空间注意力权重，使得网络能加强对同质区域的关注程度；SSFFM通过一系列的3D卷积块和2D卷积块对前层网络的特征进行融合，获得表征性能更强的空间-光谱语义信息，从而提升网络的分类性能和鲁棒性。
1）所提出方法在3个公开的高光谱数据集Indian pines，Pavia University，Houston2013进行了验证，在分别使用10%，2%和5%的训练样本下，OA可分别达到99.39%，99.67%和98.64%。
2）为了证明本文方法的鲁棒性，实验额外选取了4组在不同训练样本进行了验证。实验表明，本文所提出方法综合性能优于其他现有模型，且具有更强的鲁棒性。
3）文章对提出的RSFEM和NAM模块进行了消融分析，实验表明所提出的模块能分别对高光谱图像的光谱特征和空间特征进行提取，二者结合使用能够有效提高分类精度。

参考文献

原文顺序 | 文献年度倒序 | 文中引用次数倒序
[1]
张涛, 王彬沣, 付莹, 等. 基于深度学习的光谱图像超分辨率综述[J]. 中国图象图形学报, 2024, 29(8):2113-2136.

Zhang T, Wang B F, Fu Y, et al. Deep learning-based spectral image super-resolution:A survey[J]. Journal of Image and Graphics, 2024, 29(8):2113-2136.

[2]
马晓瑞, 哈林, 谌敦斌, 等. 融合特征优化的跨数据集高光谱图像分类[J]. 中国图象图形学报, 2024, 29(8):2175-2187.

Ma X R, Ha L, Shen D B, et al. Cross-dataset hyperspectral image classification based on fusion feature optimization[J]. Journal of Image and Graphics, 2024, 29(8):2175-2187.

[3]
袁静文, 武辰, 杜博, 等. 高分五号高光谱遥感影像的城市土地利用景观格局分析[J]. 遥感学报, 2020, 24(4):465-478.

Yuan J W, Wu C, Du B, et al. Analysis of landscape pattern on urban land use based on GF-5 hyperspectral data[J]. Journal of Remote Sensing, 2020, 24(4):465-478.

[4]
王书伟, 席磊, 邱霜, 等. 高光谱遥感森林资源监测原理与应用[J]. 安徽农业科学, 2023, 51(15):111-114,197.

Wang S W, Xi L, Qiu S, et al. Principle and application of hyperspectral remote sensing forest resource monitoring[J]. Journal of Anhui Agricultural Sciences, 2023, 51(15):111-114,197.

[5]
杨燕杰. 基于高光谱的高海拔地区矿物信息提取与分析[J]. 世界核地质科学, 2023, 40(4):1020-1028.

Yang Y J. The alteration mineral information extraction and analysis of hyperspectral data in high-altitude area[J]. World Nuclear Geoscience, 2023, 40(4):1020-1028.

[6]
吕欢欢, 张峻通, 张辉. 多尺度特征与双注意力机制的高光谱影像分类[J]. 光电子·激光, 2024, 35(2):143-154.

Lyu H H, Zhang J T, Zhang H. Multi-scale feature and dual-attention mechanism for hyperspectral image classification[J]. Journal of Optoelectronics·Laser, 2024, 35(2):143-154.

[7]
Chen Y, Lin Z, Zhao X, et al. Deep learning-based classification of hyperspectral data[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2014, 7(6):2094-2107.

[8]
Liu Q, Zhou F, Hang R, et al. Bidirectional-convolutional LSTM based spectral-spatial feature learning for hyperspectral image classification[J]. Remote Sensing, 2017, 9(12):1330.

[9]
Mou L, Ghamisi P, Zhu X X. Deep recurrent neural networks for hyperspectral image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2017, 55(7):3639-3655.

[10]
He K, Zhang X, Ren S, et al. Deep residual learning for image recognition[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). IEEE, 2016:770-778.

[11]
Ren S, He K, Girshick R, et al. Faster R-CNN:Towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2017, 39(6):1137-1149.

[12]
Shelhamer E, Long J, Darrell T. Fully convolutional networks for semantic segmentation[C]// IEEE Transactions on Pattern Analysis and Machine Intelligence. IEEE,2017:640-651.

[13]
Hu W, Huang Y, Wei L, et al. Deep convolutional neural networks for hyperspectral image classification[J]. Journal of Sensors, 2015, 2015(1):258619.

[14]
Makantasis K, Karantzalos K, Doulamis A, et al. Deep supervised learning for hyperspectral data classification through convolutional neural networks[C]//2015 IEEE International Geoscience and Remote Sensing Symposium (IGARSS). IEEE, 2015:4959-4962.

[15]
Zhu M, Jiao L, Liu F, et al. Residual spectral-spatial attention network for hyperspectral image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2021, 59(1):449-462.

[16]
Kang X, Zhuo B, Duan P. Dual-path network-based hyperspectral image classification[J]. IEEE Geoscience and Remote Sensing Letters, 2019, 16(3):447-451.

DOI

[17]
Hamida A B, Benoit A, Lambert P, et al. 3-D deep learning approach for remote sensing image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2018, 56(8):4420-4434.

[18]
Zhong Z, Li J, Luo Z, et al. Spectral-spatial residual network for hyperspectral image classification:A 3-D deep learning framework[J]. IEEE Transactions on Geoscience and Remote Sensing, 2018, 56(2):847-858.

[19]
郑宗生, 王政翰, 王振华, 等. 改进3D-Octave卷积的高光谱图像分类方法[J]. 自然资源遥感, 2024, 36(4):82-91.doi:10.6046/zrzyyg.2023171.

Zheng Z S, Wang Z H, Wang Z H, et al. Improved 3D-Octave convolutional hyperspectral image classification method[J]. Remote Sensing for Natural Resources, 2024, 36(4):82-91.doi:10.6046/zrzyyg.2023171.

[20]
Roy S K, Krishna G, Dubey S R, et al. HybridSN:Exploring 3-D-2-D CNN feature hierarchy for hyperspectral image classification[J]. IEEE Geoscience and Remote Sensing Letters, 2020, 17(2):277-281.

[21]
Roy S K, Manna S, Song T, et al. Attention-based adaptive spectral-spatial kernel ResNet for hyperspectral image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2021, 59(9):7831-7843.

[22]
Dosovitskiy A, Beyer L, Kolesnikov A, et al. An image is worth 16x16 words:Transformers for image recognition at scale[J/OL]. arXiv, 2020. https://arxiv.org/abs/2010.11929.

[23]
Roy S K, Deria A, Shah C, et al. Spectral-spatial morphological attention Transformer for hyperspectral image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2023, 61:5503615.

[24]
Chen N, Fang L, Xia Y, et al. Spectral query spatial:Revisiting the role of center pixel in transformer for hyperspectral image classification[J]. IEEE Transactions on Geoscience and Remote Sensing, 2024, 62:5402714.

[25]
Ma C, Wan M, Wu J, et al. Light self-Gaussian-attention vision Transformer for hyperspectral image classification[J]. IEEE Transactions on Instrumentation and Measurement, 2023, 72:5015712.

[26]
D’Ascoli S, Touvron H, et al. ConViT:Improving vision Transfor-mers with soft convolutional inductive biases[C]//International conference on machine learning. PMLR,2021:2286-2296.

[27]
Gao H, Chen Z, Xu F. Adaptive spectral-spatial feature fusion network for hyperspectral image classification using limited training samples[J]. International Journal of Applied Earth Observation and Geoinformation, 2022, 107:102687.

[28]
Zhao Z, Hu D, Wang H, et al. Convolutional Transformer network for hyperspectral image classification[J]. IEEE Geoscience and Remote Sensing Letters, 2022, 19:6009005.

[29]
Debes C, Merentitis A, Heremans R, et al. Hyperspectral and LiDAR data fusion:Outcome of the 2013 GRSS data fusion contest[J]. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 2014, 7(6):2405-2418.

[30]
Zhang L, Zhang L, Du B. Deep learning for remote sensing data:A technical tutorial on the state of the art[J]. IEEE Geoscience and Remote Sensing Magazine, 2016, 4(2):22-40.

Options
文章导航

/

〈 〉