[发明专利]针对深度神经网络对抗样本的对抗噪声去除方法及系统在审
申请号: | 202111252920.9 | 申请日: | 2021-10-27 |
公开(公告)号: | CN113963213A | 公开(公告)日: | 2022-01-21 |
发明(设计)人: | 杨杰;李佳文 | 申请(专利权)人: | 上海交通大学 |
主分类号: | G06V10/764 | 分类号: | G06V10/764;G06V10/82;G06V10/30;G06N3/04;G06N3/08;G06T5/00 |
代理公司: | 上海恒慧知识产权代理事务所(特殊普通合伙) 31317 | 代理人: | 徐红银;张琳 |
地址: | 200240 *** | 国省代码: | 上海;31 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 针对 深度 神经网络 对抗 样本 噪声 去除 方法 系统 | ||
1.一种针对深度神经网络对抗样本的对抗噪声去除方法,其特征在于,包括:
利用干净图像样本对条件变分自编码器与分类器进行端到端训练,获得训练后的条件变分自编码器与分类器;
将待去噪的对抗图像输入至所述训练后的条件变分自编码器和分类器,获得所述对抗图像的去噪样本与类别。
2.根据权利要求1所述的针对深度神经网络对抗样本的对抗噪声去除方法,其特征在于,所述利用干净图像样本对条件变分自编码器与分类器进行端到端训练,获得训练后的条件变分自编码器与分类器,包括:
S101,将条件变分自编码器与分类器直接连接,输入干净图像与对应的类别标签配对;
条件变分自编码器中的编码器对输入干净图像与对应类别标签进行编码至特征空间,得到特征空间中潜在变量所服从分布的均值与标准差:
μ,σ=Encoder(xclean,ytrue)
其中,μ表示编码得到的潜在变量z的均值,σ为编码得到的潜在变量z的标准差,xclean为输入干净图像,ytrue为干净图像对应的类别标签;
S102,在所述特征空间采样得到潜在变量z:
z=μ+ε·σ
ε~N(0,I)
其中ε代表服从N(0,I)正态分布的随机超参数,通过上述重采样方法,得到服从正态分布N(μ,ε)的潜在变量z;
S103,将潜在变量z输入条件变分自编码器中的解码器,重建干净输入图像:
xrecon=Decoder(z,ytrue)
其中xrecon表示解码器重建后的干净输入图像,解码过程同样是有监督的,需要额外输入类别标签ytrue;
S104,将所述重建后的干净输入图像输入分类器,得到依据重建样本预测的类别ypred:
ypred=Classifier(xrecon)
S105,将所述条件变分自编码器与分类器的损失函数组成联合损失函数,利用Adam优化算法端到端训练:
L1=xcleanlog xrecon+(1-xclean)log(1-xrecon)
L2=μ2+σ2-logσ2-1
L3=ytruelog ypred+(1-ytrue)log(1-ypred)
L=L1+L2+L3+Lreg
其中,训练后的条件变分自编码器与分类器模型的损失函数分为四部分:条件变分自编码器的损失L1与L2,分类器的分类损失L3,和控制条件变分自编码器参数大小的正则化损失Lreg,L1和L3采用交叉熵损失,L2采用KL散度损失,正则化损失函数Lreg则定义为编码器与解码器参数的二范数。
3.根据权利要求2所述的一种针对深度神经网络对抗样本的对抗噪声去除方法,其特征在于,所述条件变分自编码器包括:
基础条件变分自编码器;
在所述基础条件变分自编码器的标签输入端额外加入的多层全连接层,所述多层全连接层用于提高便签的维度和标签在编码操作中所占权重;
在所述基础条件变分自编码器的标签输入端额外加入的用于避免编码过程中出现方差漂移现象的多个BN层;以及,
在所述基础条件变分自编码器输出的均值后额外加入的用于改善编码效果的BN层。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于上海交通大学,未经上海交通大学许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/202111252920.9/1.html,转载请声明来源钻瓜专利网。