Proceedings

c-TPE: Tree-structured Parzen Estimator with Inequality Constraints for Expensive Hyperparameter Optimization

Download BibTeX Cite

Proceedings of the Thirty-Second International Joint Conference on Artificial Intelligence. Main Track. Pages 4371-4379.

https://doi.org/10.24963/ijcai.2023/486

Abstract

Hyperparameter optimization (HPO) is crucial for strong performance of deep learning algorithms and real-world applications often impose some constraints, such as memory usage, or latency on top of the performance requirement. In this work, we propose constrained TPE (c-TPE), an extension of the widely-used versatile Bayesian optimization method, tree-structured Parzen estimator (TPE), to handle these constraints. Our proposed extension goes beyond a simple combination of an existing acquisition function and the original TPE, and instead includes modifications that address issues that cause poor performance. We thoroughly analyze these modifications both empirically and theoretically, providing insights into how they effectively overcome these challenges. In the experiments, we demonstrate that c-TPE exhibits the best average rank performance among existing methods with statistical significance on 81 expensive HPO with inequality constraints. Due to the lack of baselines, we only discuss the applicability of our method to hard-constrained optimization in Appendix D. See https://arxiv.org/abs/2211.14411 for the latest version with Appendix.

  1. [Addison et al., 2022] H. Addison, KS. inversion, H. Ryan, and C. Ted. Happywhale -whale and dolphin identifica- tion. Kaggle, 2022.
  2. [Akiba et al., 2019] T. Akiba, S. Sano, T. Yanase, T. Ohta, and M. Koyama. Optuna: A next-generation hyperparam- eter optimization framework. In International Conference on Knowledge Discovery & Data Mining, 2019.
  3. [Alina et al., 2019] JE. Alina, C. Phil, B. Rodrigo, and G. Victor. Open images 2019 -object detection. Kaggle, 2019.
  4. [Arnold and Hansen, 2012] D. Arnold and N. Hansen. A (1+1)-CMA-ES for constrained optimisation. In Genetic and Evolutionary Computation Conference, 2012.
  5. [Awad et al., 2021] N. Awad, N. Mallik, and F. Hutter. DEHB: Evolutionary hyberband for scalable, robust and efficient hyperparameter optimization. arXiv:2105.09821, 2021. IJCAI
  6. [Bergstra and Bengio, 2012] J. Bergstra and Y. Bengio. Ran- dom search for hyper-parameter optimization. Journal of Machine Learning Research, 13(2), 2012.
  7. [Bergstra et al., 2011] J. Bergstra, R. Bardenet, Y. Bengio, and B. Kégl. Algorithms for hyper-parameter optimiza- tion. In Advances in Neural Information Processing Sys- tems, 2011.
  8. [Bergstra et al., 2013] J. Bergstra, D. Yamins, and D. Cox. Making a science of model search: Hyperparameter op- timization in hundreds of dimensions for vision architec- tures. In International Conference on Machine Learning, 2013.
  9. [Brochu et al., 2010] E. Brochu, V. Cora, and N. de Fre- itas. A tutorial on Bayesian optimization of expensive cost functions, with application to active user modeling and hi- erarchical reinforcement learning. arXiv:1012.2599, 2010.
  10. [Chen et al., 2018] Y. Chen, A. Huang, Z. Wang, I. Antonoglou, J. Schrittwieser, D. Silver, and N. de Fre- itas. Bayesian optimization in alphago. arXiv:1812.06855, 2018.
  11. [Deb et al., 2002] K. Deb, A. Pratap, S. Agarwal, and T. Me- yarivan. A fast and elitist multiobjective genetic algorithm: NSGA-II. IEEE Transactions on Evolutionary Computa- tion, 6(2):182-197, 2002.
  12. [Dong and Yang, 2020] X. Dong and Y. Yang. NAS-Bench- 201: Extending the scope of reproducible neural architec- ture search. arXiv:2001.00326, 2020.
  13. [Eriksson and Poloczek, 2021] D. Eriksson and M. Poloczek. Scalable constrained Bayesian opti- mization. In International Conference on Artificial Intelligence and Statistics, 2021.
  14. [Falkner et al., 2018] S. Falkner, A. Klein, and F. Hutter. BOHB: Robust and efficient hyperparameter optimization at scale. In International Conference on Machine Learn- ing, 2018.
  15. [Gardner et al., 2014] J. Gardner, M. Kusner, ZE. Xu, K. Weinberger, and J. Cunningham. Bayesian optimiza- tion with inequality constraints. In International Confer- ence on Machine Learning, 2014.
  16. [Garnett, 2022] R. Garnett. Bayesian Optimization. Cam- bridge University Press, 2022.
  17. [Garrido-Merchán et al., 2023] EC.
  18. Garrido-Merchán, D. Fernández-Sánchez, and D. Hernández-Lobato. Parallel predictive entropy search for multi-objective Bayesian optimization with constraints applied to the tuning of machine learning algorithms. Expert Systems with Applications, 215, 2023.
  19. [Gelbart et al., 2014] M. Gelbart, J. Snoek, and R. Adams. Bayesian optimization with unknown constraints. arXiv:1403.5607, 2014.
  20. [Jones et al., 1998] D. Jones, M. Schonlau, and W. Welch. Efficient global optimization of expensive black-box func- tions. Journal of Global Optimization, 13(4):455-492, 1998.
  21. [Klein and Hutter, 2019] A. Klein and F. Hutter. Tabular benchmarks for joint architecture and hyperparameter op- timization. arXiv:1905.04970, 2019.
  22. [Kushner, 1964] HJ. Kushner. A new method of locating the maximum point of an arbitrary multipeak curve in the presence of noise. Joint Automatic Control Conference, 1964.
  23. [Letham et al., 2019] B. Letham, B. Karrer, G. Ottoni, and E. Bakshy. Constrained Bayesian optimization with noisy experiments. Bayesian Analysis, 14(2):495-519, 2019.
  24. [Liaw et al., 2018] R. Liaw, E. Liang, R. Nishihara, P. Moritz, J. Gonzalez, and I. Stoica. Tune: A re- search platform for distributed model selection and train- ing. arXiv:1807.05118, 2018.
  25. [Lobato et al., 2015] JH. Lobato, M. Gelbart, M. Hoffman, R. Adams, and Z. Ghahramani. Predictive entropy search for bayesian optimization with unknown constraints. In International Conference on Machine Learning, 2015.
  26. [Loshchilov and Hutter, 2016] I. Loshchilov and F. Hutter. CMA-ES for hyperparameter optimization of deep neural networks. arXiv:1604.07269, 2016.
  27. [Loshchilov et al., 2013] I. Loshchilov, M. Schoenauer, and M. Sebag. Bi-population CMA-ES agorithms with surro- gate models and line searches. In Genetic and Evolution- ary Computation Conference, 2013.
  28. [Melis et al., 2018] G. Melis, C. Dyer, and P. Blunsom. On the state of the art of evaluation in neural language models. In International Conference on Learning Representations, 2018.
  29. [Montes et al., 2006] EM. Montes, J. Velázquez-Reyes, and CA. Coello. Modified differential evolution for con- strained optimization. In International Conference on Evo- lutionary Computation, 2006.
  30. [Nardi et al., 2019] L. Nardi, D. Koeplinger, and K. Oluko- tun. Practical design space exploration. In Interna- tional Symposium on Modeling, Analysis, and Simulation of Computer and Telecommunication Systems, pages 347- 358. IEEE, 2019.
  31. [Perrone et al., 2019] V. Perrone, I. Shcherbatyi, R. Jenat- ton, C. Archambeau, and M. Seeger. Constrained Bayesian optimization with max-value entropy search. arXiv:1910.07003, 2019.
  32. [Shahriari et al., 2016] B. Shahriari, K. Swersky, Z. Wang, R. Adams, and N. de Freitas. Taking the human out of the loop: A review of Bayesian optimization. Proceedings of the IEEE, 104(1):148-175, 2016.
  33. [Watanabe, 2023] S. Watanabe. Tree-structured Parzen esti- mator: Understanding its algorithm components and their roles for better empirical performance. arXiv:2304.11127, 2023.
  34. [Ying et al., 2019] C. Ying, A. Klein, E. Christiansen, E. Real, K. Murphy, and F. Hutter. NAS-Bench-101: To- wards reproducible neural architecture search. In Interna- tional Conference on Machine Learning, 2019.

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

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