Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning

Matthias FeurerKatharina EggenspergerStefan FalknerMarius LindauerFrank Hutter

article2022JMLR430 citations

Presents Auto-Sklearn 2.0, a truly hands-free automated machine learning framework that uses meta-learning and bandit-based budget allocation to automatically configure its own optimization strategy, outperforming its predecessor in ten minutes compared to an hour of compute time across 39 benchmark datasets.

Listen

Automated Machine Learning systems help developers and non-specialists build predictive pipelines without requiring manual trial and error. However, existing tools face major operational hurdles: evaluating complex models on large datasets often exceeds strict time and memory limits, and the systems themselves introduce difficult high-level configuration choices, such as selecting validation methods or resource allocation strategies. The article evaluates a new generation of automated tools—culminating in the development of Auto-sklearn 2.0—to determine how to deliver high predictive accuracy rapidly under rigid computational constraints without requiring human intervention.

The authors designed and evaluated their framework using extensive empirical benchmarks. They trained their meta-learning components across 208 diverse open-source classification datasets and tested performance on an independent benchmark suite of 39 standard datasets. The approach introduces pre-computed portfolios of diverse, high-performing pipelines to eliminate the need for expensive real-time meta-feature calculations. It integrates a successive halving allocation strategy that early-stops poorly performing pipelines to allocate more compute time to promising models. To make the entire framework hands-free, the researchers built an automated policy selector that uses simple, linear-time metadata (the number of rows and columns) along with a fallback safety rule to pick the optimal validation and budget allocation strategy for any unseen dataset.

The experimental findings show substantial improvements in efficiency, accuracy, and robustness. Under strict 10-minute compute limits, Auto-sklearn 2.0 achieved a lower prediction error than its predecessor, Auto-sklearn 1.0, achieved after 60 minutes of computation. Against Auto-sklearn 1.0, the new framework reduced relative error by a factor of 4.5 at the 10-minute horizon and by a factor of 3 at the 60-minute mark. On large, complex datasets where previous tools frequently failed due to memory and time limits, the new pipeline retrieval and allocation mechanisms completed execution reliably. In head-to-head comparisons against other established automated machine learning platforms across 39 benchmark datasets, Auto-sklearn 2.0 achieved the lowest average rank and won the highest number of individual datasets.

These results demonstrate that automating higher-level system design decisions eliminates common operational bottlenecks and significantly lowers the compute cost and timeline required for rapid prototyping. Practitioners can obtain higher-quality models in a fraction of the time, reducing cloud hardware expenses and democratizing advanced predictive modeling for non-expert teams. The findings also reveal that complex dataset descriptions are unnecessary; simple metadata paired with disciplined warmstart portfolios is sufficient to guide algorithm selection effectively.

Organizations deploying automated machine learning should adopt portfolio warmstarts and multi-fidelity budget allocation to handle tight operational deadlines. For automated system configuration, the evidence supports using per-instance policy selection combined with aggressive fallback mechanisms to prevent timeouts on out-of-distribution datasets. Future initiatives should focus on extending these automated selection policies across varying compute budgets and metrics, dynamically adjusting portfolio sizes, and exploring adaptive fallbacks during execution.

The findings are supported by strong statistical testing and multi-dataset cross-validation; however, certain boundaries remain. The evaluation focused primarily on classification tasks with fixed tabular constraints, and the system relies on an offline training phase across historical datasets to construct its portfolios and selector models. Readers should exercise caution when applying these configurations directly to regression tasks, complex multi-modal inputs, or environments where the target evaluation metrics differ substantially from those modeled in the offline phase.

Cover for Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning

Abstract

Automated Machine Learning (AutoML) supports practitioners and researchers with the tedious task of designing machine learning pipelines and has recently achieved substantial success. In this paper, we introduce new AutoML approaches motivated by our winning submission to the second ChaLearn AutoML challenge. We develop PoSH Auto-sklearn, which enables AutoML systems to work well on large datasets under rigid time limits by using a new, simple and meta-feature-free meta-learning technique and by employing a successful bandit strategy for budget allocation. However, PoSH Auto-sklearn introduces even more ways of running AutoML and might make it harder for users to set it up correctly. Therefore, we also go one step further and study the design space of AutoML itself, proposing a solution towards truly hands-free AutoML. Together, these changes give rise to the next generation of our AutoML system, Auto-sklearn 2.0. We verify the improvements by these additions in an extensive experimental study on 39 AutoML benchmark datasets. We conclude the paper by comparing to other popular AutoML frameworks and Auto-sklearn 1.0, reducing the relative error by up to a factor of 4.5, and yielding a performance in 10 minutes that is substantially better than what Auto-sklearn 1.0 achieves within an hour.

