一种基于多任务学习卷积神经网络的图像隐写分析方法技术

本发明专利技术公开了一种基于多任务学习卷积神经网络的图像隐写分析方法：S1、设计包括图像隐写分析主网络和像素点隐写分析分支网络的多任务学习网络模型；S2、对原始图像进行隐写得到隐写图像，保存每张图像的图像隐写分析标签和像素点隐写分析标签；S3、使用高通滤波器对多任务学习网络模型中的第一层卷积层的权值参数进行初始化，其余卷积层使用Xavier方式进行初始化；S4、将步骤S2得到的数据集随机划分为训练集和测试集，应用Adam算法在训练集上对多任务学习网络模型同时进行图像隐写分析和像素点隐写分析有监督的学习训练，在测试集上进行测试，最后保存在测试集上结果最好的模型作为最终模型；S5、使用最终模型对其余新的测试图像进行隐写分析。

【技术实现步骤摘要】
一种基于多任务学习卷积神经网络的图像隐写分析方法
本专利技术涉及图像信号处理、图像隐写分析领域，具体涉及一种基于多任务学习卷积神经网络的图像隐写分析方法。
技术介绍
图像隐写是指在图像的空域或变换域上嵌入密信，当前比较安全的隐写算法能自适应地将密信嵌入到内容复杂的区域以使之更难以被检测出来。而图像隐写分析的主要任务则是判断一张图像是否有被嵌入密信，即是一个二分类的问题。传统的隐写分析算法主要包括特征的提取和分类器的设计这两大步骤。其中，所提特征是否有足够的区分能力直接决定最后分类器的性能。而特征的提取完全依赖于研究者对隐写图像与非隐写图像间差异的理解与认识这些先验知识。另外，特征提取与分类器的训练是依次进行而不是同时优化的。深度学习，尤其是卷积神经网络在图像识别、目标检测等计算机视觉问题上取得了突破性进展。不需要依赖于手工设计的特征，卷积神经网络能自动从大量的训练数据中学到表征能力和鲁棒性都更强的特征。另外，一般在应用卷积神经网络时，特征提取和分类器的训练是端到端地联合优化的，可使最终的网络模型得到更优的性能。在利用卷积神经网络进行图像隐写分析的研究中，中科大的Qian等人首先提出使用SRM中用到的一个高通滤波器提取图像的高频特征，然后将其输入卷积神经网络，得到接近SRM算法的性能。中山大学的Ye等人使用了SRM中用到的所有30个高通滤波器初始化神经网络的第一个卷积层，同时将图像中各个像素点是否被改变过的概率评估这种先验知识加入网络，得到优于maxSRMd2的结果。这些都是卷积神经网络在图像隐写分析领域非常有价值的尝试，取得了不错的结果。在识别、检测等传统计算机视觉问题中，通过合理的设置，用同一个共享网络同时训练几个目标相近的任务，有可能可以取得比单独训练一个任务更好的结果。这是因为一方面在训练过程中加入了更多的监督信号，有助于学到更好的特征表示，另一方面可以在一定程度上防止网络过拟合到某一个任务的训练数据上，起到类似正则化的作用，提升网络的泛化能力。因此希望同样能通过多任务的学习来提高卷积神经网络在图像隐写分析任务中的性能。
技术实现思路
本专利技术的目的是针对现有技术的不足，为了更好地提升卷积神经网络在图像隐写分析任务中的性能和泛化能力，提供了一种基于多任务学习卷积神经网络的图像隐写分析方法，能够在网络训练过程中同时对图像隐写分析和像素点隐写分析这两项任务进行优化学习。本专利技术的目的可以通过如下技术方案实现：一种基于多任务学习卷积神经网络的图像隐写分析方法，所述方法包括以下步骤：S1、多任务学习网络模型设计：包括图像隐写分析主网络的设计和像素点隐写分析分支网络的设计；S2、数据集的制作：使用隐写算法在设定嵌入率下对原始图像进行隐写得到对应的隐写图像，同时保存每张图像是否有被隐写的图像隐写分析标签和每张图像中每个像素点是否有被改变的像素点隐写分析标签；S3、多任务学习网络模型初始化：使用高通滤波器对多任务学习网络模型中的第一层卷积层的权值参数进行初始化以提取图像的高频信息，其余卷积层使用Xavier方式进行初始化；S4、多任务学习网络模型的训练：将步骤S2中得到的数据集按设定比例随机划分为训练集和测试集，应用Adam算法在训练集上对多任务学习网络模型同时进行图像隐写分析和像素点隐写分析有监督的学习训练，在测试集上进行测试，最后保存在测试集上结果最好的模型作为最终模型；S5、使用在步骤S4中训练得到的在测试集上结果最好的多任务学习网络模型对其余新的测试图像进行隐写分析。进一步地，步骤S1中所述图像隐写分析主网络的第一层为由30个5×5×1的卷积核组成的卷积层，用于提取图像的高频信息，其后依次连接有五个结构相同的网络小块，其中每个网络小块均由两个30×(3×3×30)的卷积层和一个均值池化层依次连接而成，第五个网络小块后连接一个包含有两个神经元的全连接层，作为预测图像是否有被隐写的两个输出。进一步地，所述像素点隐写分析分支网络包括第一个网络小块后依次连接的一个输出通道数为2大小为2×(1×1×30)的卷积层和一个将特征图上采样2倍至输入图像大小的解卷积层，以及第二个网络小块后依次连接的一个输出通道数为2大小为2×(1×1×30)的卷积层和一个将特征图上采样4倍至输入图像大小的解卷积层；对上述得到的两个输入图像大小的特征图求和，得到的特征图作为对图像中各个像素点是否有被隐写的输出。进一步地，步骤S2中，原始图像的图像隐写分析标签为0，隐写图像的图像隐写分析标签为1；对于隐写图像，将其与对应的原始图像中的每个像素点逐个进行比对，若相同，则将其像素点隐写标签对应位置置为0，否则置为1；对于原始图像，其各个位置像素点隐写标签均置为2，在后续训练时忽略。进一步地，步骤S3中，所述高通滤波器为隐写算法SRM中使用到的30个高通滤波器。本专利技术与现有技术相比，具有如下优点和有益效果：1、本专利技术通过多任务学习的方法，训练卷积神经网络同时优化图像隐写分析和像素点隐写分析这两个任务，从而使得网络在训练过程中得到了更多的监督信号，训练过程收敛得更快。2、本专利技术采用多任务学习的技术方案，训练卷积神经网络同时优化图像隐写分析和像素点隐写分析这两个任务，从而使得网络更不容易陷入过拟合，泛化能力更强，得到更低的错误率。附图说明图1为本专利技术实施例多任务学习网络模型中结构相同的网络小块。图2为本专利技术实施例多任务学习卷积神经网络图像隐写分析方法训练阶段的多任务学习网络模型。图3(a)为本专利技术实施例原始图像的像素点值示例图、图3(b)为本专利技术实施例原始图像的像素点隐写分析标签示例图、图3(c)为本专利技术实施例隐写图像的像素点值示例图、图3(d)为本专利技术实施例隐写图像的像素点隐写分析标签示例图。图4为本专利技术实施例对多任务学习网络模型中的第一层卷积层的权值参数进行初始化的30个高通滤波器。具体实施方式下面结合实施例及附图对本专利技术作进一步详细的描述，但本专利技术的实施方式不限于此。实施例：本实施例提供了一种基于多任务学习卷积神经网络的图像隐写分析方法，所述方法包括以下步骤：S1、多任务学习网络模型设计，具体的网络参数设置和方法如下：如图2所示，本实施例的多任务学习网络模型包括图像隐写分析主网络和像素点隐写分析分支网络，其中，图像隐写分析主网络的第一层为由30个5×5×1的卷积核组成的卷积层，用于提取图像的高频信息，其后依次连接有五个结构相同的网络小块，如图1所示，其中每个网络小块均由两个30×(3×3×30)的卷积层和一个大小为3×3、步长为2的均值池化层依次连接而成，第五个网络小块后连接一个包含有两个神经元的全连接层，作为预测图像是否有被隐写的两个输出。作为像素点隐写分析分支网络的其中一部分，第一个网络小块后再接一个大小为2×(1×1×30)的卷积层，该卷积层后紧接一个大小为2×(4×4×2)，步长为2的转置卷积层将特征图上采样至输入图像大小；类似地，第2个网络小块后依次接大小2×(1×1×30)的卷积层和大小为2×(8×8×2)，步长为4的转置卷积层，同样得到长宽与输入图像一致的特征图。对上述两张特征图求和，得到长宽与输入图像相等的两张特征图，这两张特征图的各个位置分别对应输入图像各个像素点是否有被改变的类别。除第一个卷积层外，网络中的所有卷积层后都本文档来自技高网...

