Is K-fold cross validation the best model selection method for Machine Learning?
Not on the map. Why?
Abstract (arXiv)
As a technique that can compactly represent complex patterns, machine learning has significant potential for predictive inference. K-fold cross-validation (CV) is the most common approach to ascertaining the likelihood that a machine learning outcome is generated by chance, and it frequently outperforms conventional hypothesis testing. This improvement uses measures directly obtained from machine learning classifications, such as accuracy, that do not have a parametric description. To approach a frequentist analysis within machine learning pipelines, a permutation test or simple statistics from data partitions (i.e., folds) can be added to estimate confidence intervals. Unfortunately, neither parametric nor non-parametric tests solve the inherent problems of partitioning small sample-size datasets and learning from heterogeneous data sources. The fact that machine learning strongly depends on the learning parameters and the distribution of data across folds recapitulates familiar difficulties around excess false positives and replication. A novel statistical test based on K-fold CV and the Upper Bound of the actual risk (K-fold CUBV) is proposed, where uncertain predictions of machine learning with CV are bounded by the worst case through the evaluation of concentration inequalities. Probably Approximately Correct-Bayesian upper bounds for linear classifiers in combination with K-fold CV are derived and used to estimate the actual risk. The performance with simulated and neuroimaging datasets suggests that K-fold CUBV is a robust criterion for detecting effects and validating accuracy values obtained from machine learning and classical CV schemes, while avoiding excess false positives.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Builds on 10 of its 63 references that are in the map. First posted on arXiv 29 Jan 2024.
Builds on
10 of 63 references.
References · 10
- 2016Model-Agnostic Interpretability of Machine Learningcited by 219
- 2021Cross-validation: what does it estimate and how well does it do it?cited by 65
- 2013A PAC-Bayesian Tutorial with A Dropout Boundcited by 43
- 2017Cross-validation failure: small sample sizes lead to large error barscited by 43
- 2015Can parametric statistical methods be trusted for fMRI based group studies?cited by 22
- 2016Better-Than-Chance Classification for Signal Detectioncited by 8
- 2020Ensemble Deep Learning on Large, Mixed-Site fMRI Datasets in Autism and Other Taskscited by 7
- 2019Statistical Agnostic Mapping: a Framework in Neuroimaging based on Concentration Inequalitiescited by 5
- 2017The Same Analysis Approach: Practical protection against the pitfalls of novel neuroimaging analysis methodscited by 4
- 2022A hypothesis-driven method based on machine learning for neuroimaging data analysiscited by 2
Built on by
Citing papers per year: 2024: 1; 2025: 8; 2026: 1.
10 papers.
- 2025Perturbing the Derivative: Doubly Wild Refitting for Model-Free Evaluation of Opaque Machine Learning Predictorscited by 2
- 2024Statistical Agnostic Regression: a machine learning method to validate regression modelscited by 0
- 2025Uncertainty-Driven Modeling of Microporosity and Permeability in Clastic Reservoirs Using Random Forestcited by 0
- 2025The use of cross validation in the analysis of designed experimentscited by 0
- 2025Harnessing Near-Infrared Spectroscopy and Machine Learning for Traceable Classification of Hanwoo and Holstein Beefcited by 0
- 2025Jet Image Tagging Using Deep Learning: An Ensemble Modelcited by 0
- 2025Deep Learning Framework for Early Detection of Pancreatic Cancer Using Multi-Modal Medical Imaging Analysiscited by 0
- 2025Perturbing the Derivative: Wild Refitting for Model-Free Evaluation of Machine Learning Models under Bregman Lossescited by 0
- 2025A machine learning framework integrating seed traits and plasma parameters for predicting germination uplift in cropscited by 0
- 2026Enhancing Language Models for Robust Greenwashing Detectioncited by 0
- 2026Enhancing Language Models for Robust Greenwashing Detection29 Jan 2026
- 2025Perturbing the Derivative: Doubly Wild Refitting for Model-Free Evaluation of Opaque Machine Learning Predictors24 Nov 2025
- 2025A machine learning framework integrating seed traits and plasma parameters for predicting germination uplift in crops26 Oct 2025
- 2025Perturbing the Derivative: Wild Refitting for Model-Free Evaluation of Machine Learning Models under Bregman Losses2 Sep 2025
- 2025Deep Learning Framework for Early Detection of Pancreatic Cancer Using Multi-Modal Medical Imaging Analysis28 Aug 2025
- 2025Jet Image Tagging Using Deep Learning: An Ensemble Model9 Aug 2025
- 2025Harnessing Near-Infrared Spectroscopy and Machine Learning for Traceable Classification of Hanwoo and Holstein Beef24 Jun 2025
- 2025The use of cross validation in the analysis of designed experiments17 Jun 2025
- 2025Uncertainty-Driven Modeling of Microporosity and Permeability in Clastic Reservoirs Using Random Forest21 Mar 2025
- 2024Statistical Agnostic Regression: a machine learning method to validate regression models23 Feb 2024