一种基于张量分解网络的高光谱异常检测方法技术

本发明专利技术公开了一种基于张量分解网络的高光谱异常检测方法，步骤包括：步骤1、选取背景样本数据集；步骤2、通过自编码器和张量分解网络来恢复背景数据信息，先将训练集送入编码器中进行编码，通过三层编码网络逐步将输入数据降维，得到潜在特征；通过张量分解网络从潜在特征中分解出背景的低秩张量，通过解码器的解码操作，逐层恢复出背景数据；步骤3、通过重构误差进行异常检测，得到最终的检测结果图。本发明专利技术的方法，通过深度网络来模拟传统张量分解的过程，有效提高了网络的重建精度，进一步提高异常检测性能。高异常检测性能。高异常检测性能。

【技术实现步骤摘要】
一种基于张量分解网络的高光谱异常检测方法


[0001]本专利技术属于高光谱影像处理
，涉及一种基于张量分解网络的高光谱异常检测方法。

技术介绍

[0002]高光谱异常目标检测作为高光谱影像处理的一个重要子方向，在过去的数十年里，有大量算法被提出。RX(Reed
‑
Xiao li Detection,RX)算法作为高光谱异常目标检测的一个基准算法，它假设背景服从多元正态分布，通过计算待检测像素与背景之间的马氏距离来确定每一个像素的异常程度，然而由于实际场景复杂，难以满足此假设，所以RX算法检测精度有限。为了避免对背景分布的假设，李伟和杜倩提出了基于协同表达思想的CRD(Collaborative Representation
‑
based Detector,CRD)算法，此算法通过滑动双窗口构造局部字典，并通过此字典来逐个表示原始输入高光谱影像，最后表示误差作为每一个像素的异常程度。然而由于缺少先验信息，针对不同的数据集CRD算法没有一个有效的规则来制定双窗口尺寸。得益于低秩稀疏矩阵分解(Low
‑
Rank and Sparse Matrix Decomposition,LRaSM)技术的发展，大量的基于LRaSM的异常检测算法被相继提出。LRASR算法通过K
‑
Means聚类算法构造背景字典，然后通过低秩矩阵分解来进行异常检测。除此之外，张兴等提出了基于Tucker分解的异常检测算法，此方法很好地保留了原始影像的空谱信息，然后由于高光谱影像自身的高维性以及噪声多等特点，导致基于矩阵分解和张量分解的异常检测算法不够稳定。
[0003]近年来，深度学习技术为图像处理、计算机视觉等领域带来了极大的便利，基于此，为了克服上述方法的不足，亟需研制一种基于张量分解网络的高光谱异常目标检测方法，以客服了噪声和维度对异常检测算法的影响。

技术实现思路

