External validation of machine learning models - registered models and adaptive sample splitting
Giuseppe GallittoiD, Robert EnglertiD, Bálint KincsesiD, Raviteja KotikalapudiiD, Jialin LiiD, Kevin HoffschlagiD, Ulrike BingeliD, Tamás SpisákiD
Abstract
Multivariate predictive models play a crucial role in enhancing our understanding of complex biological systems and in developing innovative, replicable tools for translational medical research. However, the complexity of machine learning methods and extensive data pre-processing and feature engineering pipelines can lead to overfitting and poor generalizability. An unbiased evaluation of predictive models necessitates external validation, which involves testing the finalized model on independent data. Despite its importance, external validation is often neglected in practice due to the associated costs. Here we propose that, for maximal credibility, model discovery and external validation should be separated by the public disclosure (e.g. pre-registration) of feature processing steps and model weights. Furthermore, we introduce a novel approach to optimize the trade-off between efforts spent on training and external validation in such studies. We show on data involving more than 3000 participants from four different datasets that, for any "sample size budget", the proposed adaptive splitting approach can successfully identify the optimal time to stop model discovery so that predictive performance is maximized without risking a low powered, and thus inconclusive, external validation. The proposed design and splitting approach (implemented in the Python package "AdaptiveSplit") may contribute to addressing issues of replicability, effect size inflation and generalizability in predictive modeling studies.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 45
- FWCI
- 37.24
- Field percentile
- top 1%
- 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
- Model Validation Protocols for Machine Learning in Small Molecule Drug Discovery
Srijit Seal, Akshat Shirish Zalte, … Cas Wognum·bioRxiv·Aug 24, 2026
- Impacts of batch effects on the performance of machine learning classifiers across multiple studies
Preston Raab, William Evan Johnson, Stephen Piccolo·bioRxiv·Jun 30, 2026
- Beyond P-values: A Multi-Metric Framework for Robust Feature Selection and Predictive Modeling
Raelynn Chen, Attri Ghosh, … Ruowang Li·bioRxiv·Oct 5, 2025
- Controlling for effects of confounding variables on machine learning predictions
Richard Dinga, Lianne Schmaal, … André F. Marquand·bioRxiv·Aug 17, 2020
- Correcting machine learning models using calibrated ensembles with "mlensemble"
Tomasz Konopka·bioRxiv·Jul 26, 2021
- Power and reproducibility in the external validation of brain-phenotype predictions
Matthew Rosenblatt, Link Tejavibulya, … Dustin Scheinost·Nature Human BehaviourPeer-reviewed·Jul 31, 2024