Efficient Automatic CASH via Rising Bandits
Yang Li, Jiawei Jiang, Jinyang Gao, Yingxia Shao, Ce Zhang, Bin Cui
Abstract
The Combined Algorithm Selection and Hyperparameter optimization (CASH) is one of the most fundamental problems in Automatic Machine Learning (AutoML). The existing Bayesian optimization (BO) based solutions turn the CASH problem into a Hyperparameter Optimization (HPO) problem by combining the hyperparameters of all machine learning (ML) algorithms, and use BO methods to solve it. As a result, these methods suffer from the low-efficiency problem due to the huge hyperparameter space in CASH. To alleviate this issue, we propose the alternating optimization framework, where the HPO problem for each ML algorithm and the algorithm selection problem are optimized alternately. In this framework, the BO methods are used to solve the HPO problem for each ML algorithm separately, incorporating a much smaller hyperparameter space for BO methods. Furthermore, we introduce Rising Bandits, a CASH-oriented Multi-Armed Bandits (MAB) variant, to model the algorithm selection in CASH. This framework can take the advantages of both BO in solving the HPO problem with a relatively small hyperparameter space and the MABs in accelerating the algorithm selection. Moreover, we further develop an efficient online algorithm to solve the Rising Bandits with provably theoretical guarantees. The extensive experiments on 30 OpenML datasets demonstrate the superiority of the proposed approach over the competitive baselines.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext bd888b3a-3cd7-4bd7-a1de-10a60e637b0aCited by top-tier papers15
- Facilitating Database Tuning with Hyper-Parameter Optimization: A Comprehensive Experimental EvaluationXinyi Zhang, Zhuo Chang, Yang Li, Hong Wu et al.VLDB 2022 · 88 citations
- VolcanoML: Speeding up End-to-End AutoML via Scalable Search Space DecompositionYang Li, Yu Shen, Wentao Zhang, Jiawei Jiang et al.VLDB 2021 · 55 citations
- Which LLM to Play? Convergence-Aware Online Model Selection with Time-Increasing BanditsYu Xia, Fang Kong, Tong Yu, Liya Guo et al.WWW 2024 · 33 citations
- Hyper-Tune: Towards Efficient Hyper-parameter Tuning at ScaleYang Li, Yu Shen, Huaijun Jiang, Wentao Zhang et al.VLDB 2022 · 32 citations
- A Unified and Efficient Coordinating Framework for Autonomous DBMS TuningXinyi Zhang, Zhuo Chang, Hong Wu, Yang Li et al.SIGMOD 2023 · 17 citations
Related papers
- Put CASH on Bandits: A Max K-Armed Problem for Automated Machine LearningAmir Rezaei Balef, Claire Vernade, Katharina EggenspergerNeurIPS 2025 · 4 citations
- PSEO: Optimizing Post-hoc Stacking Ensemble Through Hyperparameter TuningBeicheng Xu, Wei Liu, Keyao Ding, Yupeng Lu et al.AAAI 2026 · 2 citations
- CASH via Optimal Diversity for Ensemble LearningPranav Poduval, Sanjay Kumar Patnala, Gaurav Oberoi, Nitish Srivasatava et al.KDD 2024 · 2 citations
- DivBO: Diversity-aware CASH for Ensemble LearningYu Shen, Yupeng Lu, Yang Li, Yaofeng Tu et al.NeurIPS 2022 · 15 citations
- Weighted Sampling for Combined Model Selection and Hyperparameter TuningDimitrios Sarigiannis, Thomas P. Parnell, Haralampos PozidisAAAI 2020 · 3 citations