【技术保护点】
1.一种基于多任务学习卷积神经网络的图像隐写分析方法，其特征在于，所述方法包括以下步骤：S1、多任务学习网络模型设计：包括图像隐写分析主网络的设计和像素点隐写分析分支网络的设计；S2、数据集的制作：使用隐写算法在设定嵌入率下对原始图像进行隐写得到对应的隐写图像，同时保存每张图像是否有被隐写的图像隐写分析标签和每张图像中每个像素点是否有被改变的像素点隐写分析标签；S3、多任务学习网络模型初始化：使用高通滤波器对多任务学习网络模型中的第一层卷积层的权值参数进行初始化以提取图像的高频信息，其余卷积层使用Xavier方式进行初始化；S4、多任务学习网络模型的训练：将步骤S2中得到的数据集按设定比例随机划分为训练集和测试集，应用Adam算法在训练集上对多任务学习网络模型同时进行图像隐写分析和像素点隐写分析有监督的学习训练，在测试集上进行测试，最后保存在测试集上结果最好的模型作为最终模型；S5、使用在步骤S4中训练得到的在测试集上结果最好的多任务学习网络模型对其余新的测试图像进行隐写分析。

【技术特征摘要】
1.一种基于多任务学习卷积神经网络的图像隐写分析方法，其特征在于，所述方法包括以下步骤：S1、多任务学习网络模型设计：包括图像隐写分析主网络的设计和像素点隐写分析分支网络的设计；S2、数据集的制作：使用隐写算法在设定嵌入率下对原始图像进行隐写得到对应的隐写图像，同时保存每张图像是否有被隐写的图像隐写分析标签和每张图像中每个像素点是否有被改变的像素点隐写分析标签；S3、多任务学习网络模型初始化：使用高通滤波器对多任务学习网络模型中的第一层卷积层的权值参数进行初始化以提取图像的高频信息，其余卷积层使用Xavier方式进行初始化；S4、多任务学习网络模型的训练：将步骤S2中得到的数据集按设定比例随机划分为训练集和测试集，应用Adam算法在训练集上对多任务学习网络模型同时进行图像隐写分析和像素点隐写分析有监督的学习训练，在测试集上进行测试，最后保存在测试集上结果最好的模型作为最终模型；S5、使用在步骤S4中训练得到的在测试集上结果最好的多任务学习网络模型对其余新的测试图像进行隐写分析。2.根据权利要求1所述的一种基于多任务学习卷积神经网络的图像隐写分析方法，其特征在于：步骤S1中所述图像隐写分析主网络的第一层为由30个5×5×1的卷积核组成的卷积层，用于提取图像的高频信息，其后依次连接有五个结构相同的网络小块，其中每个网络...

【专利技术属性】
技术研发人员：余翔宇，谭华彬，梁辉，
申请(专利权)人：华南理工大学，
类型：发明
国别省市：广东,44