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

DAG Matters! GFlowNets Enhanced Explainer For Graph Neural Networks

Huawei Noah’s Ark Lab(华为诺亚方舟实验室);Tianjin University(天津大学);National University of Singapore(新加坡国立大学)

Wenqian Li, Yinchuan Li, Zhigang Li, Jianye Hao, Yan Pang

英文摘要:

Uncovering rationales behind predictions of graph neural networks (GNNs) has received increasing attention over the years. Existing literature mainly focus on selecting a subgraph, through combinatorial optimization, to provide faithful explanations. However, the exponential size of candidate subgraphs limits the applicability of state-of-the-art methods to large-scale GNNs. We enhance on this through a different approach: by proposing a generative structure -- GFlowNets-based GNN Explainer (GFlowExplainer), we turn the optimization problem into a step-by-step generative problem. Our GFlowExplainer aims to learn a policy that generates a distribution of subgraphs for which the probability of a subgraph is proportional to its' reward. The proposed approach eliminates the influence of node sequence and thus does not need any pre-training strategies. We also propose a new cut vertex matrix to efficiently explore parent states for GFlowNets structure, thus making our approach applicable in a large-scale setting. We conduct extensive experiments on both synthetic and real datasets, and both qualitative and quantitative results show the superiority of our GFlowExplainer.

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

↑