arXiv:2410.13974cs.LGcs.CR2024-10KDD被引 5

提出针对图神经网络提示学习的隐蔽后门攻击,无需修改模型参数即可生效。

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

  • 通过操纵图提示注入后门,不改动预训练编码器
  • 攻击成功率高且在用户微调后仍有效,保持正常精度
  • 适用于评估提示学习系统安全性,适合安全研究者阅读

图提示学习(GPL)作为一种新兴方法,利用提示适配预训练图神经网络(GNN)以完成下游任务,无需全模型微调。尽管具有优势,但其对后门攻击的脆弱性尚未受到足够关注——攻击者可通过嵌入隐藏触发器操控模型行为。现有图后门攻击依赖训练时修改模型参数,但在GPL中因编码器参数冻结而不可行;此外,下游用户可能在干净数据上进行微调,进一步增加攻击难度。本文提出TGPA,一种专为GPL设计的后门攻击框架,可在不修改预训练GNN编码器的前提下,将后门注入图提示中,并保证高攻击成功率与正常准确率。针对用户微调问题,引入抗微调污染策略,确保后门在下游模型调整后依然有效。在多个数据集上的大量实验表明,即使编码器固定,TGPA仍能有效破坏GPL模型。

原文摘要 · Abstract (English)

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.

图神经网络后门攻击提示学习

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