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

XInsight:通过基于流的解释揭示图神经网络(GNNs)的模型洞察

XInsight: Revealing Model Insights for GNNs with Flow-based Explanations

Southern Methodist University(南卫理公会大学)

Eli Laird, Ayesh Madushanka, Elfi Kraka, Corey Clark

AI总结:

针对图神经网络的黑箱问题,提出基于GFlowNets的可解释算法XInsight,可生成多样化的模型解释分布,在两项图分类任务中验证了其揭示模型底层关系、辅助发现隐藏规律的价值。

摘要:

近年来,图神经网络(GNNs)领域进展迅速,在药物发现、医疗诊断和推荐系统中涌现出大量新成果。尽管这一进展意义重大,但许多网络仍属于“黑箱”,人们对网络究竟在“学习什么”知之甚少。许多高风险应用(例如药物发现)需要模型提供人类可理解的解释,以便用户识别错误并发现新知识。因此,可解释人工智能算法的开发对于我们充分发挥AI的价值至关重要。\n我们提出了一种面向GNN的可解释性算法,名为可解释洞察(eXplainable Insight,XInsight),它利用生成流网络(GFlowNets)生成模型解释的分布。由于GFlowNets生成对象的概率与奖励成正比,与仅学习最大奖励样本的现有方法相比,XInsight能够生成多样化的解释集合。我们通过为在两项图分类任务上训练的GNN生成解释,对XInsight进行了验证:一项是使用MUTAG数据集对致突变化合物进行分类,另一项是使用我们开源的合成数据集对无环图进行分类。我们通过QSAR建模分析生成的化合物,展示了XInsight解释的实用性,结果发现XInsight生成的化合物会按亲脂性(一种已知的致突变性相关因素)聚类。我们的结果表明,XInsight生成的解释分布能够揭示模型所展现的底层关系。研究结果还凸显了生成多样化解释集合的重要性,因为这让我们能够发现模型中的隐藏关系,并为后续分析提供有价值的指导。

英文摘要:

Progress in graph neural networks has grown rapidly in recent years, with many new developments in drug discovery, medical diagnosis, and recommender systems. While this progress is significant, many networks are `black boxes' with little understanding of the `what' exactly the network is learning. Many high-stakes applications, such as drug discovery, require human-intelligible explanations from the models so that users can recognize errors and discover new knowledge. Therefore, the development of explainable AI algorithms is essential for us to reap the benefits of AI. We propose an explainability algorithm for GNNs called eXplainable Insight (XInsight) that generates a distribution of model explanations using GFlowNets. Since GFlowNets generate objects with probabilities proportional to a reward, XInsight can generate a diverse set of explanations, compared to previous methods that only learn the maximum reward sample. We demonstrate XInsight by generating explanations for GNNs trained on two graph classification tasks: classifying mutagenic compounds with the MUTAG dataset and classifying acyclic graphs with a synthetic dataset that we have open-sourced. We show the utility of XInsight's explanations by analyzing the generated compounds using QSAR modeling, and we find that XInsight generates compounds that cluster by lipophilicity, a known correlate of mutagenicity. Our results show that XInsight generates a distribution of explanations that uncovers the underlying relationships demonstrated by the model. They also highlight the importance of generating a diverse set of explanations, as it enables us to discover hidden relationships in the model and provides valuable guidance for further analysis.

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

↑