[0004]本专利技术的目的是提供一种基于张量分解网络的高光谱异常检测方法，解决现有基于传统张量分解模型过程复杂、易受噪声和数据维度干扰的问题。
[0005]本专利技术采用的技术方案是，一种基于张量分解网络的高光谱异常检测方法，按照以下步骤具体实施：
[0006]步骤1、选取背景样本数据集，
[0007]输入的原始高光谱影像记为上角标W、H和D分别表示原始高光谱影像的宽、高和波段数；
[0008]步骤2、通过自编码器和张量分解网络来恢复背景数据信息，
[0009]先将训练集送入编码器中进行编码，通过三层编码网络逐步将输入数据降维：D
→
128
→
64
→
24，得到潜在特征
[0010]通过张量分解网络从潜在特征中分解出背景的低秩张量
张量分解网络包括RTGB模块和LTGB模块，RTGB模块根据输入生成对应的秩为一的张量，而LTGB模块采用残差结构和RTGB模块生成R个秩为一的张量，然后将R个秩为一的张量进行通道拼接，再使用1
×
1卷积降维得到背景低秩张量具体过程是：
[0011]对于输入的潜在特征先分别从通道方向、宽度方向以及高度方向取全局平均池化，并将得到的三个方向向量先经过1
×
1卷积，再进行Sigmoid激活，接着通过克罗内克积生成秩为一的张量，具体表达式如下：
[0012][0013]其中，O
i
表示第i个秩为一的张量，分别表示通道、宽度以及高度方向的平均池化向量，表示克罗内克积，
[0014]将上一个RTGB模块的输入与输出的差作为下一个RTGB模块的输入，生成下一个秩为一的张量，重复此操作，直到生成R个秩为一的张量，再将R个秩为一的张量进行通道拼接，经过卷积后得到背景低秩张量表达式如下：
[0015]L
rank
＝Conv(tf.concat(O1,O2,...,O
R
))
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(3)
[0016]其中，tf.concat(
·
)表示通道拼接操作，在得到低秩张量后，通过解码器的解码操作，逐层恢复出背景数据
[0017]步骤3、通过重构误差进行异常检测，得到最终的检测结果图。
[0018]本专利技术的有益效果是：1)通过深度网络来模拟传统张量分解(CP分解)的过程，从而克服了传统张量分解模型复杂、难以找到最优解这一缺陷。2)在原有均方误差的基础上引入了结构相似性损失，克服了图像重建过程中的模糊问题，从而有效的提高了网络的重建精度，进一步提高异常检测性能。
附图说明
[0019]图1是本专利技术方法采用的算法流程简图；
[0020]图2是本专利技术方法采用的背景样本选取模块简图；
[0021]图3是本专利技术方法采用的张量生成网络简图；
[0022]图4a是Texas Coast数据集的伪彩色图，图4b是Texas Coast数据集标签，图4c是RX方法的处理结果图，图4d是LRX方法的处理结果图，图4e是CRD方法的处理结果图，图4f是LRASR方法的处理结果图，图4g是DAEAD方法的处理结果图，图4h是本专利技术方法的处理结果图；
[0023]图5是采用现有异常检测算法RX、LRX、CRD、LRASR、DAEAD以及本专利技术方法对高光谱异常目标进行检测的ROC曲线对比图；
[0024]图6是采用现有异常检测算法RX、LRX、CRD、LRASR、DAEAD以及本专利技术方法对高光谱异常目标进行检测的AUC值对比图。
具体实施方式
[0025]下面结合附图和具体实施方式对本专利技术进行详细说明。
[0026]本专利技术方法的整体思路是：首先，使用马氏距离来计算输入高光谱图像上每一个像素点的异常响应值，根据预先设置的背景样本选取比率来剔除响应值最大的可疑异常像素；然后，将选取的背景样本训练集输入到编码器编码，将编码特征送入到张量分解网络得到低秩张量，再通过解码器解码得到恢复的背景数据；最后在残差图像上使用均方误差进行异常检测。
[0027]本专利技术的方法，按照以下步骤具体实施：
[0028]步骤1、选取背景样本数据集，
[0029]输入的原始高光谱影像记为上角标W、H和D分别表示原始高光谱影像的宽、高和波段数，
[0030]参照图2，首先通过马氏距离来计算原始高光谱影像X上每一个像素点的异常响应值，它的大小为假设预先设置的背景样本的选取比率为将异常响应值最小的个像素置为1，其余像素标记为0，其中WH表示总像素数目，从而得到一个权值矩阵其中，1表示要选取的背景样本，0表示剔除掉的潜在异常像素，背景样本训练集的表达式如下：
[0031]X
train
＝X
⊙
W
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(1)
[0032]其中，
⊙
表示像素点乘本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于张量分解网络的高光谱异常检测方法，其特征在于，按照以下步骤具体实施：步骤1、选取背景样本数据集，输入的原始高光谱影像记为上角标W、H和D分别表示原始高光谱影像的宽、高和波段数；步骤2、通过自编码器和张量分解网络来恢复背景数据信息，先将训练集送入编码器中进行编码，通过三层编码网络逐步将输入数据降维：D
→
128
→
64
→
24，得到潜在特征通过张量分解网络从潜在特征中分解出背景的低秩张量张量分解网络包括RTGB模块和LTGB模块，RTGB模块根据输入生成对应的秩为一的张量，而LTGB模块采用残差结构和RTGB模块生成R个秩为一的张量，然后将R个秩为一的张量进行通道拼接，再使用1
×
1卷积降维得到背景低秩张量具体过程是：对于输入的潜在特征先分别从通道方向、宽度方向以及高度方向取全局平均池化，并将得到的三个方向向量先经过1
×
1卷积，再进行Sigmoid激活，接着通过克罗内克积生成秩为一的张量，具体表达式如下：其中，O
i
表示第i个秩为一的张量，分别表示通道、宽度以及高度方向的平均池化向量，表示克罗内克积，将上一个RTGB模块的输入与输出的差作为下一个RTGB模块的输入，生成下一个秩为一的张量，重复此操作，直到生成R个秩为一的张量，再将R个秩为一的张量进行通道拼接，经过卷积后得到背景低秩张量表达式如下：L
rank
＝Conv(tf.concat(O1,O2,...,O
R
))
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(3)其中，tf.concat(
·
)表示通道拼接操作，在得到低秩张量后，通过解码器...

【专利技术属性】
技术研发人员：孙帮勇，赵哲，
申请(专利权)人：西安理工大学，
类型：发明
国别省市：