Interpreting Graph Neural Networks with Myerson Values for Cheminformatics Approaches

09 July 2024, Version 2
This content is a preprint and has not undergone peer review at the time of posting.

Abstract

Graph neural networks (GNNs) are a natural choice to represent chemical data, due to their inherent ability to handle arbitrary input topologies. They avoid the need to convert molecules into molecular fingerprints with a fixed vector length. However, like most deep learning models, GNNs are not interpretable and common explainability methods fail because of the variable input size. We introduce a novel method to interpret the predictions of GNNs based on Myerson values from cooperative game theory. Myerson values are closely related to Shapley values , which have been adapted to explain a wide variety of machine learning model predictions. Applying these approaches to GNNs have, however, proven to be challenging because of their varying graph size. Our approach treats a GNN as a coalition game and the nodes of an input layer graph as players. The Myerson value of a node then determines the contribution to the prediction of the model, with only connected nodes contributing to coalitions. All Myerson values add up to the predicted value of the model allowing for a simple and intuitive interpretation of the prediction. Because calculating Myerson values becomes computationally infeasible for large graphs, we have also implemented a scalable approximation technique using Monte Carlo sampling. We developed the technique for applications in cheminformatics and drug discovery, but it can also be used in any application that uses GNNs. The effectiveness of our approach is validated through successful applications to two proof-of-concept datasets (logP and molecular weight) as well as a real-world dataset featuring kinase inhibitors, highlighting its broad applicability and promise in explaining graph-based cheminformatic models.

Keywords

Explainable AI
Artificial Intelligence
Neural Networks
Graph Neural Networks
Game Theory

Comments

Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.