mlsanityData to 7 Oct 2026

Don't Waste Your Time: Early Stopping Cross-Validation

arXiv 2405.03389 ↗First posted 6 May 2024cs.LG cs.AIcited by 6

Not on the map. Why?

Abstract (arXiv)

State-of-the-art automated machine learning systems for tabular data often employ cross-validation; ensuring that measured performances generalize to unseen data, or that subsequent ensembling does not overfit. However, using k-fold cross-validation instead of holdout validation drastically increases the computational cost of validating a single configuration. While ensuring better generalization and, by extension, better performance, the additional cost is often prohibitive for effective model selection within a time budget. We aim to make model selection with cross-validation more effective. Therefore, we study early stopping the process of cross-validation during model selection. We investigate the impact of early stopping on random search for two algorithms, MLP and random forest, across 36 classification datasets. We further analyze the impact of the number of folds by considering 3-, 5-, and 10-folds. In addition, we investigate the impact of early stopping with Bayesian optimization instead of random search and also repeated cross-validation. Our exploratory study shows that even a simple-to-understand and easy-to-implement method consistently allows model selection to converge faster; in ~94% of all datasets, on average by ~214%. Moreover, stopping cross-validation enables model selection to explore the search space more exhaustively by considering +167% configurations on average within one hour, while also obtaining better overall performance.

Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 17 of its 48 references that are in the map. First posted on arXiv 6 May 2024.

Builds on

17 of 48 references.

References · 17

  1. 2016Hyperband: A Novel Bandit-Based Approach to Hyperparameter Optimizationcited by 688
  2. 2014OpenML: networked science in machine learningcited by 523
  3. 2018BOHB: Robust and Efficient Hyperparameter Optimization at Scalecited by 466
  4. 2020AutoGluon-Tabular: Robust and Accurate AutoML for Structured Datacited by 368
  5. 2021SMAC3: A Versatile Bayesian Optimization Package for Hyperparameter Optimizationcited by 229
  6. 2015Non-stochastic Best Arm Identification and Hyperparameter Optimizationcited by 219
  7. 2020Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learningcited by 174
  8. 2014ParamILS: An Automatic Algorithm Configuration Frameworkcited by 88
  9. 2019An Open Source AutoML Benchmarkcited by 85
  10. 2019OpenML-Python: an extensible Python API for OpenMLcited by 62
  11. 2022AMLB: an AutoML Benchmarkcited by 55
  12. 2021LightAutoML: AutoML Solution for a Large Financial Services Ecosystemcited by 32
5 more
  1. 2023TabRepo: A Large Scale Repository of Tabular Model Evaluations and its AutoML Applicationscited by 32
  2. 2023PriorBand: Practical Hyperparameter Optimization in the Age of Deep Learningcited by 31
  3. 2021Multi-objective Asynchronous Successive Halvingcited by 22
  4. 2020GAMA: a General Automated Machine learning Assistantcited by 15
  5. 2021Fast and Informative Model Selection using Learning Curve Cross-Validationcited by 15

Built on by

Cited by year

Citing papers per year: 2024: 3; 2025: 3.

6 papers.