arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2505.03424cs.LGcs.AI
收藏

图神经网络分析解释与防御框架 GNN-AID

Framework GNN-AID: Graph Neural Network Analysis Interpretation and Defense

ISP RAS Research Center for Trusted Artificial Intelligence(俄罗斯科学院可信人工智能研究信息与处理中心);Ivannikov Institute for System Programming of the Russian Academy of Sciences(俄罗斯科学院伊万诺夫系统编程研究所);Moscow Institute of Physics and Technology (National Research University)(莫斯科物理技术学院(国家研究大学));Lomonosov Moscow State University(罗蒙诺索夫莫斯科国立大学);AIRI

Kirill Lukyanov, Mikhail Drobyshevskiy, Georgii Sazonov, Mikhail Soloviov, Ilya Makarov

AI总结:

本文提出GNN-AID框架,旨在解决图数据中可解释性与鲁棒性结合的问题,提供分析、解释和防御工具,支持多种攻击与防御方法,并提供可视化和MLOps技术以提升可重复性。

摘要:

随着可信AI(TAI)需求的增加,机器学习模型的可解释性和鲁棒性变得尤为重要。然而,许多现有工具忽视了图数据,并且很少将这两个方面结合起来。图神经网络(GNNs)已成为一种流行的方法,在各种任务中取得了最佳结果。我们介绍了GNN-AID(图神经网络分析、解释和防御),这是一个开源框架,旨在解决这一差距。GNN-AID是基于PyTorch-Geometric构建的Python库,支持先进的信任方法和架构层,允许用户使用攻击、防御和可解释性方法分析图数据集和GNN行为。GNN-AID还包含一个网页界面,提供图可视化工具和无代码功能,如交互式模型构建器,简化了GNN的探索和分析。该框架还支持MLOps技术,确保可重复性和结果版本控制,以高效地跟踪和重新访问分析结果。GNN-AID是一个灵活的工具,帮助开发人员创建、分析和定制图模型,同时提供预构建的数据集和模型以快速实验。研究人员可以使用该框架探索可解释性和鲁棒性之间的高级主题,测试防御策略,并结合方法以保护 against 不同类型的攻击。我们还展示了在图数据上应用防御措施时,防御对抗和污染攻击可能会产生冲突,突显了防御策略之间的复杂联系。

英文摘要:

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}

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