arXiv · 2208.04946
Attention Hijacking in Trojan Transformers
Abstract
Trojan attacks pose a severe threat to AI systems. Recent works on Transformer models received explosive popularity and the self-attentions are now indisputable. This raises a central question: Can we reveal the Trojans through attention mechanisms in BERTs and ViTs? In this paper, we investigate the attention hijacking pattern in Trojan AIs, \ie, the trigger token ``kidnaps'' the attention weights when a specific trigger is present. We observe the consistent attention hijacking pattern in Trojan Transformers from both Natural Language Processing (NLP) and Computer Vision (CV) domains. This intriguing property helps us to understand the Trojan mechanism in BERTs and ViTs. We also propose an Attention-Hijacking Trojan Detector (AHTD) to discriminate the Trojan AIs from the clean ones.
Explore related subjects
Keep this discovery
Weimin Lyu, Songzhu Zheng, Tengfei Ma, Haibin Ling, Chao Chen. 2022-08-09. Attention Hijacking in Trojan Transformers. https://arxiv.org/abs/2208.04946
Cite the original work for its findings. Save a collection to share your selection of sources.