mlsanityData to 14 Oct 2026

Is K-fold cross validation the best model selection method for Machine Learning?

arXiv 2401.16407 ↗First posted 29 Jan 2024stat.ML cs.LG eess.IVcited by 10

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

  1. 2016Model-Agnostic Interpretability of Machine Learningcited by 219
  2. 2021Cross-validation: what does it estimate and how well does it do it?cited by 65
  3. 2013A PAC-Bayesian Tutorial with A Dropout Boundcited by 43
  4. 2017Cross-validation failure: small sample sizes lead to large error barscited by 43
  5. 2015Can parametric statistical methods be trusted for fMRI based group studies?cited by 22
  6. 2016Better-Than-Chance Classification for Signal Detectioncited by 8
  7. 2020Ensemble Deep Learning on Large, Mixed-Site fMRI Datasets in Autism and Other Taskscited by 7
  8. 2019Statistical Agnostic Mapping: a Framework in Neuroimaging based on Concentration Inequalitiescited by 5
  9. 2017The Same Analysis Approach: Practical protection against the pitfalls of novel neuroimaging analysis methodscited by 4
  10. 2022A hypothesis-driven method based on machine learning for neuroimaging data analysiscited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 1; 2025: 8; 2026: 1.

10 papers.