Table of Contents

  • 1. Introduction
  • 2. Problem Statement
  • 2.1 Time-bounded AutoML
  • 2.2 Generalization of AutoML
  • 3. Part I: Portfolio Successive Halving in PoSH Auto-sklearn
  • 3.1 Portfolio Building
  • 3.1.1 Approach
  • 3.1.2 Theoretical Properties of the Greedy Algorithm
  • 3.2 Budget Allocation using Successive Halving
  • 3.2.1 Approach
  • 3.3 Practical Considerations and Challenge Results
  • 3.4 Experimental Setup
  • 3.4.1 Datasets
  • 3.4.2 Meta-data Generation
  • 3.4.3 Other Experimental Details
  • 3.5 Experimental Results
  • 3.5.1 Portfolio vs. KND
  • 3.5.2 PoSH Auto-sklearn vs Auto-sklearn 1.0
  • 4. Part II: Automating Design Decisions in AutoML
  • 4.1 Automated Policy Selection
  • 4.1.1 Approach
  • 4.2 Experimental Results
  • 4.3 Ablation
  • 4.3.1 Do we need per-dataset selection?
  • 4.3.2 Do we need different model selection strategies?
  • 4.3.3 Do we still need to warm-start Bayesian optimization?
  • 5. Comparison to other AutoML systems
  • 5.1 Integration and setup
  • 5.2 Results
  • 6. Related Work
  • 6.1 Related Work on Portfolios
  • 6.2 Related Work on Successive Halving
  • 6.3 Related Work on Algorithm Selection
  • 6.4 Background on AutoML Systems and Their Components
  • 6.4.1 Components of AutoML systems
  • 6.4.2 AutoML systems
  • 7. Discussion and Conclusion
  • Acknowledgments
  • Appendix A. Additional pseudo-code
  • Appendix B. Additional results and experiments
  • B.1 Early Stopping and Retrieving Intermittent Results
  • B.2 Performance Without Post-Hoc Ensembling
  • B.3 Unaggregated results
  • Appendix C. Theoretical properties of the greedy algorithm
  • C.1 Definitions
  • C.2 Choosing on the test set
  • C.3 Choosing on the validation set
  • C.4 Successive Halving
  • C.5 Further equalities
  • Appendix D. Implementation Details
  • D.1 Software
  • D.2 Configuration Space
  • D.3 Successive Halving hyperparameters
  • Appendix E. Datasets
  • References

Knowls

  1. Knowl 1 — No paper content was available for extraction

    limitation

    The attached file did not expose any readable text of the paper, so no method, model, theorem, algorithm, experiment, or result from the paper could be identified or summarized. This entry is a placeholder stating that absence; it does not report any finding from the paper itself.

Coverage note — No knowls could be extracted: the attached file's content was not accessible to me, so I cannot identify the paper's methods, results, or other contributions without fabricating them. No contributed material was deliberately omitted; rather, none could be read at all. To obtain valid knowls, please re-attach the paper with its text content included.

Citation

MLA
Feurer, M., et al. “Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning”. Journal of Machine Learning Research, vol. 23, no. 261, 2022, pp. 1–1, https://www.jmlr.org/papers/v23/21-0992.html.
APA
Feurer, M., Eggensperger, K., Falkner, S., Lindauer, M., & Hutter, F. (2022). Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning. Journal of Machine Learning Research, 23(261), 1–61. https://www.jmlr.org/papers/v23/21-0992.html
Chicago
Feurer, M., K. Eggensperger, S. Falkner, M. Lindauer, and F. Hutter. 2022. “Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning”. Journal of Machine Learning Research 23 (261): 1–61. https://www.jmlr.org/papers/v23/21-0992.html.
Harvard
Feurer, M. et al. (2022) “Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning”, Journal of Machine Learning Research, 23(261), pp. 1–61. Available at: https://www.jmlr.org/papers/v23/21-0992.html.
Vancouver
1. Feurer M, Eggensperger K, Falkner S, Lindauer M, Hutter F (2022) Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning. Journal of Machine Learning Research 23:1–61

BibTeX

@article{JMLR:v23:21-0992,
  author  = {Matthias Feurer and Katharina Eggensperger and Stefan Falkner and Marius Lindauer and Frank Hutter},
  title   = {Auto-Sklearn 2.0: Hands-free AutoML via Meta-Learning},
  journal = {Journal of Machine Learning Research},
  year    = {2022},
  volume  = {23},
  number  = {261},
  pages   = {1--61},
  url     = {http://jmlr.org/papers/v23/21-0992.html}
}
Metadata:DOI registry

Source Code

This paper has an official code repository available. Click below to access the source code.

View Repository

Access the Paper

This paper is available from its original source. Click below to access the PDF.

Open PDF
License: https://creativecommons.org/licenses/by/4.0/