Model Validation Protocols for Machine Learning in Small Molecule Drug Discovery
Srijit SealiD, Akshat Shirish ZalteiD, David Alencar AraripeiD, Renan Augusto GomesiD, Deepa KoraniiD, Mrinal ShekhariD, Vishal B. SiramshettyiD, Arijit PatraiD, Zhongyu MouiD, Xiang Yu, Daniel KühniD, Nils WeskampiD, Jeremy R. AshiD, Alan C. ChengiD, Cheng Fang, Daniel J. Price, Matteo AldeghiiD, Raquel Rodríguez-PéreziD, Djork-Arné ClevertiD, Ola EngkvistiD, Kristine DeibleriD, David RouquiéiD, Michael ReutlingeriD, Nicola J. Richmond, Jon AinsleyiD, , Cas WognumiD
Abstract
Machine learning (ML) models for molecular property prediction are increasingly deployed in drug discovery, yet their adoption in real-world scenarios requires an understanding of the conditions in which a model succeeds or fails. While standardized benchmarks are powerful instruments to measure and unlock progress in ML research, they should not be blindly treated as the end goal. Especially static and retrospective benchmarks, in which no true unknown test set is employed, limit our ability to robustly validate a model's performance. Building on the collective expertise of a cross-industry consortium, we present a model validation framework consisting of five recommendations that would enable the community to move beyond aggregate metrics toward understanding where and why molecular property prediction models fail. We connect evaluation choices to real-world applications and case studies encountered in pharmaceutical research. The framework proposes splitting strategies that mimic realistic distribution shifts and expose common failure modes. We apply the recommended framework to a recently released dataset of absorption, distribution, metabolism, and excretion (ADME) properties. Across two complementary model algorithms, our case studies reveal four distinct failure modes (extrapolation, interpolation, representation, and evaluation), showing that model errors arise not only from distribution shift but also from limitations in molecular representations. Our results show that commonly used evaluation protocols can significantly overestimate performance and may not detect important model failure modes. All software and data are released via https://github.com/srijitseal/polaris.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 0
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Small molecule machine learning: All models are wrong, some may not even be useful
Fleming Kretschmer, Jan Seipp, … Sebastian Böcker·Nature CommunicationsPeer-reviewed·Jan 9, 2025
- Step Forward Cross Validation for Bioactivity Prediction: Out of Distribution Validation in Drug Discovery
Udit Surya Saha, Michele Vendruscolo, … Srijit Seal·bioRxiv·Jul 2, 2024
- Do Larger Models Really Win in Drug Discovery?A Benchmark Assessment of Model Scaling in AI-Driven Molecular Property and Activity Prediction
Jinjiang Guo, Sheng Ding·bioRxiv·Jun 8, 2026
- Accountable Prediction of Drug ADMET Properties with Molecular Descriptors
Nilavo Boral, Promita Ghosh, … Malay Bhattacharyya·bioRxiv·Jun 29, 2022
- Predicting FDA approvability of small-molecule drugs
Chih-Han Huang, Justine Hsu, … Ming‐Jing Hwang·bioRxiv·Oct 15, 2022
- Critical Assessment of ML models for ADMET Prediction in TDC leaderboards
Ihor Koleiev, Roman Stratiichuk, … Алан Нафієв·bioRxiv·Feb 28, 2026