Adaptive Kernel Design for Bayesian Optimization Is a Piece of CAKE with LLMs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Suwandi, Richard Cornelius, Yin, Feng, Wang, Juntao, Li, Renjie, Chang, Tsung-Hui, Theodoridis, Sergios
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911171116793856
author Suwandi, Richard Cornelius
Yin, Feng
Wang, Juntao
Li, Renjie
Chang, Tsung-Hui
Theodoridis, Sergios
author_facet Suwandi, Richard Cornelius
Yin, Feng
Wang, Juntao
Li, Renjie
Chang, Tsung-Hui
Theodoridis, Sergios
contents The efficiency of Bayesian optimization (BO) relies heavily on the choice of the Gaussian process (GP) kernel, which plays a central role in balancing exploration and exploitation under limited evaluation budgets. Traditional BO methods often rely on fixed or heuristic kernel selection strategies, which can result in slow convergence or suboptimal solutions when the chosen kernel is poorly suited to the underlying objective function. To address this limitation, we propose a freshly-baked Context-Aware Kernel Evolution (CAKE) to enhance BO with large language models (LLMs). Concretely, CAKE leverages LLMs as the crossover and mutation operators to adaptively generate and refine GP kernels based on the observed data throughout the optimization process. To maximize the power of CAKE, we further propose BIC-Acquisition Kernel Ranking (BAKER) to select the most effective kernel through balancing the model fit measured by the Bayesian information criterion (BIC) with the expected improvement at each iteration of BO. Extensive experiments demonstrate that our fresh CAKE-based BO method consistently outperforms established baselines across a range of real-world tasks, including hyperparameter optimization, controller tuning, and photonic chip design. Our code is publicly available at https://github.com/richardcsuwandi/cake.
format Preprint
id arxiv_https___arxiv_org_abs_2509_17998
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Kernel Design for Bayesian Optimization Is a Piece of CAKE with LLMs
Suwandi, Richard Cornelius
Yin, Feng
Wang, Juntao
Li, Renjie
Chang, Tsung-Hui
Theodoridis, Sergios
Machine Learning
Artificial Intelligence
The efficiency of Bayesian optimization (BO) relies heavily on the choice of the Gaussian process (GP) kernel, which plays a central role in balancing exploration and exploitation under limited evaluation budgets. Traditional BO methods often rely on fixed or heuristic kernel selection strategies, which can result in slow convergence or suboptimal solutions when the chosen kernel is poorly suited to the underlying objective function. To address this limitation, we propose a freshly-baked Context-Aware Kernel Evolution (CAKE) to enhance BO with large language models (LLMs). Concretely, CAKE leverages LLMs as the crossover and mutation operators to adaptively generate and refine GP kernels based on the observed data throughout the optimization process. To maximize the power of CAKE, we further propose BIC-Acquisition Kernel Ranking (BAKER) to select the most effective kernel through balancing the model fit measured by the Bayesian information criterion (BIC) with the expected improvement at each iteration of BO. Extensive experiments demonstrate that our fresh CAKE-based BO method consistently outperforms established baselines across a range of real-world tasks, including hyperparameter optimization, controller tuning, and photonic chip design. Our code is publicly available at https://github.com/richardcsuwandi/cake.
title Adaptive Kernel Design for Bayesian Optimization Is a Piece of CAKE with LLMs
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2509.17998