1.一种基于图像预处理的对抗样本防御方法,其特征在于,包括如下步骤:
步骤1:构建防御模型AGD;所述防御模型AGD共有17层网络,由三大模块构成,其中,前
12层网络为扩张卷积模块,中间四层网络为特征增强模块,最后一层网络为注意力模块;
所述步骤1在扩张卷积模块中,通过将第2层、第5层、第9层以及第12层的卷积层由普通卷积改为扩张卷积,达到增大感受野来映射图像中更多的上下文信息,从而有助于获取到*更多的对抗扰动特征,公式(1)对扩张卷积模块的过程进行了表示,其中x代表输入的对抗样本,fDC代表扩张卷积模块实现的作用,ODC代表扩张卷积模块的输出,*ODC=fDC(x) (1)
所述步骤1在特征增强模块中,将输入的对抗样本特征与第16层网络所输出的特征进*
行融合,公式(2)对特征增强模块的过程进行了表示,其中x 代表输入的对抗样本,ODC代表扩张卷积模块的输出,fFE代表特征增强模块实现的作用,OFE代表扩张卷积模块的输出,*OFE=fFE(x ,ODC) (2)
所述步骤1在最后一层网络上加入了CBAM注意力模块来进行注意力机制的实现;CBAM注意力机制主要包括两部分,分别为通道注意力和空间注意力,注意力模块利用通道与空间这两个维度,对AGD防御模型中的中间特征图分别采用两个独立的注意力机制来得到权重系数,公式(3)表示了注意力模块的处理过程,其中OFE代表特征增强模块的输出,也是注意力模块的输入,fAB代表注意力模块实现的作用, 代表注意力模块的输出,即对抗扰动信息的残差图像,所述步骤1在AGD防御模型损失函数的设计中,将扰动去除前后图像在目标模型高层网络中输出结果之间的差异作为损失函数,如公式(4)所示,其中,代表经AGD处理去除了对抗扰动的图像,x代表原始图像,Fl代表目标模型中第l层网络的输出结果,||·||代表L1范数,步骤2:构建图像重建模型;基于超分辨率设计图像重建模型,对AGD防御模型处理后的样本进行重建,减轻AGD对正常样本产生的影响;
所述步骤2的图像重建模型由三个模块组成,分别是特征提取模块、多尺度特征融合模块以及亚像素重建模块;
所述步骤2在特征提取模块,对AGD处理后的图像进行初始特征的提取,作为后续模块的输入,利用了一个1×1的卷积提取AGD防御模型处理后图像的初始特征,该过程用公式(5)表示,其中,代表输入的低分辨率图像,即AGD防御模型处理后得到的图像,FE代表卷积核大小为1×1的卷积层对 特征的提取过程,OE代表提取初始特征后得到的特征图;
所述步骤2在多尺度特征融合模块,采用多个大小不同的卷积核对初始特征进行卷积操作,以提取不同维度的特征信息,并利用特征级联实现对多尺度特征的融合,提高模型获取到的图像特征的丰富度,多尺度特征融合模块增加了网络的宽度,能够提取到不同维度的特征信息,并使这些特征信息得到充分利用,该过程可用公式(6)表示,OM=FM(OE), (6)其中,OE代表特征提取模块提取的初始特征,FM代表多尺度特征融合模块的作用,OM代表经过多尺度特征融合后得到的新特征信息,在多尺度特征融合模块,包含三种不同尺度的卷积核,卷积核的大小分别为3×3、5×5和7×7,使用不同尺寸的卷积核,获取输入图像不同区域大小的特征信息,来提高特征的丰富度,同时,利用残差融合的方式,避免特征信息在网络传递中的丢失,保证每层网络提取到的特征信息都能被充分利用,此外,将不同尺度卷积核所提取的多个特征进行级联融合后,采用一个大小为1×1的卷积进行降维处理;
所述步骤2的亚像素重建模块,在亚像素重建前,将多尺度特征融合模块所得到的高级特征与低分辨率图像中的低级特征进行残差融合,之后,亚像素重建模块对融合后的特征进行亚像素卷积操作,重建高分辨率图像,该模块可用公式(7)表示,其中x′代表经图像重建模型重建后的图像,x′=FS(OE,OM), (7)
在图像重建模型的训练过程中,采用像素级的均方误差MSE来作为损失函数,将AGD防御模型处理后的样本输入到图像重建模型中,输出的重建后图像需与原始图像尽可能相似,因此,图像重建模型的训练目标就是最小化重建后样本与原始图像之间的差异,损失函数可用公式(8)表示,其中, 代表参数集, 代表图像重建模型的输出,||·||代表L2范数,
在训练过程中,使用随机梯度下降法来优化损失函数;
步骤3:构建防御框架SR‑AGD;结合AGD防御模型和图像重建模型的特点,构建防御框架SR‑AGD,实现对对抗攻击的防御效果。
2.根据权利要求1所述的基于图像预处理的对抗样本防御方法,其特征在于,所述步骤
3将对抗样本直接输入到SR‑AGD防御框架中,首先利用AGD防御模型去除对抗样本中的对抗扰动,得到处理后的样本,然后将其送入图像重建模型进行处理,在图像重建模型的重建过程中,进一步去除了残余的对抗扰动,最终实现对对抗攻击的防御效果。