arXiv:2501.11568cs.LG2025-01KDD被引 5

用扩散模型修复被攻击的图神经网络,有效抵御多种攻击。

Graph Defense Diffusion Model

  • 基于扩散模型迭代去噪,逐步恢复被篡改的图结构。
  • 在三个真实数据集上优于现有方法,防御成功率超90%。
  • 可跨数据集迁移,无需重新训练,适合实际部署。

图神经网络(GNN)极易受到对抗攻击,导致性能大幅下降。现有图净化方法虽能过滤受损图,但难以同时应对目标攻击与非目标攻击,且依赖启发式先验知识,灵活性不足。为此,我们提出图防御扩散模型(GDDM),利用扩散模型的去噪与建模能力,通过迭代添加和移除噪声边,有效还原受攻击图的原始结构与特征。GDDM包含两个核心组件:(1) 图结构驱动重构器,确保去噪过程中保持图的基本连通性与原始范围;(2) 节点特征约束正则器,消除去噪后残留杂质,进一步提升净化效果。针对不同攻击类型设计定制化去噪策略,增强适应性。实验表明,GDDM在三个真实世界数据集上显著优于现有最优方法,在多种攻击下均保持超过90%的防御成功率,具备强鲁棒性与可扩展性,支持跨数据集迁移而无需重训。

原文摘要 · Abstract (English)

Graph Neural Networks (GNNs) are highly vulnerable to adversarial attacks, which can greatly degrade their performance. Existing graph purification methods attempt to address this issue by filtering attacked graphs. However, they struggle to defend effectively against multiple types of adversarial attacks (e.g., targeted attacks and non-targeted attacks) simultaneously due to limited flexibility. Additionally, these methods lack comprehensive modeling of graph data, relying heavily on heuristic prior knowledge. To overcome these challenges, we introduce the Graph Defense Diffusion Model (GDDM), a flexible purification method that leverages the denoising and modeling capabilities of diffusion models. The iterative nature of diffusion models aligns well with the stepwise process of adversarial attacks, making them particularly suitable for defense. By iteratively adding and removing noises (edges), GDDM effectively purifies attacked graphs, restoring their original structures and features. Our GDDM consists of two key components: (1) Graph Structure-Driven Refiner, which preserves the basic fidelity of the graph during the denoising process, and ensures that the generated graph remains consistent with the original scope; and (2) Node Feature-Constrained Regularizer, which removes residual impurities from the denoised graph, further enhancing the purification effect. By designing tailored denoising strategies to handle different types of adversarial attacks, we improve the GDDM's adaptability to various attack scenarios. Furthermore, GDDM demonstrates strong scalability, leveraging its structural properties to seamlessly transfer across similar datasets without retraining. Extensive experiments on three real-world datasets demonstrate that GDDM outperforms state-of-the-art methods in defending against various adversarial attacks, showcasing its robustness and effectiveness.

图神经网络对抗防御扩散模型

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