arXiv:2412.08555cs.LG2024-12

Grimm可无缝防御图神经网络中毒攻击,不改动原模型且高效并行。

Grimm: A Plug-and-Play Perturbation Rectifier for Graph Neural Networks Defending against Poisoning Attacks

  • 基于节点特征轨迹异常检测,构建类免疫系统防御机制。
  • 在多种GNN模型上实现0.8%~3.2%准确率提升,有效修复恶意边。
  • 无需修改原模型,适合部署于各类图神经网络场景。

近期研究揭示了图神经网络(GNN)在节点分类任务中易受对抗性投毒攻击。现有防御方法需替换原始GNN模型,无论其类型如何,这会损害先前提升实际性能的研究成果。本文提出Grimm,首个即插即用的防御模型。仅需从任意层提取特征,Grimm即可无缝修正扰动。通过分析训练过程中节点特征轨迹(FTs)的变化,理论上证明受害节点的FTs必然出现可区分异常。受生物神经与免疫系统平行性的启发,构建了完整的类免疫系统用于GNN。Grimm不仅能在训练中检测并修正异常特征轨迹,还能并行运行,实现监测、检测与修复的独立执行。实验验证其四大优势:1)无害性,不主动干扰训练;2)并行性,各功能独立于训练过程;3)通用性,兼容GCN、GAT、GraphSAGE等主流模型;4)可迁移性,异常检测器可跨系统迁移,实现一步修正。

原文摘要 · Abstract (English)

Recent studies have revealed the vulnerability of graph neural networks (GNNs) to adversarial poisoning attacks on node classification tasks. Current defensive methods require substituting the original GNNs with defense models, regardless of the original's type. This approach, while targeting adversarial robustness, compromises the enhancements developed in prior research to boost GNNs' practical performance. Here we introduce Grimm, the first plug-and-play defense model. With just a minimal interface requirement for extracting features from any layer of the protected GNNs, Grimm is thus enabled to seamlessly rectify perturbations. Specifically, we utilize the feature trajectories (FTs) generated by GNNs, as they evolve through epochs, to reflect the training status of the networks. We then theoretically prove that the FTs of victim nodes will inevitably exhibit discriminable anomalies. Consequently, inspired by the natural parallelism between the biological nervous and immune systems, we construct Grimm, a comprehensive artificial immune system for GNNs. Grimm not only detects abnormal FTs and rectifies adversarial edges during training but also operates efficiently in parallel, thereby mirroring the concurrent functionalities of its biological counterparts. We experimentally confirm that Grimm offers four empirically validated advantages: 1) Harmlessness, as it does not actively interfere with GNN training; 2) Parallelism, ensuring monitoring, detection, and rectification functions operate independently of the GNN training process; 3) Generalizability, demonstrating compatibility with mainstream GNNs such as GCN, GAT, and GraphSAGE; and 4) Transferability, as the detectors for abnormal FTs can be efficiently transferred across different systems for one-step rectification.

图神经网络对抗攻击防御机制即插即用

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