非平衡高光谱图像分类方法.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114882368A(43)申请公布日2022.08.09(21)申请号202210626027.6(22)申请日2022.06.02(71)申请人西安电子科技大学地址710071陕西省西安市太白南路2号(72)发明人席博博李娇娇刁妍李云松刘薇宋锐刘松林(74)专利代理机构陕西电子工业专利中心61205专利代理师王品华(51)Int.Cl.G06V20/10(2022.01)G06V10/764(2022.01)G06V10/82(2022.01)G06N3/04(2006.01)G06N3/08(2006.01)权利要求书3页说明书8页附图3页(54)发明名称非平衡高光谱图像分类方法(57)摘要本发明公开了一种基于深度生成光谱‑空间分类器的非平衡高光谱图像分类方法，主要解决现有技术在面对非平衡高光谱分类任务中小类别分类精度不佳的问题。其实现方案为：获取高光谱图像并进行波段选择，将其划分为训练样本与测试样本；构建包含两阶段三维编码器、三维解码器、小类别上采样模块和分类器的非平衡高光谱图像的分类网络；利用训练样本对分类网络进行训练，设置总体损失函数，初始化网络训练参数，采用梯度下降方法更新分类网络，直至达到最大迭代次数；将测试样本输入到训练好的分类网络，得到分类结果。本发明提高了高光谱图像中小类别的分类精度，增强了鲁棒性，可用于矿产勘测，生态监测，智慧农业及医疗诊断。CN114882368ACN114882368A权利要求书1/3页1.一种非平衡高光谱图像的分类方法，其特征在于，包括以下步骤：1)获取高光谱图像数据并进行波段选择，按照固定尺寸将波段选择后的高光谱图像分割出光谱‑空间特征邻域块，再将其按1：99的比例划分为训练样本和测试样本；2)构建非平衡高光谱图像分类网络模型：2a)建立由三个三维卷积模块与二维卷积模块级联，并与两个并联的全连接网络级联，之后再级联重参数模块组成了三维编码器；2b)在隐含空间设置小类别上采样模块；2c)建立由1个全连接层和多个反卷积层构成的三维解码器；2d)建立由1个全连接层构成的图像分类器模块；2e)将三维编码器与小类别上采样模块级联，将三维解码器与分类器并联，小类别上采样模块的输出同时并行输入三维解码器与分类器，形成非平衡高光谱图像分类网络模型；3)对非平衡高光谱图像分类网络模型进行训练：3a)构建该网络模型的总体损失函数LTotal＝λ·Lmmd+LPdRec+Lcls，其中Lmmd为基于最大化均值差异的正则项损失函数，LPdRec为基于邻域距离的重构损失函数，Lcls为基于交叉熵的分类损失函数，λ为参数；3b)初始化非平衡高光谱图像分类网络模型的参数包括三维编码器中的权重参数w、偏置参数b和参数θ＝{θ1,θ2}、三维解码器中的权重参数w、偏置参数b和参数小类样本特征上采样模块中的均值参数和方差参数及分类器参数η；3c)将训练样本输入到非平衡高光谱图像分类网络模型，采用梯度下降方法，循环更新非平衡高光谱图像分类网络中的参数，以减小总损失函数的梯度值，直至达到最大迭代次数T为止，得到训练好的非平衡高光谱图像分类网络模型；4)将测试样本输入到训练好的非平衡高光谱图像分类网络模型，输出高光谱图像分类结果图。2.根据权利要求1所述的方法，其特征在于：所述1)中获取高光谱图像数据并进行波段选择，是利用主成分分析法对高光谱图像进行波段选择处理，实现特征降维，并保留前K个主成分；所述1)中按照固定尺寸将波段选择后的高光谱图像分割出光谱‑空间特征邻域块，是采用主成分波段选取样本，每个样本的尺寸为S×S×K，得到光谱‑空间邻域特征，其中S表示邻域的空间大小；所述1)中的训练图像表示为X＝{x1,...xi...,xN}，训练图像对应的标签为Y＝{y1,...yi...,yN}，其中xi为第i个训练图像，yi为第i个标签，i的范围为从1到N，N表示训练图像的数目。3.根据权利要求1所述的方法，其特征在于：所述2a)中三个三维卷积模块，均由卷积层、标准化层和激活层级联组成，该三个卷积层的卷积核分别为(3，3，7)、(3，3，5)和(3，3，3)，该三个标准化层与三个激活层结构相同，其结构为：第一卷积层‑>第一标准化层‑>第一激活层‑>第二卷积层‑>第二标准化层‑>第二激活层‑>第三卷积层‑>第三标准化层‑>第三激活层。4.根据权利要求1所述的方法，其特征在于：所述2a)中二维卷积模块由卷积核为(3，3)2CN114882368A权利要求书2/3页的卷积层和标准化层、激活层级联组成，其结构为：卷积层‑>标准化层‑>激活层。5.根据权利要求1所述的方法，其特征在于：所述2a)重参数模块，是从高斯分布中采样出变量ε，再结合全连接层的输出计算出输入图像的潜