Are You Using Reliable Graph Prompts? Trojan Prompt Attacks on Graph Neural Networks

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Lin, Minhua, Zhang, Zhiwei, Dai, Enyan, Wu, Zongyu, Wang, Yilong, Zhang, Xiang, Wang, Suhang
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912401549426688
author Lin, Minhua
Zhang, Zhiwei
Dai, Enyan
Wu, Zongyu
Wang, Yilong
Zhang, Xiang
Wang, Suhang
author_facet Lin, Minhua
Zhang, Zhiwei
Dai, Enyan
Wu, Zongyu
Wang, Yilong
Zhang, Xiang
Wang, Suhang
contents Graph Prompt Learning (GPL) has been introduced as a promising approach that uses prompts to adapt pre-trained GNN models to specific downstream tasks without requiring fine-tuning of the entire model. Despite the advantages of GPL, little attention has been given to its vulnerability to backdoor attacks, where an adversary can manipulate the model's behavior by embedding hidden triggers. Existing graph backdoor attacks rely on modifying model parameters during training, but this approach is impractical in GPL as GNN encoder parameters are frozen after pre-training. Moreover, downstream users may fine-tune their own task models on clean datasets, further complicating the attack. In this paper, we propose TGPA, a backdoor attack framework designed specifically for GPL. TGPA injects backdoors into graph prompts without modifying pre-trained GNN encoders and ensures high attack success rates and clean accuracy. To address the challenge of model fine-tuning by users, we introduce a finetuning-resistant poisoning approach that maintains the effectiveness of the backdoor even after downstream model adjustments. Extensive experiments on multiple datasets under various settings demonstrate the effectiveness of TGPA in compromising GPL models with fixed GNN encoders.
format Preprint
id arxiv_https___arxiv_org_abs_2410_13974
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Are You Using Reliable Graph Prompts? Trojan Prompt Attacks on Graph Neural Networks
Lin, Minhua
Zhang, Zhiwei
Dai, Enyan
Wu, Zongyu
Wang, Yilong
Zhang, Xiang
Wang, Suhang
Machine Learning
Cryptography and Security
Graph Prompt Learning (GPL) has been introduced as a promising approach that uses prompts to adapt pre-trained GNN models to specific downstream tasks without requiring fine-tuning of the entire model. Despite the advantages of GPL, little attention has been given to its vulnerability to backdoor attacks, where an adversary can manipulate the model's behavior by embedding hidden triggers. Existing graph backdoor attacks rely on modifying model parameters during training, but this approach is impractical in GPL as GNN encoder parameters are frozen after pre-training. Moreover, downstream users may fine-tune their own task models on clean datasets, further complicating the attack. In this paper, we propose TGPA, a backdoor attack framework designed specifically for GPL. TGPA injects backdoors into graph prompts without modifying pre-trained GNN encoders and ensures high attack success rates and clean accuracy. To address the challenge of model fine-tuning by users, we introduce a finetuning-resistant poisoning approach that maintains the effectiveness of the backdoor even after downstream model adjustments. Extensive experiments on multiple datasets under various settings demonstrate the effectiveness of TGPA in compromising GPL models with fixed GNN encoders.
title Are You Using Reliable Graph Prompts? Trojan Prompt Attacks on Graph Neural Networks
topic Machine Learning
Cryptography and Security
url https://arxiv.org/abs/2410.13974