Proceedings

Best Arm Identification with Retroactively Increased Sampling Budget for More Resource-Efficient HPO

Download BibTeX Cite

Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 3742-3750.

https://doi.org/10.24963/ijcai.2024/414

Abstract

Hyperparameter optimization (HPO) is indispensable for achieving optimal performance in machine learning tasks. A popular class of methods in this regard is based on Successive Halving (SHA), which casts HPO into a pure-exploration multi-armed bandit problem under finite sampling budget constraints. This is accomplished by considering hyperparameter configurations as arms and rewards as the negative validation losses. While enjoying theoretical guarantees as well as working well in practice, SHA comes, however, with several hyperparameters itself, one of which is the maximum budget that can be allocated to evaluate a single arm (hyperparameter configuration). Although there are already solutions to this meta hyperparameter optimization problem, such as the doubling trick or asynchronous extensions of SHA, these are either practically inefficient or lack theoretical guarantees. In this paper, we propose incremental SHA (iSHA), a synchronous extension of SHA, allowing to increase the maximum budget a posteriori while still enjoying theoretical guarantees. Our empirical analysis of HPO problems corroborates our theoretical findings and shows that iSHA is more resource-efficient than existing SHA-based approaches.

  1. [Abbasi-Yadkori et al., 2018] Yasin Abbasi-Yadkori, Peter Bartlett, Victor Gabillon, Alan Malek, and Michal Valko. Best of Both Worlds: Stochastic & Adversarial Best-arm Identification. In Proceedings of the 31st Conference On Learning Theory, volume 75 of Proceedings of Machine Learning Research, pages 918-949. PMLR, 06-09 Jul 2018.
  2. [Awad et al., 2021] Noor H. Awad, Neeratyoy Mallik, and Frank Hutter. DEHB: Evolutionary Hyberband for Scal- able, Robust and Efficient Hyperparameter Optimization. In Proceedings of the 30th International Joint Conference on Artificial Intelligence, pages 2147-2153, 2021. IJCAI
  3. [Azizi et al., 2022] Mohammad Javad Azizi, Branislav Kve- ton, and Mohammad Ghavamzadeh. Fixed-Budget Best- Arm Identification in Structured Bandits. In Proceedings of the Thirty-First International Joint Conference on Artifi- cial Intelligence, IJCAI, pages 2798-2804. ijcai.org, 2022. IJCAI
  4. [Bergstra et al., 2011] James Bergstra, Rémi Bardenet, Yoshua Bengio, and Balázs Kégl. Algorithms for Hyper-Parameter Optimization. In Advances in Neural Information Processing Systems, pages 2546-2554, 2011.
  5. [Bischl et al., 2023] Bernd Bischl, Martin Binder, Michel Lang, Tobias Pielok, Jakob Richter, Stefan Coors, Janek Thomas, Theresa Ullmann, Marc Becker, Anne-Laure Boulesteix, et al. Hyperparameter Optimization: Foun- dations, Algorithms, Best Practices and Open Challenges. Wiley Interdisciplinary Reviews: Data Mining and Knowl- edge Discovery, 13, 2023.
  6. [Bohdal et al., 2023] Ondrej Bohdal, Lukas Balles, Martin Wistuba, Beyza Ermis, Cédric Archambeau, and Giovanni Zappella. PASHA: Efficient HPO and NAS with Progres- sive Resource Allocation. In The 11th International Con- ference on Learning Representations, 2023.
  7. [Brandt et al., 2022] Jasmin Brandt, Viktor Bengs, Björn Haddenhorst, and Eyke Hüllermeier. Finding Optimal Arms in Non-stochastic Combinatorial Bandits with Semi- bandit Feedback and Finite Budget. In Advances in Neural Information Processing Systems, 2022.
  8. [Brandt et al., 2023] Jasmin Brandt, Elias Schede, Björn Haddenhorst, Viktor Bengs, Eyke Hüllermeier, and Kevin Tierney. AC-Band: A Combinatorial Bandit-Based Ap- proach to Algorithm Configuration. Proceedings of the AAAI Conference on Artificial Intelligence, 37(10):12355- 12363, 2023.
  9. [Carpentier and Valko, 2015] Alexandra Carpentier and Michal Valko. Simple regret for infinitely many armed bandits. In Proceedings of the 32nd International Conference on Machine Learning, ICML, volume 37 of JMLR Workshop and Conference Proceedings, pages 1133-1141. JMLR.org, 2015.
  10. [Falkner et al., 2018] Stefan Falkner, Aaron Klein, and Frank Hutter. BOHB: Robust and Efficient Hyperparame- ter Optimization at Scale. In Proceedings of the 35th Inter- national Conference on Machine Learning, volume 80 of Proceedings of Machine Learning Research, pages 1436- 1445. PMLR, 2018.
  11. [Feurer and Hutter, 2019] Matthias Feurer and Frank Hutter. Hyperparameter Optimization. In Automated Machine Learning -Methods, Systems, Challenges, The Springer Series on Challenges in Machine Learning, pages 3-33. Springer, 2019.
  12. [Frazier, 2018] Peter I. Frazier. A Tutorial on Bayesian Op- timization. CoRR, abs/1807.02811, 2018.
  13. [Gong and Sellke, 2023] Xiao-Yue Gong and Mark Sellke. Asymptotically Optimal Pure Exploration for Infinite- Armed Bandits. CoRR, abs/2306.01995, 2023.
  14. [Hutter et al., 2011] Frank Hutter, Holger H. Hoos, and Kevin Leyton-Brown. Sequential Model-Based Optimiza- tion for General Algorithm Configuration. In Learning and Intelligent Optimization -5th International Confer- ence, volume 6683 of Lecture Notes in Computer Science, pages 507-523. Springer, 2011.
  15. [Jamieson and Talwalkar, 2016a] Kevin Jamieson and Ameet Talwalkar. Non-stochastic Best Arm Identification and Hyperparameter Optimization. In Proceedings of the 19th International Conference on Artificial Intelligence and Statistics, volume 51 of Proceedings of Machine Learning Research, pages 240-248. PMLR, 2016.
  16. [Jamieson and Talwalkar, 2016b] Kevin G. Jamieson and Ameet Talwalkar. Non-stochastic Best Arm Identifica- tion and Hyperparameter Optimization. In Proceedings of the 19th International Conference on Artificial Intelli- gence and Statistics, AISTATS, volume 51 of JMLR Work- shop and Conference Proceedings, pages 240-248, 2016.
  17. [Karnin et al., 2013] Zohar Karnin, Tomer Koren, and Oren Somekh. Almost Optimal Exploration in Multi-Armed Bandits. In International Conference on Machine Learn- ing, pages 1238-1246. PMLR, 2013.
  18. [Kato et al., 2022] Masahiro Kato, Kaito Ariu, Masaaki Imaizumi, Masatoshi Uehara, Masahiro Nomura, and Chao Qin. Optimal Fixed-Budget Best Arm Identification using the Augmented Inverse Probability Weighting Es- timator in Two-Armed Gaussian Bandits with Unknown Variances. CoRR, abs/2201.04469, 2022.
  19. [Lattimore and Szepesvári, 2020] Tor Lattimore and Csaba Szepesvári. Bandit Algorithms. Cambridge University Press, 2020.
  20. [Li et al., 2018] Lisha Li, Kevin Jamieson, Giulia DeSalvo, Afshin Rostamizadeh, and Ameet Talwalkar. Hyper- band: A Novel Bandit-Based Approach to Hyperparame- ter Optimization. Journal of Machine Learning Research, 18(185):1-52, 2018.
  21. [Li et al., 2020] Liam Li, Kevin Jamieson, Afshin Ros- tamizadeh, Ekaterina Gonina, Jonathan Ben-Tzur, Moritz Hardt, Benjamin Recht, and Ameet Talwalkar. A System for Massively Parallel Hyperparameter Tuning. Proceed- ings of Machine Learning and Systems, 2:230-246, 2020.
  22. [Mallik et al., 2023] Neeratyoy Mallik, Edward Bergman, Carl Hvarfner, Danny Stoll, Maciej Janowski, Marius Lin- dauer, Luigi Nardi, and Frank Hutter. PriorBand: Practical Hyperparameter Optimization in the Age of Deep Learn- ing. arXiv preprint arXiv:2306.12370, 2023.
  23. [Mendes et al., 2021] Pedro Mendes, Maria Casimiro, and Paolo Romano. HyperJump: Accelerating HyperBand via Risk Modelling. arXiv preprint arXiv:2108.02479, 2021.
  24. [Pfisterer et al., 2022] Florian Pfisterer, Lennart Schneider, Julia Moosbauer, Martin Binder, and Bernd Bischl. YAHPO Gym -An Efficient Multi-Objective Multi- Fidelity Benchmark for Hyperparameter Optimization. In International Conference on Automated Machine Learn- ing, AutoML 2022, volume 188 of Proceedings of Machine Learning Research, pages 3/1-39. PMLR, 2022.
  25. [Rice, 1976] John R. Rice. The Algorithm Selection Prob- lem. Advances in Computers, 15:65-118, 1976.
  26. [Shen, 2019] Cong Shen. Universal Best Arm Identification. IEEE Trans. Signal Process., 67(17):4464-4478, 2019.
  27. [Tornede et al., 2023] Tanja Tornede, Alexander Tornede, Jonas Hanselle, Felix Mohr, Marcel Wever, and Eyke Hüllermeier. Towards green automated machine learning: Status quo and future directions. Journal of Artificial In- telligence Research, 77:427-457, 2023.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards