论文标题

一种有效的融合方法来增强CNN的鲁棒性

An Effective Fusion Method to Enhance the Robustness of CNN

论文作者

Ma, Yating, Lian, Zhichao

论文摘要

随着技术的发展,卷积神经网络的应用改善了我们生活的便利。但是,在图像分类字段中,已经发现,当将某些扰动添加到图像中时,CNN会将其错误分类。因此,已经提出了各种防御方法。先前的方法仅考虑了如何将模块合并到网络中以提高鲁棒性,但并未关注模块合并的方式。在本文中,我们设计了一种新的融合方法来增强CNN的鲁棒性。我们使用基于DOT产品的方法将Denoising模块添加到RESNET18和注意机制,以进一步提高模型的鲁棒性。 CIFAR10上的实验结果表明,我们的方法比FGSM和PGD攻击下的最新方法更好,并且更好。

With the development of technology rapidly, applications of convolutional neural networks have improved the convenience of our life. However, in image classification field, it has been found that when some perturbations are added to images, the CNN would misclassify it. Thus various defense methods have been proposed. The previous approach only considered how to incorporate modules in the network to improve robustness, but did not focus on the way the modules were incorporated. In this paper, we design a new fusion method to enhance the robustness of CNN. We use a dot product-based approach to add the denoising module to ResNet18 and the attention mechanism to further improve the robustness of the model. The experimental results on CIFAR10 have shown that our method is effective and better than the state-of-the-art methods under the attack of FGSM and PGD.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源