English

Backdoor Attack Against Vision Transformers via Attention Gradient-Based Image Erosion

Computer Vision and Pattern Recognition 2024-11-19 v1 Artificial Intelligence Cryptography and Security

Abstract

Vision Transformers (ViTs) have outperformed traditional Convolutional Neural Networks (CNN) across various computer vision tasks. However, akin to CNN, ViTs are vulnerable to backdoor attacks, where the adversary embeds the backdoor into the victim model, causing it to make wrong predictions about testing samples containing a specific trigger. Existing backdoor attacks against ViTs have the limitation of failing to strike an optimal balance between attack stealthiness and attack effectiveness. In this work, we propose an Attention Gradient-based Erosion Backdoor (AGEB) targeted at ViTs. Considering the attention mechanism of ViTs, AGEB selectively erodes pixels in areas of maximal attention gradient, embedding a covert backdoor trigger. Unlike previous backdoor attacks against ViTs, AGEB achieves an optimal balance between attack stealthiness and attack effectiveness, ensuring the trigger remains invisible to human detection while preserving the model's accuracy on clean samples. Extensive experimental evaluations across various ViT architectures and datasets confirm the effectiveness of AGEB, achieving a remarkable Attack Success Rate (ASR) without diminishing Clean Data Accuracy (CDA). Furthermore, the stealthiness of AGEB is rigorously validated, demonstrating minimal visual discrepancies between the clean and the triggered images.

Keywords

Cite

@article{arxiv.2410.22678,
  title  = {Backdoor Attack Against Vision Transformers via Attention Gradient-Based Image Erosion},
  author = {Ji Guo and Hongwei Li and Wenbo Jiang and Guoming Lu},
  journal= {arXiv preprint arXiv:2410.22678},
  year   = {2024}
}

Comments

Accepted by IEEE GLOBECOM 2024

R2 v1 2026-06-28T19:40:37.170Z