提出DMGNN模型,有效检测并清除图神经网络中的隐蔽后门攻击。
DMGNN: Detecting and Mitigating Backdoor Attacks in Graph Neural Networks
- 基于反事实解释预测标签转移,识别隐藏的后门触发器。
- 通过逆向采样剪枝过滤触发特征,降低攻击成功率至5%。
- 兼顾防御效果与模型性能,适合高安全要求的图学习场景。
近期研究表明,图神经网络(GNN)极易受到多种对抗攻击。其中,图后门攻击尤为严重:攻击者在训练阶段注入触发器并修改目标标签,使模型在特定输入下误分类。根据触发器特征分布差异,攻击可分为分布外(OOD)和分布内(ID)两类——前者触发器与正常样本特征差异明显,后者则几乎与正常样本一致。现有方法可有效防御OOD攻击,但对隐蔽的ID攻击束手无策,且在防御过程中损害主任务准确率。为此,本文提出DMGNN,能同时抵御OOD与ID攻击,在消除隐蔽性的同时提升模型性能。具体地,通过反事实解释预测标签转移,精准识别隐藏的触发器;进一步设计逆向采样剪枝方法,从数据层面筛选并剔除触发特征。在多个公开图数据集上的实验表明,DMGNN显著优于现有最优防御方法,将攻击成功率降至5%,模型性能损失控制在3.5%以内。
原文摘要 · Abstract (English)
Recent studies have revealed that GNNs are highly susceptible to multiple adversarial attacks. Among these, graph backdoor attacks pose one of the most prominent threats, where attackers cause models to misclassify by learning the backdoored features with injected triggers and modified target labels during the training phase. Based on the features of the triggers, these attacks can be categorized into out-of-distribution (OOD) and in-distribution (ID) graph backdoor attacks, triggers with notable differences from the clean sample feature distributions constitute OOD backdoor attacks, whereas the triggers in ID backdoor attacks are nearly identical to the clean sample feature distributions. Existing methods can successfully defend against OOD backdoor attacks by comparing the feature distribution of triggers and clean samples but fail to mitigate stealthy ID backdoor attacks. Due to the lack of proper supervision signals, the main task accuracy is negatively affected in defending against ID backdoor attacks. To bridge this gap, we propose DMGNN against OOD and ID graph backdoor attacks that can powerfully eliminate stealthiness to guarantee defense effectiveness and improve the model performance. Specifically, DMGNN can easily identify the hidden ID and OOD triggers via predicting label transitions based on counterfactual explanation. To further filter the diversity of generated explainable graphs and erase the influence of the trigger features, we present a reverse sampling pruning method to screen and discard the triggers directly on the data level. Extensive experimental evaluations on open graph datasets demonstrate that DMGNN far outperforms the state-of-the-art (SOTA) defense methods, reducing the attack success rate to 5% with almost negligible degradation in model performance (within 3.5%).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。