[发明专利]一种基于多任务学习卷积神经网络的图像隐写分析方法

说明书

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

技术领域

本发明涉及图像信号处理、图像隐写分析领域，具体涉及一种基于多任务学习卷积神经网络的图像隐写分析方法。

背景技术

图像隐写是指在图像的空域或变换域上嵌入密信，当前比较安全的隐写算法能自适应地将密信嵌入到内容复杂的区域以使之更难以被检测出来。而图像隐写分析的主要任务则是判断一张图像是否有被嵌入密信，即是一个二分类的问题。

传统的隐写分析算法主要包括特征的提取和分类器的设计这两大步骤。其中，所提特征是否有足够的区分能力直接决定最后分类器的性能。而特征的提取完全依赖于研究者对隐写图像与非隐写图像间差异的理解与认识这些先验知识。另外，特征提取与分类器的训练是依次进行而不是同时优化的。

深度学习，尤其是卷积神经网络在图像识别、目标检测等计算机视觉问题上取得了突破性进展。不需要依赖于手工设计的特征，卷积神经网络能自动从大量的训练数据中学到表征能力和鲁棒性都更强的特征。另外，一般在应用卷积神经网络时，特征提取和分类器的训练是端到端地联合优化的，可使最终的网络模型得到更优的性能。在利用卷积神经网络进行图像隐写分析的研究中，中科大的Qian等人首先提出使用SRM中用到的一个高通滤波器提取图像的高频特征，然后将其输入卷积神经网络，得到接近SRM算法的性能。中山大学的Ye等人使用了SRM中用到的所有30个高通滤波器初始化神经网络的第一个卷积层，同时将图像中各个像素点是否被改变过的概率评估这种先验知识加入网络，得到优于maxSRMd2的结果。这些都是卷积神经网络在图像隐写分析领域非常有价值的尝试，取得了不错的结果。

在识别、检测等传统计算机视觉问题中，通过合理的设置，用同一个共享网络同时训练几个目标相近的任务，有可能可以取得比单独训练一个任务更好的结果。这是因为一方面在训练过程中加入了更多的监督信号，有助于学到更好的特征表示，另一方面可以在一定程度上防止网络过拟合到某一个任务的训练数据上，起到类似正则化的作用，提升网络的泛化能力。因此希望同样能通过多任务的学习来提高卷积神经网络在图像隐写分析任务中的性能。

发明内容

本发明的目的是针对现有技术的不足，为了更好地提升卷积神经网络在图像隐写分析任务中的性能和泛化能力，提供了一种基于多任务学习卷积神经网络的图像隐写分析方法，能够在网络训练过程中同时对图像隐写分析和像素点隐写分析这两项任务进行优化学习。

本发明的目的可以通过如下技术方案实现：

一种基于多任务学习卷积神经网络的图像隐写分析方法，所述方法包括以下步骤：

S1、多任务学习网络模型设计：包括图像隐写分析主网络的设计和像素点隐写分析分支网络的设计；

S2、数据集的制作：使用隐写算法在设定嵌入率下对原始图像进行隐写得到对应的隐写图像，同时保存每张图像是否有被隐写的图像隐写分析标签和每张图像中每个像素点是否有被改变的像素点隐写分析标签；

S3、多任务学习网络模型初始化：使用高通滤波器对多任务学习网络模型中的第一层卷积层的权值参数进行初始化以提取图像的高频信息，其余卷积层使用Xavier方式进行初始化；