Proceedings

Automated Random Embedding for Practical Bayesian Optimization with Unknown Effective Dimension

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 4732-4740.

https://doi.org/10.24963/ijcai.2026/527

Abstract

Bayesian optimization is widely employed for optimizing complex black-box functions but struggles with the curse of dimensionality. Random embedding, as a dimension reduction strategy, simplifies tasks that possess the effective dimension by optimizing within a low-dimensional subspace. However, determining the effective dimension of a task in advance remains a significant challenge, which influences the selection of the subspace dimensionality and the optimization performance. Traditional methods use fixed subspace dimensions provided by experts or rely on trial and error to estimate subspace dimensions with resources consumed. To this end, this paper proposes an automated random embedding for high-dimensional Bayesian optimization with unknown effective dimension, called Dynamic Shared Embedding Bayesian Optimization (DSEBO). DSEBO starts with a low dimension and switches to a higher subspace if the solutions in the current subspace show preliminary convergence. DSEBO dynamically determines the dimension of the next subspace based on the quality of the solutions in different subspaces and shares the queried solutions with the new subspace for a better initialization. Theoretically, we derive a regret bound for DSEBO and demonstrate that DSEBO can better balance approximation and optimization errors. Extensive experiments on functions with dimensionality of varying magnitudes and real-world tasks with unknown effective dimensions reveal that, compared with state-of-the-art methods, alternating optimization across different subspaces results in significant improvements in high-dimensional optimization, both in terms of optimization regret and time.

  1. [Amos and Yarats, 2020] Brandon Amos and Denis Yarats. The differentiable cross-entropy method. In Proceedings of the 37th International Conference on Machine Learning, volume 119, pages 291-302, Virtual Event, 2020.
  2. [Audibert et al., 2010] Jean-Yves Audibert, Sébastien Bubeck, and Rémi Munos. Best arm identification in multi-armed bandits. In Proceedings of the 23rd Conference on Learning Theory, pages 41-53, Haifa, Israel, 2010.
  3. [Auer et al., 2002] Peter Auer, Nicolò Cesa-Bianchi, and Paul Fischer. Finite-time analysis of the multiarmed bandit problem. Machine Learning, 47:235-256, 2002.
  4. [Bardou et al., 2024] Anthony Bardou, Patrick Thiran, and Thomas Begin. Relaxing the additivity constraints in decen- tralized no-regret high-dimensional Bayesian optimization. In Proceedings of the 12th International Conference on Learning Representations, 2024.
  5. [Binois and Wycoff, 2022] Mickael Binois and Nathan Wycoff. A survey on high-dimensional Gaussian process modeling with application to Bayesian optimization. ACM Transactions on Evolutionary Learning and Optimization, 2(2):8:1-8:26, 2022.
  6. [Boyd and Vandenberghe, 2004] Stephen Boyd and Lieven Vandenberghe. Convex Optimization. Cambridge Univer- sity Press, Cambridge, 2004.
  7. [Cartis et al., 2023] Coralia Cartis, Estelle M. Massart, and Adilet Otemissov. Bound-constrained global optimization of functions with low effective dimensionality using mul- tiple random embeddings. Mathematical Programming, 198(1):997-1058, 2023.
  8. [Eckmann et al., 2022] Peter Eckmann, Kunyang Sun, Bo Zhao, Mudong Feng, Michael K. Gilson, and Rose Yu. LIMO: Latent inceptionism for targeted molecule generation. In Proceedings of the 39th International Conference on Machine Learning, volume 162, pages 5777-5792, Baltimore, MD, 2022.
  9. [Elsken et al., 2019] Thomas Elsken, Jan Hendrik Metzen, and Frank Hutter. Neural architecture search: A survey. Journal of Machine Learning Research, 20:55:1-55:21, 2019.
  10. [Eriksson and Jankowiak, 2021] David Eriksson and Martin Jankowiak. High-dimensional Bayesian optimization with sparse axis-aligned subspaces. In Proceedings of the 37th Conference on Uncertainty in Artificial Intelligence, pages 493-503, Virtual Event, 2021.
  11. [Eriksson et al., 2019] David Eriksson, Michael Pearce, Ja- cob R. Gardner, Ryan Turner, and Matthias Poloczek. Scal- able global optimization via local Bayesian optimization. In Advances in Neural Information Processing Systems 32, pages 5497-5508, Vancouver, Canada, 2019.
  12. [Freund and Schapire, 1997] Yoav Freund and Robert E. Schapire. A decision-theoretic generalization of on-line learning and an application to boosting. Journal of Com- puter and System Sciences, 55(1):119-139, 1997.
  13. [Garnett et al., 2014] Roman Garnett, Michael A. Osborne, and Philipp Hennig. Active learning of linear embeddings for Gaussian processes. In Proceedings of the 30th Confer- ence on Uncertainty in Artificial Intelligence, pages 230- 239, Quebec, Canada, 2014.
  14. [Garnett, 2023] Roman Garnett. Bayesian Optimization. Cambridge University Press, Cambridge, 2023.
  15. [Gómez-Bombarelli et al., 2018] Rafael Gómez-Bombarelli, Jennifer N Wei, David Duvenaud, José Miguel Hernández- Lobato, Benjamín Sánchez-Lengeling, Dennis Sheberla, Jorge Aguilera-Iparraguirre, Timothy D Hirzel, Ryan P Adams, and Alán Aspuru-Guzik. Automatic chemical design using a data-driven continuous representation of molecules. ACS Central Science, 4(2):268-276, 2018.
  16. [Hvarfner et al., 2024] Carl Hvarfner, Erik Orm Hellsten, and Luigi Nardi. Vanilla Bayesian optimization performs great in high dimensions. In Proceedings of the 41st Interna- tional Conference on Machine Learning, volume 235, pages 20793-20817, Vienna, Austria, 2024.
  17. [Jin et al., 2022] Tianyuan Jin, Pan Xu, Xiaokui Xiao, and Anima Anandkumar. Finite-time regret of Thompson sam- pling algorithms for exponential family multi-armed ban- dits. In Advances in Neural Information Processing Systems 35, pages 38475-38487, New Orleans, LA, 2022.
  18. [Karnin et al., 2013] Zohar Shay Karnin, Tomer Koren, and Oren Somekh. Almost optimal exploration in multi-armed bandits. In Proceedings of the 30th International Confer- ence on Machine Learning, pages 1238-1246, Atlanta, GA, 2013.
  19. [Langford and Zhang, 2007] John Langford and Tong Zhang. The epoch-greedy algorithm for multi-armed bandits with side information. In Advances in Neural Information Pro- cessing Systems 20, pages 817-824, Vancouver, Canada, 2007.
  20. [Letham et al., 2020] Benjamin Letham, Roberto Calandra, Akshara Rai, and Eytan Bakshy. Re-examining linear em- beddings for high-dimensional Bayesian optimization. In Advances in Neural Information Processing Systems 33, Virtual Event, 2020.
  21. [Loshchilov et al., 2019] Ilya Loshchilov, Tobias Glasmach- ers, and Hans-Georg Beyer. Large scale black-box op- timization by limited-memory matrix adaptation. IEEE Transactions on Evolutionary Computation, 23(2):353-358, 2019.
  22. [Mehta and Grosan, 2015] Dhagash Mehta and Crina Grosan. A collection of challenging optimization problems in sci- ence, engineering and economics. In Proceedings of the 17th IEEE Congress on Evolutionary Computation, pages 2697-2704, Sendai, Japan, 2015.
  23. [Nayebi et al., 2019] Amin Nayebi, Alexander Munteanu, and Matthias Poloczek. A framework for Bayesian op- timization in embedded subspaces. In Proceedings of the 36th International Conference on Machine Learning, vol- ume 97, pages 4752-4761, Long Beach, CA, 2019.
  24. [Papenmeier et al., 2022] Leonard Papenmeier, Luigi Nardi, and Matthias Poloczek. Increasing the scope as you learn: Adaptive Bayesian optimization in nested subspaces. In Advances in Neural Information Processing Systems 35, pages 11586-11601, New Orleans, LA, 2022.
  25. [Qian and Yu, 2021] Hong Qian and Yang Yu. Derivative- free reinforcement learning: A review. Frontiers of Com- puter Science, 15(6):156336, 2021.
  26. [Qian et al., 2016] Hong Qian, Yi-Qi Hu, and Yang Yu. Derivative-free optimization of high-dimensional non- convex functions by sequential random embeddings. In Proceedings of the 25th International Joint Conference on Artificial Intelligence, pages 1946-1952, New York, NY, 2016.
  27. [Qin et al., 2010] Tao Qin, Tie-Yan Liu, Jun Xu, and Hang Li. LETOR: A benchmark collection for research on learning to rank for information retrieval. Information Retrieval, 13(4):346-374, 2010.
  28. [Santoni et al., 2024] Maria Laura Santoni, Elena Raponi, Re- nato De Leone, and Carola Doerr. Comparison of high- dimensional Bayesian optimization algorithms on BBOB. ACM Transactions on Evolutionary Learning and Optimiza- tion, 4(3):17:1-17:33, 2024.
  29. [Sehic et al., 2022] Kenan Sehic, Alexandre Gramfort, Joseph Salmon, and Luigi Nardi. LassoBench: A high-dimensional hyperparameter optimization benchmark suite for Lasso. In Proceedings of the 1st International Conference on Automated Machine Learning, volume 188, pages 2/1-24, Baltimore, MD, 2022.
  30. [Shahriari et al., 2016] Bobak Shahriari, Kevin Swersky, Ziyu Wang, Ryan P. Adams, and Nando de Freitas. Taking the human out of the loop: A review of Bayesian optimiza- tion. Proceedings of the IEEE, 104(1):148-175, 2016.
  31. [Song et al., 2022] Lei Song, Ke Xue, Xiaobin Huang, and Chao Qian. Monte Carlo tree search based variable se- lection for high dimensional Bayesian optimization. In Advances in Neural Information Processing Systems 35, pages 28488-28501, New Orleans, LA, 2022.
  32. [Srinivas et al., 2010] Niranjan Srinivas, Andreas Krause, Sham M. Kakade, and Matthias W. Seeger. Gaussian pro- cess optimization in the bandit setting: No regret and exper- imental design. In Proceedings of the 27th International Conference on Machine Learning, pages 1015-1022, Haifa, Israel, 2010.
  33. [Sutton and Barto, 2018] Richard S Sutton and Andrew G Barto. Reinforcement Learning: An Introduction. MIT Press, Cambridge, 2018.
  34. [Wang et al., 2013] Ziyu Wang, Masrour Zoghi, Frank Hut- ter, David Matheson, and Nando de Freitas. Bayesian optimization in high dimensions via random embeddings. In Proceedings of the 23rd International Joint Conference on Artificial Intelligence, pages 1778-1784, Beijing, China, 2013.
  35. [Wang et al., 2016] Ziyu Wang, Frank Hutter, Masrour Zoghi, David Matheson, and Nando de Freitas. Bayesian opti- mization in a billion dimensions via random embeddings. Journal of Artificial Intelligence Research, 55:361-387, 2016.
  36. [Xu et al., 2025] Zhitong Xu, Haitao Wang, Jeff M Phillips, and Shandian Zhe. Standard Gaussian process is all you need for high-dimensional Bayesian optimization. In Pro- ceedings of the 13th International Conference on Learning Representations, 2025.
  37. [Yu et al., 2025] Yang Yu, Hong Qian, and Yi-Qi Hu. Derivative-Free Optimization: Theoretical Foundations, Algorithms, and Applications. Springer, 2025.
  38. [Zhang et al., 2019] Miao Zhang, Huiqi Li, and Steven W. Su. High dimensional Bayesian optimization via super- vised dimension reduction. In Proceedings of the 28th International Joint Conference on Artificial Intelligence, pages 4292-4298, Macao, China, 2019. IJCAI
  39. [Zhou et al., 2019] Zhi-Hua Zhou, Yang Yu, and Chao Qian. Evolutionary Learning: Advances in Theories and Algo- rithms. Springer Publishing Company, Incorporated, Berlin, 2019.
  40. [Ziomek and Bou-Ammar, 2023] Juliusz Krysztof Ziomek and Haitham Bou-Ammar. Are random decompositions all we need in high dimensional Bayesian optimisation? In Proceedings of the 40th International Conference on Machine Learning, volume 202, pages 43347-43368, Hon- olulu, HI, 2023.

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

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