arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.03093cs.LGcs.CRcs.SI

归纳图遗忘

Inductive Graph Unlearning

King Abdullah University of Science and Technology(阿卜杜拉国王科技大学);SDAIA-KAUST Center of Excellence in Data Science and Artificial Intelligence(沙特数据与人工智能局-阿卜杜拉国王科技大学数据科学与人工智能卓越中心);Iowa State University(爱荷华州立大学);Computational Bioscience Research Center(计算生物科学研究中心)

Cheng-Long Wang, Mengdi Huai, Di Wang

AI总结:

针对现有图遗忘方法仅适用于静态直推式设置的问题,提出引导式归纳图遗忘框架GUIDE,通过引导图划分、子图修复和相似度聚合实现动态图的高效遗忘。

摘要:

作为在机器学习中实现“被遗忘权”的一种方式,机器遗忘旨在完全移除待删除样本对已训练模型的贡献和信息,而不影响其他样本的贡献。近期,已提出许多机器遗忘框架,其中大多数聚焦于图像和文本数据。为了将机器遗忘扩展到图数据,人们提出了GraphEraser。然而,一个关键问题是GraphEraser专门为直推式图设置设计,其中图是静态的,且测试节点的属性和边在训练期间是可见的。它不适用于归纳设置,在归纳设置中图可能是动态的,且测试图信息事先不可见。这种归纳能力对于具有不断演化的图(如社交媒体和交易网络)的生产机器学习系统至关重要。为了填补这一空白,我们提出了引导式归纳图遗忘框架(GUIDE)。GUIDE由三个组件组成:具有公平性和平衡性的引导图划分、高效的子图修复以及基于相似度的聚合。在实验上,我们在几个归纳基准和不断演化的交易图上评估了我们的方法。总体而言,GUIDE可以高效地应用于归纳图学习任务,因为其图划分成本低,无论是在计算还是结构信息上。代码将在此处提供:https://github.com/Happy2Git/GUIDE。

英文摘要:

As a way to implement the "right to be forgotten" in machine learning, \textit{machine unlearning} aims to completely remove the contributions and information of the samples to be deleted from a trained model without affecting the contributions of other samples. Recently, many frameworks for machine unlearning have been proposed, and most of them focus on image and text data. To extend machine unlearning to graph data, \textit{GraphEraser} has been proposed. However, a critical issue is that \textit{GraphEraser} is specifically designed for the transductive graph setting, where the graph is static and attributes and edges of test nodes are visible during training. It is unsuitable for the inductive setting, where the graph could be dynamic and the test graph information is invisible in advance. Such inductive capability is essential for production machine learning systems with evolving graphs like social media and transaction networks. To fill this gap, we propose the \underline{\bf G}\underline{\bf U}ided \underline{\bf I}n\underline{\bf D}uctiv\underline{\bf E} Graph Unlearning framework (GUIDE). GUIDE consists of three components: guided graph partitioning with fairness and balance, efficient subgraph repair, and similarity-based aggregation. Empirically, we evaluate our method on several inductive benchmarks and evolving transaction graphs. Generally speaking, GUIDE can be efficiently implemented on the inductive graph learning tasks for its low graph partition cost, no matter on computation or structure information. The code will be available here: https://github.com/Happy2Git/GUIDE.

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

↑