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

野外全能对抗训练

Omnipotent Adversarial Training in the Wild

Nanyang Technological University(南洋理工大学);Tsinghua University(清华大学)

Guanlin Li, Kangjie Chen, Yuan Xu, Han Qiu, Tianwei Zhang

AI总结:

本文提出全能对抗训练(OAT)策略,通过引入预言机提供正确标签和logits调整对抗训练克服数据不平衡,在不平衡且有噪声的数据集上显著提升模型的干净准确率和对抗鲁棒性。

摘要:

对抗训练是鲁棒深度学习中的一个重要课题,但社区对其实际应用的关注不足。在本文中,我们旨在解决一个现实世界中的挑战,即在不平衡且有噪声的数据集上训练模型,以同时实现高干净准确率和对抗鲁棒性,并提出我们提出的全能对抗训练(OAT)策略。OAT包含两种创新方法来应对训练集中的不完美之处。我们首先在对抗训练过程中引入一个预言机(oracle),以帮助模型学习正确的数据-标签条件分布。这个精心设计的预言机可以为对抗训练提供正确的标签标注。我们进一步提出logits调整对抗训练来克服数据不平衡问题,这可以帮助模型学习贝叶斯最优分布。我们的综合评估结果表明,在数据不平衡和标签噪声场景的复杂组合下,OAT相比其他基线方法在干净准确率上提高了超过20%,在鲁棒准确率上提高了超过10%。代码可以在 https://github.com/GuanlinLee/OAT 找到。

英文摘要:

Adversarial training is an important topic in robust deep learning, but the community lacks attention to its practical usage. In this paper, we aim to resolve a real-world challenge, i.e., training a model on an imbalanced and noisy dataset to achieve high clean accuracy and adversarial robustness, with our proposed Omnipotent Adversarial Training (OAT) strategy. OAT consists of two innovative methodologies to address the imperfection in the training set. We first introduce an oracle into the adversarial training process to help the model learn a correct data-label conditional distribution. This carefully-designed oracle can provide correct label annotations for adversarial training. We further propose logits adjustment adversarial training to overcome the data imbalance issue, which can help the model learn a Bayes-optimal distribution. Our comprehensive evaluation results show that OAT outperforms other baselines by more than 20% clean accuracy improvement and 10% robust accuracy improvement under complex combinations of data imbalance and label noise scenarios. The code can be found in https://github.com/GuanlinLee/OAT.

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

↑