arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.01705cs.CVcs.AI
收藏

下游迁移攻击:针对预训练视觉Transformer的下游模型对抗攻击

Downstream Transfer Attack: Adversarial Attacks on Downstream Models with Pre-trained Vision Transformers

Fudan University(复旦大学);The University of Melbourne(墨尔本大学);Ant Group(蚂蚁集团);Tongji University(同济大学)

Weijie Zheng, Xingjun Ma, Hanxun Huang, Zuxuan Wu, Yu-Gang Jiang

AI总结:

本文提出下游迁移攻击(DTA)方法,利用预训练ViT中最脆弱层和余弦相似度损失生成高可迁移对抗样本,在多种预训练与微调设置下平均攻击成功率超90%,并可用于提升模型鲁棒性。

摘要:

随着视觉Transformer(ViTs)和自监督学习(SSL)技术的进步,预训练的大型ViTs已成为计算机视觉应用的新基础模型。然而,研究表明,与卷积神经网络(CNNs)一样,ViTs也容易受到对抗攻击,输入中的细微扰动即可欺骗模型做出错误预测。本文研究了这种对抗脆弱性从预训练ViT模型到下游任务的迁移性。我们专注于逐样本迁移攻击,并提出了一种名为下游迁移攻击(Downstream Transfer Attack,DTA)的新攻击方法。对于给定的测试图像,DTA利用预训练ViT模型生成对抗样本,然后将该对抗样本应用于攻击在下游数据集上微调后的模型版本。在攻击过程中,DTA以余弦相似度损失为指导,识别并利用预训练模型中最脆弱的层来生成高度可迁移的攻击。通过在3种不同预训练方法、3种微调方案和10个不同下游数据集上对预训练ViTs进行的大量实验,我们表明DTA实现了超过90%的平均攻击成功率(ASR),大幅超越现有方法。当与对抗训练结合使用时,我们的DTA生成的对抗样本可以显著提高模型对不同下游迁移攻击的鲁棒性。

英文摘要:

With the advancement of vision transformers (ViTs) and self-supervised learning (SSL) techniques, pre-trained large ViTs have become the new foundation models for computer vision applications. However, studies have shown that, like convolutional neural networks (CNNs), ViTs are also susceptible to adversarial attacks, where subtle perturbations in the input can fool the model into making false predictions. This paper studies the transferability of such an adversarial vulnerability from a pre-trained ViT model to downstream tasks. We focus on \emph{sample-wise} transfer attacks and propose a novel attack method termed \emph{Downstream Transfer Attack (DTA)}. For a given test image, DTA leverages a pre-trained ViT model to craft the adversarial example and then applies the adversarial example to attack a fine-tuned version of the model on a downstream dataset. During the attack, DTA identifies and exploits the most vulnerable layers of the pre-trained model guided by a cosine similarity loss to craft highly transferable attacks. Through extensive experiments with pre-trained ViTs by 3 distinct pre-training methods, 3 fine-tuning schemes, and across 10 diverse downstream datasets, we show that DTA achieves an average attack success rate (ASR) exceeding 90\%, surpassing existing methods by a huge margin. When used with adversarial training, the adversarial examples generated by our DTA can significantly improve the model's robustness to different downstream transfer attacks.

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

↑