arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.12435cs.CVcs.LG
收藏

从卷积层特征图视角理解对抗鲁棒性

Understanding Adversarial Robustness from Feature Maps of Convolutional Layers

Yantai University(烟台大学);East China Normal University(华东师范大学)

Cong Xu, Wei Zhang, Jun Wang, Min Yang

AI总结:

该研究从卷积层特征图角度探究网络抗扰动能力,发现平均池化前更大的特征图可提升抗扰性,据此通过上采样或调步长改进架构,在多个基准模型上实现准确率与鲁棒性双提升。

摘要:

神经网络的对抗鲁棒性主要取决于两个因素:模型容量与抗扰动能力。本文从卷积层特征图的角度研究网络的抗扰动能力。我们的理论分析发现,平均池化前的卷积特征图尺寸越大,对扰动的抵抗能力越强,但该结论对最大池化不成立。这一发现为鲁棒神经网络的设计带来了新启发,也推动我们将这些结论用于改进现有架构。我们提出的修改方案十分简单,仅需对输入进行上采样,或微调下采样算子的步长配置。我们在AlexNet、VGG、ResNet18、PreActResNet18等多个基准神经网络架构上验证了所提方法,在多种攻击与防御机制下,自然准确率与对抗鲁棒性均能取得显著提升。代码可在https://github.com/MTandHJ/rcm 获取。

英文摘要:

The adversarial robustness of a neural network mainly relies on two factors: model capacity and anti-perturbation ability. In this paper, we study the anti-perturbation ability of the network from the feature maps of convolutional layers. Our theoretical analysis discovers that larger convolutional feature maps before average pooling can contribute to better resistance to perturbations, but the conclusion is not true for max pooling. It brings new inspiration to the design of robust neural networks and urges us to apply these findings to improve existing architectures. The proposed modifications are very simple and only require upsampling the inputs or slightly modifying the stride configurations of downsampling operators. We verify our approaches on several benchmark neural network architectures, including AlexNet, VGG, RestNet18, and PreActResNet18. Non-trivial improvements in terms of both natural accuracy and adversarial robustness can be achieved under various attack and defense mechanisms. The code is available at \url{https://github.com/MTandHJ/rcm}.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