arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2212.02003cs.LGcs.CRcs.CV

基于信息增益的贝叶斯学习可证明地约束鲁棒对抗防御的风险

Bayesian Learning with Information Gain Provably Bounds Risk for a Robust Adversarial Defense

University of Adelaide(阿德莱德大学)

Bao Gia Doan, Ehsan Abbasnejad, Javen Qinfeng Shi, Damith C. Ranasinghe

AI总结:

本文针对贝叶斯神经网络对抗训练中的模式坍缩问题,提出防止模式坍缩并引入信息增益目标以约束对抗风险,在CIFAR-10和STL-10上实现最高20%的鲁棒性提升。

摘要:

我们提出了一种新算法,用于学习对对抗攻击具有鲁棒性的深度神经网络模型。先前算法表明,经过对抗训练的贝叶斯神经网络(BNN)能够提供更强的鲁棒性。我们认识到,用于近似贝叶斯模型多模态后验分布的对抗学习方法可能导致模式坍缩;因此,模型在鲁棒性和性能方面的表现是次优的。为此,我们首先提出防止模式坍缩,以更好地近似多模态后验分布。其次,基于一个直觉——鲁棒模型应当忽略扰动,只考虑输入中的信息性内容——我们构思并形式化了一个信息增益目标,用于度量并强制从良性训练样本和对抗训练样本中学习到的信息保持相似。重要的是,我们证明并展示了最小化信息增益目标可以使对抗风险趋近于传统经验风险。我们相信,我们的工作为对抗训练BNN的原则性方法奠定了基础。我们的模型在CIFAR-10和STL-10数据集上,面对扰动为0.035的PGD攻击时,与对抗训练和Adv-BNN相比,鲁棒性显著提升——最高可达20%。

英文摘要:

We present a new algorithm to learn a deep neural network model robust against adversarial attacks. Previous algorithms demonstrate an adversarially trained Bayesian Neural Network (BNN) provides improved robustness. We recognize the adversarial learning approach for approximating the multi-modal posterior distribution of a Bayesian model can lead to mode collapse; consequently, the model's achievements in robustness and performance are sub-optimal. Instead, we first propose preventing mode collapse to better approximate the multi-modal posterior distribution. Second, based on the intuition that a robust model should ignore perturbations and only consider the informative content of the input, we conceptualize and formulate an information gain objective to measure and force the information learned from both benign and adversarial training instances to be similar. Importantly. we prove and demonstrate that minimizing the information gain objective allows the adversarial risk to approach the conventional empirical risk. We believe our efforts provide a step toward a basis for a principled method of adversarially training BNNs. Our model demonstrate significantly improved robustness--up to 20%--compared with adversarial training and Adv-BNN under PGD attacks with 0.035 distortion on both CIFAR-10 and STL-10 datasets.

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

↑