一种用于图像特征提取的融减自动编码器算法的制作方法

[0001]
本发明涉及图像特征提取技术领域，具体为一种用于图像特征提取的融减自动编码器算法。


背景技术：

[0002]
目前现存的各种自动编码器在特征提取部分主要是利用逐层卷积和池化进行特征提取和降维，在解码恢复方面本质上都是根据各自的任务背景对损失函数其实施一定的松弛或限制以提高算法性能。但由于自动编码器算法是基于无监督学习原理，上述方法容易使得模型在训练过程中更加难以收敛，并且在网络初始化初期就会出现梯度爆炸或梯度消失的情况；同时，在图像特征提取过程中仍存在特征提取过程中特征缺乏融合、信息损失过多以及模型容易早熟等问题。
[0003]
基于此，本发明设计了一种用于图像特征提取的融减自动编码器算法，以解决上述提到的问题。


技术实现要素：

[0004]
本发明的目的在于提供一种用于图像特征提取的融减自动编码器算法，在自动编码器的编码阶段建立特征层之间的联系，在解码阶段提升了解码效率和质量；设计的重构损失函数能够用于特征层之间的优化和避免模型早熟。
[0005]
为实现上述目的，本发明提供如下技术方案：一种用于图像特征提取的融减自动编码器算法，包括建立融减网络结构模型和联合重构损失函数模型，
[0006]
所述建立融减网络结构模型包括以下步骤：
[0007]
s1：将输入信号增加随机噪声形成受损信号；
[0008]
s2：利用4组卷积和池化对受损信号进行特征提取及降维；
[0009]
s3：在特征提取过程中在第4特征层对第2特征层进行融合、在第5特征层对第3特征层进行融合；
[0010]
所述建立联合重构损失函数模型包括以下步骤：
[0011]
s4：在相互对称的编码第3特征层和解码第1特征层之间计算均方误差损失；并与恢复信号和输入信号的均方误差损失形成联合重构损失，并将其一起进行优化。
[0012]
与现有技术相比，本发明的有益效果是：本发明通过融减自动编码器中的融减网络结构能够建立特征层之间的联系和提升解码效率，同时该算法的联合重构损失函数也能够加强编码和解码阶段对应层之间的联系和避免模型早熟。
附图说明
[0013]
为了更清楚地说明本发明实施例的技术方案，下面将对实施例描述所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本发明的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附
图。
[0014]
图1为本发明融减自动编码器网络结构框架图；
[0015]
图2为本发明融减自动编码器网络参数表图。
具体实施方式
[0016]
下面将结合本发明实施例中的附图，对本发明实施例中的技术方案进行清楚、完整地描述，显然，所描述的实施例仅仅是本发明一部分实施例，而不是全部的实施例。基于本发明中的实施例，本领域普通技术人员在没有作出创造性劳动前提下所获得的所有其它实施例，都属于本发明保护的范围。
[0017]
请参阅图1-2，本发明提供一种技术方案：一种用于图像特征提取的融减自动编码器算法，包括建立融减网络结构模型和联合重构损失函数模型，
[0018]
所述建立融减网络结构模型包括以下步骤：
[0019]
s1：将输入信号增加随机噪声形成受损信号；
[0020]
s2：利用4组卷积和池化对受损信号进行特征提取及降维；
[0021]
s3：在特征提取过程中在第4特征层对第2特征层进行融合、在第5特征层对第3特征层进行融合；
[0022]
所述建立联合重构损失函数模型包括以下步骤：
[0023]
s4：在相互对称的编码第3特征层和解码第1特征层之间计算均方误差损失；并与恢复信号和输入信号的均方误差损失形成联合重构损失，并将其一起进行优化。
[0024]
为测试融减网络结构及联合重构损失函数的有效性，首先，我们分别使用融减网络结构对卷积降噪自编码器进行改进以及使用联合重构损失函数优化对卷积降噪自编码器进行优化；其次，利用上述改进后的两个模型与原始卷积降噪自编码器肺部ct数据集进行特征提取；最后，对比所提取特征在svm、k-means和cart分类器上的分类准确率，以此验证融减网络结构及联合重构损失函数的有效性。实验结果表明与卷积降噪自编码器相比，利用融减网络结构改进后的卷积降噪自编码器和使用联合重构损失函数优化的卷积降噪自编码器所提取特征的肺炎筛查准确率在k-means分类器上平均提高了6.45％，在svm分类器上平均提高了6.95％，在cart分类器上平均提高了2.95％。
[0025]
为验证融减自动编码器的特征提取性能，实验中分别使用传统特征提取算法中的分层梯度方向直方图和融减自动编码器在肺部ct数据集进行特征提取，实验结果显示与分层梯度方向直方图和卷积降噪自编码器相比，使用融减自动编码器所提取特征的肺炎筛查准确率在k-means分类器上平均提高 19.78％，在svm分类器上平均提高11.86％，在cart分类器上平均提高 11.68％，这表明融减自动编码器在图像特征提取方面的优异性能；其次，基于融减自动编码器所提取的特征在不同分类器上的分类准确率均大于97％，说明模型所提取特征对于肺部异常图像的鲁棒性；最后，融减自动编码器在验证集和测试集上所提取特征的分类准确率近乎相同，这表明模型有较强抗拟合能力。
[0026]
在本说明书的描述中，参考术语“一个实施例”、“示例”、“具体示例”等的描述意指结合该实施例或示例描述的具体特征、结构、材料或者特点包含于本发明的至少一个实施例或示例中。在本说明书中，对上述术语的示意性表述不一定指的是相同的实施例或示例。而且，描述的具体特征、结构、材料或者特点可以在任何的一个或多个实施例或示例中以合
适的方式结合。
[0027]
以上公开的本发明优选实施例只是用于帮助阐述本发明。优选实施例并没有详尽叙述所有的细节，也不限制该发明仅为所述的具体实施方式。显然，根据本说明书的内容，可作很多的修改和变化。本说明书选取并具体描述这些实施例，是为了更好地解释本发明的原理和实际应用，从而使所属技术领域技术人员能很好地理解和利用本发明。本发明仅受权利要求书及其全部范围和等效物的限制。