中文

关于对抗训练抵御常见 corruptions 的有效性

机器学习 2022-01-05 v2 人工智能 计算机视觉与模式识别 机器学习

摘要

关于针对常见 corruptions 的鲁棒性的文献中,对于对抗训练是否能在此设定下提升性能尚无共识。首先,我们表明,当以适当选择的扰动半径使用时,p\ell_p 对抗训练可作为抵御常见 corruptions 的强基线,同时提升准确率和校准性。随后我们解释了为何对抗训练优于使用简单高斯噪声的数据增强——后者已被观察到是常见 corruptions 上一个有意义的基线。与此相关,我们识别出高斯增强过拟合到用于训练的特定标准差时的 σ\sigma-过拟合现象,这对常见 corruption 准确率有显著的 detrimental 影响。我们讨论了如何缓解此问题,进而如何通过引入以学习到的感知图像块相似度作为距离度量的对抗训练的高效松弛来进一步增强 p\ell_p 对抗训练。通过在 CIFAR-10 和 ImageNet-100 上的实验,我们表明我们的方法不仅改进了 p\ell_p 对抗训练基线,而且与 AugMix、DeepAugment、ANT 和 SIN 等数据增强方法具有累积增益,从而在常见 corruptions 上取得最先进的性能。我们实验的代码公开于 https://github.com/tml-epfl/adv-training-corruptions。

关键词

引用

@article{arxiv.2103.02325,
  title  = {On the effectiveness of adversarial training against common corruptions},
  author = {Klim Kireev and Maksym Andriushchenko and Nicolas Flammarion},
  journal= {arXiv preprint arXiv:2103.02325},
  year   = {2022}
}

备注

New calibration results, more comprehensive experimental evaluation (e.g., new results with AugMix+JSD and DeepAugment)