Framework GNN-AID: Graph Neural Network Analysis Interpretation and Defense
By: Kirill Lukyanov , Mikhail Drobyshevskiy , Georgii Sazonov and more
Potential Business Impact:
Helps AI understand and protect graph data.
The growing need for Trusted AI (TAI) highlights the importance of interpretability and robustness in machine learning models. However, many existing tools overlook graph data and rarely combine these two aspects into a single solution. Graph Neural Networks (GNNs) have become a popular approach, achieving top results across various tasks. We introduce GNN-AID (Graph Neural Network Analysis, Interpretation, and Defense), an open-source framework designed for graph data to address this gap. Built as a Python library, GNN-AID supports advanced trust methods and architectural layers, allowing users to analyze graph datasets and GNN behavior using attacks, defenses, and interpretability methods. GNN-AID is built on PyTorch-Geometric, offering preloaded datasets, models, and support for any GNNs through customizable interfaces. It also includes a web interface with tools for graph visualization and no-code features like an interactive model builder, simplifying the exploration and analysis of GNNs. The framework also supports MLOps techniques, ensuring reproducibility and result versioning to track and revisit analyses efficiently. GNN-AID is a flexible tool for developers and researchers. It helps developers create, analyze, and customize graph models, while also providing access to prebuilt datasets and models for quick experimentation. Researchers can use the framework to explore advanced topics on the relationship between interpretability and robustness, test defense strategies, and combine methods to protect against different types of attacks. We also show how defenses against evasion and poisoning attacks can conflict when applied to graph data, highlighting the complex connections between defense strategies. GNN-AID is available at \href{https://github.com/ispras/GNN-AID}{github.com/ispras/GNN-AID}
Similar Papers
Explainable Android Malware Detection and Malicious Code Localization Using Graph Attention
Cryptography and Security
Finds hidden bad code in phone apps.
Graph Neural Networks in Modern AI-aided Drug Discovery
Biomolecules
Helps find new medicines faster.
Enhancing Explainability of Graph Neural Networks Through Conceptual and Structural Analyses and Their Extensions
Artificial Intelligence
Explains how computer graphs make decisions.