Graph Representation Learning for Drug Discovery: Integrating Graph Neural Networks with Molecular Property Prediction and Explainable AI

Authors

  • Eggert Ruppert Department of Computer Science and Engineering, University at Buffalo, Buffalo, NY, USA. Author
  • Nedda Antonetti Department of Electrical Engineering and Computer Science, University of Missouri, Columbia, MO, USA. Author

Keywords:

graph representation learning; graph neural networks; molecular property prediction; explainable artificial intelligence; drug discovery; AI governance; socio-technical systems; robustness; fairness; sustainability

Abstract

Graph representation learning has become a central computational paradigm in drug discovery because molecular entities are naturally structured as graphs of atoms, bonds, and stereochemical relationships. This paper examines the integration of graph neural networks, molecular property prediction, and explainable artificial intelligence from a system-level perspective. Rather than treating predictive performance as the sole criterion, the analysis foregrounds architectural trade-offs, data infrastructure, deployment constraints, robustness, fairness, sustainability, and governance. Graph neural networks provide inductive biases suited to molecular topology, but their expressiveness, scalability, and interpretability depend on choices about message passing, pooling, pretraining, and uncertainty estimation. Explainability methods can support medicinal chemistry by identifying substructures and interaction patterns, yet explanations must be validated against domain knowledge and must remain stable under distribution shift. The paper argues that drug discovery platforms require hybrid socio-technical architectures that couple model development with assay data governance, provenance tracking, human oversight, and regulatory documentation. Cross-domain comparisons with materials discovery, protein modeling, and clinical decision support illustrate recurring tensions between accuracy, transparency, and operational efficiency. The discussion also addresses environmental costs of large-scale training, equitable access to computational resources, and policy mechanisms for accountable artificial intelligence in pharmaceutical research. The conclusion proposes that future progress depends less on isolated algorithmic novelty than on interoperable infrastructures, reproducible benchmarks, and institutional arrangements that align graph representation learning with public health goals.

References

1. Atz, K., Grisoni, F., & Schneider, G. (2021). Geometric deep learning on molecular representations. Nature Machine Intelligence, 3(12), 1023–1032. https://doi.org/10.1038/s42256-021-00418-4

2. Gilmer, J., Schoenholz, S. S., Riley, P. F., Vinyals, O., & Dahl, G. E. (2017). Neural message passing for quantum chemistry. In Proceedings of the 34th International Conference on Machine Learning (pp. 1263–1272). PMLR.

3. Duvenaud, D. K., Maclaurin, D., Iparraguirre, J., Bombarell, R., Hirzel, T., Aspuru-Guzik, A., & Adams, R. P. (2015). Convolutional networks on graphs for learning molecular fingerprints. In Advances in Neural Information Processing Systems 28 (pp. 2224–2232).

4. Kearnes, S., McCloskey, K., Berndl, M., Pande, V., & Riley, P. (2016). Molecular graph convolutions: Moving beyond fingerprints. Journal of Computer-Aided Molecular Design, 30(8), 595–608. https://doi.org/10.1007/s10822-016-9938-8

5. Sanchez-Lengeling, B., Wei, J. N., Lee, B. K., Coley, C. W., Wang, K., & Aspuru-Guzik, A. (2020). Evaluating attribution for graph neural networks. In Advances in Neural Information Processing Systems 33 (pp. 5898–5910).

6. Ying, R., Bourgeois, D., You, J., Zitnik, M., & Leskovec, J. (2019). GNNExplainer: Generating explanations for graph neural networks. In Advances in Neural Information Processing Systems 32 (pp. 9240–9251).

7. Baldassarre, F., & Azizpour, H. (2019). Explainability techniques for graph convolutional networks. arXiv preprint arXiv:1905.13686.

8. Jiménez-Luna, J., Grisoni, F., & Schneider, G. (2020). Drug discovery with explainable artificial intelligence. Nature Machine Intelligence, 2(10), 573–584. https://doi.org/10.1038/s42256-020-00236-4

9. Vamathevan, J., Clark, D., Czodrowski, P., Dunham, I., Ferran, E., Lee, G., ... & Zhao, S. (2019). Applications of machine learning in drug discovery and development. Nature Reviews Drug Discovery, 18(6), 463–477. https://doi.org/10.1038/s41573-019-0024-5

10. Ran, Z. O. U., Yang, L. I. U., Cong, L. I., Yijiao, Z. H. A. N. G., & Yanqing, H. U. (2023). Graph representation learning: A review. Journal of Beijing Normal University (Natural Science), 59(5), 716-724.

11. Chen, H., Engkvist, O., Wang, Y., Olivecrona, M., & Blaschke, T. (2018). The rise of deep learning in drug discovery. Drug Discovery Today, 23(6), 1241–1250. https://doi.org/10.1016/j.drudis.2018.01.039

12. Stokes, J. M., Yang, K., Swanson, K., Jin, W., Cubillos-Ruiz, A., Donghia, N. M., ... & Collins, J. J. (2020). A deep learning approach to antibiotic discovery. Cell, 180(4), 688–702. https://doi.org/10.1016/j.cell.2020.01.021

13. Zhavoronkov, A., Ivanenkov, Y. A., Aliper, A., Veselov, M. S., Aladinskiy, V. A., Aladinskaya, A. V., ... & Aspuru-Guzik, A. (2019). Deep learning enables rapid identification of potent DDR1 kinase inhibitors. Nature Biotechnology, 37(9), 1038–1040. https://doi.org/10.1038/s41587-019-0224-x

14. Xu, K., Hu, W., Leskovec, J., & Jegelka, S. (2019). How powerful are graph neural networks? In International Conference on Learning Representations.

15. Hamilton, W. L., Ying, R., & Leskovec, J. (2017). Inductive representation learning on large graphs. In Advances in Neural Information Processing Systems 30 (pp. 1024–1034).

16. Kipf, T. N., & Welling, M. (2017). Semi-supervised classification with graph convolutional networks. In International Conference on Learning Representations.

17. Veličković, P., Cucurull, G., Casanova, A., Romero, A., Liò, P., & Bengio, Y. (2018). Graph attention networks. In International Conference on Learning Representations.

18. Battaglia, P. W., Hamrick, J. B., Bapst, V., Sanchez-Gonzalez, A., Zambaldi, V., Malinowski, M., ... & Pascanu, R. (2018). Relational inductive biases, deep learning, and graph networks. arXiv preprint arXiv:1806.01261.

19. Rajpurkar, P., Chen, E., Banerjee, O., & Topol, E. J. (2022). AI in health and medicine. Nature Medicine, 28(1), 31–38. https://doi.org/10.1038/s41591-021-01614-0

20. European Commission. (2021). Proposal for a regulation laying down harmonised rules on artificial intelligence (Artificial Intelligence Act). https://eur-lex.europa.eu/legal-content/EN/TXT/?uri=CELEX:52021PC0206

Downloads

Published

2026-08-25

How to Cite

Graph Representation Learning for Drug Discovery: Integrating Graph Neural Networks with Molecular Property Prediction and Explainable AI. (2026). Journal of Advanced Artificial Intelligence Research, 5(1). https://www.jaair.org/index.php/home/article/view/286