Banner Banner

Peering inside the black box: Learning the relevance of many-body functions in Neural Network potentials

Klara Bonneau
Jonas Lederer
Clark Templeton
David Rosenberger
Klaus-Robert Müller
Cecilia Clementi

July 05, 2024

Machine learned potentials are becoming a popular tool to define an effec tive energy model for complex systems, either incorporating electronic structure effects at the atomistic resolution, or effectively renormalizing part of the atom istic degrees of freedom at a coarse-grained resolution. One of the main criticisms to machine learned potentials is that the energy inferred by the network is not as 1 interpretable as in more traditional approaches where a simpler functional form is used. Here we address this problem by extending tools recently proposed in the nascent field of Explainable Artificial Intelligence (XAI) to coarse-grained poten tials based on graph neural networks (GNN). We demonstrate the approach on three different coarse-grained systems including two fluids (methane and water) and the protein NTL9. On these examples, we show that the neural network potentials can be in practice decomposed in relevance contributions to differ ent orders, that can be directly interpreted and provide physical insights on the systems of interest. Keywords: Neural network potentials, molecular dynamics, coarse-graining, explainable AI