A Patch-wise Adversarial Denoising Could Enhance the Robustness of Adversarial Training

Shuai Zhao,Shibin Liu,Boyuan Zhang,Yang Zhai,Ziyi Liu,Yahong Han
DOI: https://doi.org/10.1109/icme57554.2024.10688077
2024-01-01
Abstract:The adversarial examples have demonstrated the vulnerability of machine learning models. While the data augmentation strategy has been a cornerstone in circumventing overfitting within the realm of standard training paradigms, the prior research has proved the limited efficiency of data augmentation in ameliorating overfitting concerns, specifically within the scope of adversarial training. In this work, we have shown that a data augmentation strategy could enhance the generalization of adversarial training. Our framework, Adversarial Denoising-based Training(ADT), first utilizes the adversarial denoising strategy to improve the generalization of adversarial training. Specifically, we use patch-wise adversarial denoising as a data augmentation strategy to boost the robustness of adversarial training. We have evaluated our method on the MNIST, CIFAR-10, Tiny-ImageNet, and GTSRB datasets. The result shows that our framework could successfully improve the adversarial robustness of models against both digital and physical adversarial examples.
What problem does this paper attempt to address?