Explainable artificial intelligence reveals divergent learning in pharmacophore-based hierarchical pooling graph neural networks

Hierarchical pooling is a promising mechanism to enhance graph neural networks ({GNNs}) by enabling multi-scale representation learning. Rationalization of hierarchical {GNN} predictions remains an underexplored area. In this work, we investigate the impact of hierarchical pooling on {GNNs} for molecular property prediction. We designed architectural variants integrating pharmacophore features with pooling {GNNs} at different levels. {GNN} models with pharmacophore-based graph reduction or hierarchical pooling achieved comparable compound classification performance. Explainable artificial intelligence ({XAI}) methods were applied to compare feature importance and substructure attribution for the different model architectures. Qualitative and quantitative analyses of the resulting explanations demonstrated that the {GNN} variants had different internal learning characteristics. {GNN} models based on reduced graphs matched the prediction accuracy of models based on complete graph representations following different variant-dependent learning strategies.

Citation information

Urrutia, Maria Julia Teja; Mastropietro, Andrea; Bajorath, Jürgen: Explainable artificial intelligence reveals divergent learning in pharmacophore-based hierarchical pooling graph neural networks, Scientific Reports, 2026, 16, 1, 19794, June, Nature Publishing Group, https://www.nature.com/articles/s41598-026-59947-0, Urrutia.etal.2026a,

Associated Lamarr Researchers

LAMARR Person 500x500 Bajorath 2 - Lamarr Institute for Machine Learning (ML) and Artificial Intelligence (AI)

Prof. Dr. Jürgen Bajorath

Area Chair Life Sciences & Health to the profile