From Sorting Algorithms to Scalable Kernels: Bayesian Optimization in High-Dimensional Permutation Spaces
Zikai Xie, Linjiang Chen
Abstract
Bayesian Optimization (BO) is a powerful tool for black-box optimization, but its application to high-dimensional permutation spaces is severely limited by the challenge of defining scalable representations. The current state-of-the-art BO approach for permutation spaces relies on an exhaustive pairwise comparison, inducing a dense representation that is impractical for large-scale permutations. To break this barrier, we introduce a novel framework for generating efficient permutation representations via kernel functions derived from sorting algorithms. Within this framework, the Mallows kernel can be viewed as a special instance derived from enumeration sort. Further, we introduce the Merge Kernel , which leverages the divide-and-conquer structure of merge sort to produce a compact, to achieve the lowest possible complexity with no information loss and effectively capture permutation structure. Our central thesis is that the Merge Kernel performs competitively with the Mallows kernel in low-dimensional settings, but significantly outperforms it in both optimization performance and computational efficiency as the dimension grows. Extensive evaluations on various permutation optimization benchmarks confirm our hypothesis, demonstrating that the Merge Kernel provides a scalable and more effective solution for Bayesian optimization in high-dimensional permutation spaces, thereby unlocking the potential for tackling previously intractable problems such as large-scale feature ordering and combinatorial neural architecture search.
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.
Builds on5
- BoTorch: A Framework for Efficient Monte-Carlo Bayesian OptimizationMaximilian Balandat, Brian Karrer, Daniel R. Jiang, Samuel Daulton et al.NeurIPS 2020 · 686 citations
- Optimizing Sequential Experimental Design with Deep Reinforcement LearningTom Blau, Edwin V. Bonilla, Iadine Chades, Amir DezfouliICML 2022 · 62 citations
- Bayesian Optimization for Categorical and Category-Specific Continuous InputsDang Nguyen, Sunil Gupta, Santu Rana, Alistair Shilton et al.AAAI 2020 · 59 citations
- Bayesian Optimization over Permutation SpacesAryan Deshwal, Syrine Belakaria, Janardhan Rao Doppa, Dae Hyun KimAAAI 2022 · 27 citations
- Batch Bayesian Optimization on Permutations using the Acquisition Weighted KernelChangYong Oh, Roberto Bondesan, Efstratios Gavves, Max WellingNeurIPS 2022 · 20 citations
Related papers
- Mercer Features for Efficient Combinatorial Bayesian OptimizationAryan Deshwal, Syrine Belakaria, Janardhan Rao DoppaAAAI 2021 · 39 citations
- Omnipresent Yet Overlooked: Heat Kernels in Combinatorial Bayesian OptimizationColin Doumont, Victor Picheny, Viacheslav Borovitskiy, Henry B. MossNeurIPS 2025 · 2 citations
- Scalable First-Order Bayesian Optimization via Structured Automatic DifferentiationSebastian E. Ament, Carla P. GomesICML 2022 · 12 citations
- Think Global and Act Local: Bayesian Optimisation over High-Dimensional Categorical and Mixed Search SpacesXingchen Wan, Vu Nguyen, Huong Ha, Bin Xin Ru et al.ICML 2021 · 79 citations
- Interpretable Neural Architecture Search via Bayesian Optimisation with Weisfeiler-Lehman KernelsBin Xin Ru, Xingchen Wan, Xiaowen Dong, Michael A. OsborneICLR 2021 · 116 citations
