Home » Publication » 27951

Dettaglio pubblicazione

2023, MACHINE LEARNING, Pages -

Explainable AI in drug discovery: self-interpretable graph neural network for molecular property prediction using concept whitening (01a Articolo in rivista)

Proietti Michela, Ragno Alessio, Rosa Biagio La, Ragno Rino, Capobianco Roberto

Molecular property prediction is a fundamental task in the field of drug discovery. Several works use graph neural networks to leverage molecular graph representations. Although they have been successfully applied in a variety of applications, their decision process is not transparent. In this work, we adapt concept whitening to graph neural networks. This approach is an explainability method used to build an inherently interpretable model, which allows identifying the concepts and consequently the structural parts of the molecules that are relevant for the output predictions. We test popular models on several benchmark datasets from MoleculeNet. Starting from previous work, we identify the most significant molecular properties to be used as concepts to perform classification. We show that the addition of concept whitening layers brings an improvement in both classification performance and interpretability. Finally, we provide several structural and conceptual explanations for the predictions.
keywords
© Università degli Studi di Roma "La Sapienza" - Piazzale Aldo Moro 5, 00185 Roma