提出MADE方法,有效防御图神经网络的后门攻击。
MADE: Graph Backdoor Defense with Masked Unlearning
- 设计对抗性掩码生成机制,保留干净子图并抑制触发器影响。
- 在多个图分类任务中,攻击成功率降至不足5%,准确率保持高位。
- 适用于药物发现等敏感领域,对图像防御方法不适用的场景有效。
图神经网络(GNN)在社交网络分析、蛋白质设计等图相关任务中表现优异,但近期研究揭示其易受后门攻击:攻击者通过向训练数据注入触发器,使模型在输入包含触发模式时错误预测目标标签。此类漏洞对药物发现等敏感应用构成重大安全风险。尽管图像后门防御研究丰富,但直接应用于图数据效果不佳。本文首先分析图像与图后门攻击的关键差异,提出新型防御方法MADE——通过对抗性掩码生成机制,选择性保留干净子图,并利用掩码调节边权重,有效消除触发器影响。大量实验表明,MADE在多种图分类任务中显著降低攻击成功率(ASR),同时保持高分类准确率。
原文摘要 · Abstract (English)
Graph Neural Networks (GNNs) have garnered significant attention from researchers due to their outstanding performance in handling graph-related tasks, such as social network analysis, protein design, and so on. Despite their widespread application, recent research has demonstrated that GNNs are vulnerable to backdoor attacks, implemented by injecting triggers into the training datasets. Trained on the poisoned data, GNNs will predict target labels when attaching trigger patterns to inputs. This vulnerability poses significant security risks for applications of GNNs in sensitive domains, such as drug discovery. While there has been extensive research into backdoor defenses for images, strategies to safeguard GNNs against such attacks remain underdeveloped. Furthermore, we point out that conventional backdoor defense methods designed for images cannot work well when directly implemented on graph data. In this paper, we first analyze the key difference between image backdoor and graph backdoor attacks. Then we tackle the graph defense problem by presenting a novel approach called MADE, which devises an adversarial mask generation mechanism that selectively preserves clean sub-graphs and further leverages masks on edge weights to eliminate the influence of triggers effectively. Extensive experiments across various graph classification tasks demonstrate the effectiveness of MADE in significantly reducing the attack success rate (ASR) while maintaining a high classification accuracy.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。