Proceedings

Speeding Up Multi-Objective Hyperparameter Optimization by Task Similarity-Based Meta-Learning for the Tree-Structured Parzen Estimator

Download BibTeX Cite

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

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

Abstract

Hyperparameter optimization (HPO) is a vital step in improving performance in deep learning (DL). Practitioners are often faced with the trade-off between multiple criteria, such as accuracy and latency. Given the high computational needs of DL and the growing demand for efficient HPO, the acceleration of multi-objective (MO) optimization becomes ever more important. Despite the significant body of work on meta-learning for HPO, existing methods are inapplicable to MO tree-structured Parzen estimator (MO-TPE), a simple yet powerful MO-HPO algorithm. In this paper, we extend TPE’s acquisition function to the meta-learning setting using a task similarity defined by the overlap of top domains between tasks. We also theoretically analyze and address the limitations of our task similarity. In the experiments, we demonstrate that our method speeds up MO-TPE on tabular HPO benchmarks and attains state-of-the-art performance. Our method was also validated externally by winning the AutoML 2022 competition on “Multiobjective Hyperparameter Optimization for Transformers”. See https://arxiv.org/abs/2212.06751 for the latest version with Appendix.

  1. [Bader and Zitzler, 2011] J. Bader and E. Zitzler. HypE: An algorithm for fast hypervolume-based many-objective op- timization. Evolutionary Computation, 19, 2011.
  2. [Bergstra and Bengio, 2012] J. Bergstra and Y. Bengio. Ran- dom search for hyper-parameter optimization. Journal of Machine Learning Research, 13(2), 2012.
  3. [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.
  4. [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.
  5. [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.
  6. [Candelieri et al., 2022] A. Candelieri, A. Ponti, and F. Archetti. Fair and green hyperparameter optimization via multi-objective and multiple information source Bayesian optimization. arXiv:2205.08835, 2022.
  7. [Chen et al., 2018] Y. Chen, A. Huang, Z. Wang, I. Antonoglou, J. Schrittwieser, D. Silver, and N. de Freitas. Bayesian optimization in AlphaGo. arXiv:1812.06855, 2018.
  8. [Chugh et al., 2016] T. Chugh, Y. Jin, K. Miettinen, J. Haka- nen, and K. Sindhya. A surrogate-assisted reference vector guided evolutionary algorithm for computationally expen- sive many-objective optimization. Evolutionary Computa- tion, 22, 2016.
  9. [Deb et al., 2002] K. Deb, A. Pratap, S. Agarwal, and T. Me- yarivan. A fast and elitist multiobjective genetic algorithm: NSGA-II. Evolutionary Computation, 6, 2002.
  10. [Eggensperger et al., 2021] K. Eggensperger, P. Müller, N. Mallik, M. Feurer, R. Sass, A. Klein, N. Awad, M. Lin- dauer, and F. Hutter. HPOBench: A collection of repro- ducible multi-fidelity benchmark problems for HPO. In Advances in Neural Information Processing Systems Track on Datasets and Benchmarks, 2021.
  11. [Emmerich et al., 2011] MTM. Emmerich, AH. Deutz, and JW. Klinkenberg. Hypervolume-based expected improve- ment: Monotonicity properties and exact computation. In Congress of Evolutionary Computation, 2011.
  12. [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.
  13. [Feurer et al., 2015] M. Feurer, JT. Springenberg, and F. Hutter. Initializing Bayesian hyperparameter optimiza- tion via meta-learning. In AAAI Conference on Artificial Intelligence, 2015.
  14. [Feurer et al., 2018] M. Feurer, B. Letham, F. Hutter, and E. Bakshy. Practical transfer learning for Bayesian opti- mization. arXiv:1802.02219, 2018.
  15. [Fonseca and Fleming, 1996] CM. Fonseca and PJ. Flem- ing. On the performance assessment and comparison of stochastic multiobjective optimizers. In International Conference on Parallel Problem Solving from Nature, 1996.
  16. [Garnett, 2022] R. Garnett. Bayesian Optimization. Cam- bridge University Press, 2022.
  17. [Guo et al., 2018] D. Guo, Y. Jin, J. Ding, and T. Chai. Het- erogeneous ensemble-based infill criterion for evolution- ary multiobjective optimization of expensive problems. Transactions on Cybernetics, 49, 2018.
  18. [Henderson et al., 2018] P. Henderson, R. Islam, P. Bach- man, J. Pineau, D. Precup, and D. Meger. Deep reinforce- ment learning that matters. In AAAI Conference on Artifi- cial Intelligence, 2018.
  19. [Hernández-Lobato et al., 2016] D. Hernández-Lobato, J. Hernandez-Lobato, A. Shah, and R. Adams. Predictive entropy search for multi-objective Bayesian optimization. In International Conference on Machine Learning, 2016.
  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, 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. [Knowles, 2006] J. Knowles. ParEGO: A hybrid algorithm with on-line landscape approximation for expensive mul- tiobjective optimization problems. Evolutionary Compu- tation, 10, 2006.
  23. [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.
  24. [Nomura et al., 2021] M. Nomura, S. Watanabe, Y. Aki- moto, Y. Ozaki, and M. Onishi. Warm starting CMA-ES for hyperparameter optimization. In AAAI Conference on Artificial Intelligence, 2021.
  25. [Ozaki et al., 2020] Y. Ozaki, Y. Tanigaki, S. Watanabe, and M. Onishi. Multiobjective tree-structured Parzen estimator for computationally expensive optimization problems. In Genetic and Evolutionary Computation Conference, 2020.
  26. [Ozaki et al., 2022] Y. Ozaki, Y. Tanigaki, S. Watanabe, M. Nomura, and M. Onishi. Multiobjective tree-structured Parzen estimator. Journal of Artificial Intelligence Re- search, 73, 2022.
  27. [Pan et al., 2018] L. Pan, C. He, Y. Tian, H. Wang, X. Zhang, and Y. Jin. A classification-based surrogate-assisted evolu- tionary algorithm for expensive many-objective optimiza- tion. Evolutionary Computation, 23, 2018.
  28. [Perrone et al., 2018] V. Perrone, R. Jenatton, M. Seeger, and C. Archambeau. Scalable hyperparameter transfer learning. In Advances in Neural Information Processing Systems, 2018.
  29. [Perrone et al., 2019] V. Perrone, H. Shen, MW. Seeger, C. Archambeau, and R. Jenatton. Learning search spaces for Bayesian optimization: Another view of hyperparame- ter transfer learning. Advances in Neural Information Pro- cessing Systems, 2019.
  30. [Ponweiser et al., 2008] W. Ponweiser, T. Wagner, D. Bier- mann, and M. Vincze. Multiobjective optimization on a limited budget of evaluations using model-assisted S- metric selection. In International Conference on Parallel Problem Solving from Nature, 2008.
  31. [Salinas et al., 2020] D. Salinas, H. Shen, and V. Perrone. A quantile-based approach for hyperparameter transfer learning. In International Conference on Machine Learn- ing, 2020.
  32. [Schmucker et al., 2020] R. Schmucker, M. Donini, V. Per- rone, MB. Zafar, and C. Archambeau. Multi-objective multi-fidelity hyperparameter optimization with applica- tion to fairness. In Meta-Learning Workshop at Advances in Neural Information Processing Systems, 2020.
  33. [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. IEEE, 104, 2016.
  34. [Song et al., 2022] J. Song, L. Yu, W. Neiswanger, and S. Er- mon. A general recipe for likelihood-free Bayesian opti- mization. In International Conference on Machine Learn- ing, 2022.
  35. [Springenberg et al., 2016] JT. Springenberg, A. Klein, S. Falkner, and F. Hutter. Bayesian optimization with ro- bust Bayesian neural networks. In Advances in Neural In- formation Processing Systems, 2016.
  36. [Swersky et al., 2013] K. Swersky, J. Snoek, and R. Adams. Multi-task Bayesian optimization. In Advances in Neural Information Processing Systems, 2013.
  37. [Vanschoren, 2019] J. Vanschoren. Meta-learning. In Auto- mated Machine Learning, pages 35-61. Springer, 2019.
  38. [Volpp et al., 2020] M. Volpp, LP. Fröhlich, K. Fischer, A. Doerr, S. Falkner, F. Hutter, and C. Daniel. Meta- learning acquisition functions for transfer learning in Bayesian optimization. In International Conference on Learning Representations, 2020.
  39. [Wang and Jegelka, 2017] Z. Wang and S. Jegelka. Max- value entropy search for efficient Bayesian optimization. In International Conference on Machine Learning, 2017.
  40. [Watanabe and Hutter, 2022] S. Watanabe and F. Hutter. c- TPE: Generalizing tree-structured Parzen estimator with inequality constraints for continuous and categorical hy- perparameter optimization. Gaussian Processes, Spa- tiotemporal Modeling, and Decision-making Systems Workshop at Advances in Neural Information Processing Systems, 2022.
  41. [Watanabe and Hutter, 2023] S. Watanabe and F. Hutter. c- TPE: Tree-structured Parzen estimator with inequality constraints for expensive hyperparameter optimization. arXiv:2211.14411, 2023. IJCAI
  42. [Watanabe et al., 2023] S. Watanabe, A. Bansal, and F. Hut- ter. PED-ANOVA: Efficiently quantifying hyperparame- ter importance in arbitrary subspaces. arXiv:2304.10255, 2023. IJCAI
  43. [Watanabe, 2023a] S. Watanabe. Python tool for visualizing variability of Pareto fronts over multiple runs. 2023.
  44. [Watanabe, 2023b] S. Watanabe. Tree-structured Parzen estimator: Understanding its algorithm components and their roles for better empirical performance. arXiv:2304.11127, 2023.
  45. [Wistuba et al., 2015] M. Wistuba, N. Schilling, and L. Schmidt-Thieme. Hyperparameter search space pruning-a new component for sequential model-based hyperparameter optimization. In Joint European Confer- ence on Machine Learning and Knowledge Discovery in Databases, 2015.
  46. [Wistuba et al., 2016] M. Wistuba, N. Schilling, and L. Schmidt-Thieme. Two-stage transfer surrogate model for automatic hyperparameter optimization. In Joint Eu- ropean Conference on Machine Learning and Knowledge Discovery in Databases, 2016.
  47. [Zhang and Duh, 2020] X. Zhang and K. Duh. Reproducible and efficient benchmarks for hyperparameter optimization of neural machine translation systems. Transactions of the Association for Computational Linguistics, 8, 2020.
  48. [Zhang and Li, 2007] Q. Zhang and H. Li. MOEA/D: A mul- tiobjective evolutionary algorithm based on decomposi- tion. Evolutionary Computation, 11, 2007.

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

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