arXiv:2412.04776cs.CVcs.CR2024-12

提出新型视觉变压器后门攻击,无需篡改标签即可隐蔽植入攻击

Megatron: Evasive Clean-Label Backdoor Attacks against Vision Transformer

  • 通过注意力机制设计新损失函数生成隐蔽触发器
  • 在多个数据集上实现超90%攻击成功率且位置偏移仍有效
  • 对人工观察和主流防御策略均具备更强隐蔽性

视觉变压器在各类视觉任务中表现优异,但其对后门攻击的脆弱性尚未充分研究。现有工作多聚焦于污染标签攻击,若训练者修正标签则可能失效。本文提出Megatron,一种针对视觉变压器的逃避式干净标签后门攻击,攻击者无需修改数据标注即可植入后门。为生成有效触发器,基于变换器中的注意力机制定制了两种损失:潜在损失使带触发样本与目标类别干净样本在最后一层注意力上对齐;注意力扩散损失强调包含触发区域的注意力扩散区。理论分析支持注意力扩散损失的设计合理性。在CIFAR-10、GTSRB、CIFAR-100和Tiny ImageNet上的大量实验表明,即使测试时触发器位置轻微偏移,Megatron仍可实现超过90%的攻击成功率。此外,相较于基线方法,Megatron在人类视觉检查及多种防御策略(如DBAVT、BAVT、Beatrix、TeCo、SAGE)下展现出更优的隐蔽性。

原文摘要 · Abstract (English)

Vision transformers have achieved impressive performance in various vision-related tasks, but their vulnerability to backdoor attacks is under-explored. A handful of existing works focus on dirty-label attacks with wrongly-labeled poisoned training samples, which may fail if a benign model trainer corrects the labels. In this paper, we propose Megatron, an evasive clean-label backdoor attack against vision transformers, where the attacker injects the backdoor without manipulating the data-labeling process. To generate an effective trigger, we customize two loss terms based on the attention mechanism used in transformer networks, i.e., latent loss and attention diffusion loss. The latent loss aligns the last attention layer between triggered samples and clean samples of the target label. The attention diffusion loss emphasizes the attention diffusion area that encompasses the trigger. A theoretical analysis is provided to underpin the rationale behind the attention diffusion loss. Extensive experiments on CIFAR-10, GTSRB, CIFAR-100, and Tiny ImageNet demonstrate the effectiveness of Megatron. Megatron can achieve attack success rates of over 90% even when the position of the trigger is slightly shifted during testing. Furthermore, Megatron achieves better evasiveness than baselines regarding both human visual inspection and defense strategies (i.e., DBAVT, BAVT, Beatrix, TeCo, and SAGE).

后门攻击视觉变压器隐蔽性注意力机制

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