Skip to main navigation Skip to search Skip to main content

A Simple and Yet Fairly Effective Defense for Graph Neural Networks

  • KTH Royal Institute of Technology
  • Laboratoire d'Informatique (LIX)

Research output: Contribution to journalConference articlepeer-review

Abstract

Graph Neural Networks (GNNs) have emerged as the dominant approach for machine learning on graph-structured data. However, concerns have arisen regarding the vulnerability of GNNs to small adversarial perturbations. Existing defense methods against such perturbations suffer from high time complexity and can negatively impact the model's performance on clean graphs. To address these challenges, this paper introduces NoisyGNNs, a novel defense method that incorporates noise into the underlying model's architecture. We establish a theoretical connection between noise injection and the enhancement of GNN robustness, highlighting the effectiveness of our approach. We further conduct extensive empirical evaluations on the node classification task to validate our theoretical findings, focusing on two popular GNNs: the GCN and GIN. The results demonstrate that NoisyGNN achieves superior or comparable defense performance to existing methods while minimizing added time complexity. The NoisyGNN approach is model-agnostic, allowing it to be integrated with different GNN architectures. Successful combinations of our NoisyGNN approach with existing defense techniques demonstrate even further improved adversarial defense results. Our code is publicly available at: https://github.com/Sennadir/NoisyGNN.

Original languageEnglish
Pages (from-to)21063-21071
Number of pages9
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume38
Issue number19
DOIs
Publication statusPublished - 25 Mar 2024
Event38th AAAI Conference on Artificial Intelligence, AAAI 2024 - Vancouver, Canada
Duration: 20 Feb 202427 Feb 2024

Fingerprint

Dive into the research topics of 'A Simple and Yet Fairly Effective Defense for Graph Neural Networks'. Together they form a unique fingerprint.

Cite this