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

使用 GNN 的差分隐私图分类

Differentially Private Graph Classification with GNNs

Technical University of Munich(慕尼黑工业大学);Helmholtz Zentrum München(慕尼黑亥姆霍兹中心);University of Zurich(苏黎世大学);Imperial College London(伦敦帝国理工学院)

Tamara T. Mueller, Johannes C. Paetzold, Chinmay Prabhakar, Dmitrii Usynin, Daniel Rueckert, Georgios Kaissis

AI总结:

针对图结构带来的差分隐私训练挑战,本文基于 DP-SGD 实现多图数据集上的隐私图级分类,并系统评估架构、超参数与可解释性,建立稳健基线。

摘要:

图神经网络(GNN)已确立其作为许多机器学习应用的最先进模型的地位,例如社交网络、蛋白质相互作用和分子分析。这些数据集中有若干包含隐私敏感数据。采用差分隐私的机器学习是一种很有前景的技术,能够在提供隐私保护形式化保证的同时,从敏感数据中获得洞见。然而,由于图内在结构连通性所带来的挑战,GNN 的差分隐私训练迄今仍未得到充分研究。在本工作中,我们将差分隐私引入图级分类,这是图上机器学习的关键应用之一。我们的方法适用于多图数据集上的深度学习,并依赖差分隐私随机梯度下降(DP-SGD)。我们在多种合成数据集和公开数据集上展示了结果,并评估了不同 GNN 架构和训练超参数对差分隐私图分类模型性能的影响。最后,我们应用可解释性技术来评估在隐私和非隐私设置下是否学习到相似的表示,并为该领域未来的工作建立稳健的基线。

英文摘要:

Graph Neural Networks (GNNs) have established themselves as the state-of-the-art models for many machine learning applications such as the analysis of social networks, protein interactions and molecules. Several among these datasets contain privacy-sensitive data. Machine learning with differential privacy is a promising technique to allow deriving insight from sensitive data while offering formal guarantees of privacy protection. However, the differentially private training of GNNs has so far remained under-explored due to the challenges presented by the intrinsic structural connectivity of graphs. In this work, we introduce differential privacy for graph-level classification, one of the key applications of machine learning on graphs. Our method is applicable to deep learning on multi-graph datasets and relies on differentially private stochastic gradient descent (DP-SGD). We show results on a variety of synthetic and public datasets and evaluate the impact of different GNN architectures and training hyperparameters on model performance for differentially private graph classification. Finally, we apply explainability techniques to assess whether similar representations are learned in the private and non-private settings and establish robust baselines for future work in this area.

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

↑