arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2303.10139cs.LGcs.AI

Distill n' Explain:使用简单代理模型解释图神经网络

Distill n' Explain: explaining graph neural networks using simple surrogates

Federal Institute of Ceará(塞阿拉联邦学院);Getulio Vargas Foundation(热图利奥·瓦加斯基金会);Aalto university(阿尔托大学)

Tamara Pereira, Erik Nascimento, Lucas E. Resck, Diego Mesquita, Amauri Souza

AI总结:

针对图神经网络解释成本与模型复杂度绑定的问题,提出基于知识蒸馏的DnX及快速版FastDnX,通过简单代理GNN和凸规划提取解释,性能优于现有方法且速度快数个数量级,并提供了理论支撑。

摘要:

解释图神经网络(GNN)的节点预测通常可归结为寻找能保留预测结果的图子结构。寻找这类结构通常需要沿GNN进行反向传播,这使得GNN的复杂度(如层数)与解释成本直接绑定。这自然引出一个问题:我们能否通过解释更简单的代理GNN来打破这种绑定?为回答该问题,我们提出Distill n' Explain(DnX)方法。首先,DnX通过知识蒸馏学习一个代理GNN;随后,DnX通过求解一个简单的凸规划提取节点级或边级的解释。我们还提出了FastDnX,这是DnX的更快版本,利用了代理模型的线性分解特性。实验表明,DnX和FastDnX的性能通常优于当前最先进的GNN解释器,同时速度快数个数量级。此外,我们还通过理论结果支撑了实证发现,该理论将代理模型的质量(即蒸馏误差)与解释的忠实度关联起来。

英文摘要:

Explaining node predictions in graph neural networks (GNNs) often boils down to finding graph substructures that preserve predictions. Finding these structures usually implies back-propagating through the GNN, bonding the complexity (e.g., number of layers) of the GNN to the cost of explaining it. This naturally begs the question: Can we break this bond by explaining a simpler surrogate GNN? To answer the question, we propose Distill n' Explain (DnX). First, DnX learns a surrogate GNN via knowledge distillation. Then, DnX extracts node or edge-level explanations by solving a simple convex program. We also propose FastDnX, a faster version of DnX that leverages the linear decomposition of our surrogate model. Experiments show that DnX and FastDnX often outperform state-of-the-art GNN explainers while being orders of magnitude faster. Additionally, we support our empirical findings with theoretical results linking the quality of the surrogate model (i.e., distillation error) to the faithfulness of explanations.

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

↑