arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.03340cs.CVcs.CRcs.LG

面向未知对抗攻击可泛化鲁棒性的元不变性防御

Meta Invariance Defense Towards Generalizable Robustness to Unknown Adversarial Attacks

Chongqing University(重庆大学);Peng Cheng Lab(鹏城实验室);Zhejiang University(浙江大学);Chongqing University of Posts and Telecommunications(重庆邮电大学)

Lei Zhang, Yuhang Zhou, Yi Yang, Xinbo Gao

AI总结:

针对部署阶段未知攻击的零样本防御问题,本文提出MID,通过攻击者池采样任务和多层级多一致性元蒸馏学习攻击不变特征,在分类与图像再生上均取得可泛化鲁棒性。

摘要:

尽管深度神经网络(DNN)模型为计算机视觉任务提供了高性能解决方案,但它已被证明极易受到对抗攻击。当前防御主要集中于已知攻击,而对未知攻击的对抗鲁棒性却被严重忽视。此外,常用的自适应学习和微调技术并不适用于对抗防御,因为模型部署时本质上是一个零样本问题。因此,为应对这一挑战,我们提出了一种与攻击无关的防御方法,名为元不变性防御(Meta Invariance Defense,MID)。具体而言,从人工构建的攻击者池(Attacker Pool)中随机采样多种对抗攻击组合,以构成针对未知攻击的不同防御任务;在这些任务中,学生编码器通过多一致性蒸馏接受监督,并依据元原则学习攻击不变特征。所提出的MID具有两个优点:1)在良性样本与对抗样本之间进行像素级、特征级和预测级的全面蒸馏,有助于发现攻击不变性。2)模型同时实现了对高级图像分类中不可感知对抗扰动的鲁棒性,以及低级鲁棒图像再生中的攻击抑制。在ImageNet等众多基准上的理论和实证研究验证了MID在多种攻击下的可泛化鲁棒性和优越性。

英文摘要:

Despite providing high-performance solutions for computer vision tasks, the deep neural network (DNN) model has been proved to be extremely vulnerable to adversarial attacks. Current defense mainly focuses on the known attacks, but the adversarial robustness to the unknown attacks is seriously overlooked. Besides, commonly used adaptive learning and fine-tuning technique is unsuitable for adversarial defense since it is essentially a zero-shot problem when deployed. Thus, to tackle this challenge, we propose an attack-agnostic defense method named Meta Invariance Defense (MID). Specifically, various combinations of adversarial attacks are randomly sampled from a manually constructed Attacker Pool to constitute different defense tasks against unknown attacks, in which a student encoder is supervised by multi-consistency distillation to learn the attack-invariant features via a meta principle. The proposed MID has two merits: 1) Full distillation from pixel-, feature- and prediction-level between benign and adversarial samples facilitates the discovery of attack-invariance. 2) The model simultaneously achieves robustness to the imperceptible adversarial perturbations in high-level image classification and attack-suppression in low-level robust image regeneration. Theoretical and empirical studies on numerous benchmarks such as ImageNet verify the generalizable robustness and superiority of MID under various attacks.

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

↑