A new pipeline for cross-validation fold-aware machine learning prediction of clinical outcomes addresses hidden data-leakage in omics based 'predictors'.
Abstract
MotivationMachine learning (ML) approaches are increasingly applied to high-dimensional biological data in which features are often dataset-dependent. In many omics workflows, features are computed using information derived from the entire dataset, such as correlations between variables, clustering structures, or enrichment scores. We refer to these as global dataset features, defined as features whose computation depends on properties of the full dataset. In such cases, standard validation strategies can fail, especially when evaluating on independent datasets, due to information leakage that leads to overly optimistic performance estimates. ResultsTo address this challenge, we present pipeML, a flexible and modular machine learning framework designed to support leakage-free model training through custom cross-validation (CV) fold construction. pipeML enables users to recompute global dataset features independently within each CV fold, ensuring strict separation between training and test data, while preserving compatibility with a wide range of ML algorithms for both classification and survival tasks. Using real-world biological datasets, we demonstrate that pipeML enables leakage-free model evaluation when global dataset features are used. We argue that overestimation of model performance during CV can lead to overoptimistic expectations for validation on independent datasets. By explicitly addressing data leakage and offering a transparent, modular workflow, pipeML provides a robust solution for developing and validating ML models in complex biological settings. AvailabilityThe pipeML R package as well as a tutorial are available at https://github.com/VeraPancaldiLab/pipeML Contactvera.pancaldi@inserm.fr or marcelo.hurtado@inserm.fr Supplementary informationAvailable at Bioinformatics online.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 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
- The effects of data leakage on connectome-based machine learning models
Matthew Rosenblatt, Link Tejavibulya, … Dustin Scheinost·Nature CommunicationsPeer-reviewed·Feb 28, 2024
- A consistent evaluation of miRNA-disease association prediction models
Thi Ngan Dong, Megha Khosla·bioRxiv·May 4, 2020
- Widespread data leakage inflates accuracy and corrupts biomarker discovery in cancer drug response prediction
Amir Asiaee, Jared Strauch, … Kevin R. Coombes·bioRxiv·Apr 5, 2026
- PPML-Omics: a Privacy-Preserving federated Machine Learning method protects patients' privacy in omic data
Juexiao Zhou, Siyuan Chen, … Xin Gao·Science AdvancesPeer-reviewed·Jan 31, 2024
- DataSAIL: Data Splitting Against Information Leakage
Roman Joeres, David B. Blumenthal, Olga V. Kalinina·bioRxiv·Nov 15, 2023
- External validation of machine learning models - registered models and adaptive sample splitting
Giuseppe Gallitto, Robert Englert, … Tamás Spisák·GigaSciencePeer-reviewed·Jan 1, 2025