arXiv:2412.10681cs.CV2024-12

单像素即可攻破ViT模型,揭示其脆弱性机制

One Pixel is All I Need

  • 用梯度敏感图定位ViT关键敏感区域,指导生成弱触发器
  • 仅需1个像素污染,训练1轮即可让所有验证集失效
  • 对齐视觉注意力模式,适合研究模型鲁棒性者参考

视觉变换器(ViTs)在各类视觉任务中表现卓越,但其对后门攻击的鲁棒性引发关注。此类攻击通过关联特定触发器与目标标签,使模型在触发器存在时预测指定标签,而正常图像仍能正确识别。我们发现,相较于卷积神经网络(CNNs),ViTs对准触发器(即与原始触发器相似但不同的模式)具有更高的攻击成功率;此外,干净样本中某些后门特征会抑制原始触发器,进一步提升准触发器效果。为深入理解并利用这些漏洞,我们提出扰动敏感度分布图(Perturbation Sensitivity Distribution Map, PSDM),通过在大量输入上计算并求和梯度,揭示模型对输入微小变化的敏感性。在ViTs中,PSDM显示中心像素比边缘更敏感,呈现块状模式。基于此,我们设计了“WorstVIT”——一种简单却高效的针对ViT的数据中毒后门攻击。该方法仅需极低污染率,训练一轮,仅修改一个像素,即可成功攻击全部验证图像。

原文摘要 · Abstract (English)

Vision Transformers (ViTs) have achieved record-breaking performance in various visual tasks. However, concerns about their robustness against backdoor attacks have grown. Backdoor attacks involve associating a specific trigger with a target label, causing the model to predict the attacker-specified label when the trigger is present, while correctly identifying clean images.We found that ViTs exhibit higher attack success rates for quasi-triggers(patterns different from but similar to the original training triggers)compared to CNNs. Moreover, some backdoor features in clean samples can suppress the original trigger, making quasi-triggers more effective.To better understand and exploit these vulnerabilities, we developed a tool called the Perturbation Sensitivity Distribution Map (PSDM). PSDM computes and sums gradients over many inputs to show how sensitive the model is to small changes in the input. In ViTs, PSDM reveals a patch-like pattern where central pixels are more sensitive than edges. We use PSDM to guide the creation of quasi-triggers.Based on these findings, we designed "WorstVIT," a simple yet effective data poisoning backdoor for ViT models. This attack requires an extremely low poisoning rate, trains for just one epoch, and modifies a single pixel to successfully attack all validation images.

ViT后门攻击数据中毒鲁棒性

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。