ICLR 2026poster0 citations

Scaling Multi-Task Bayesian Optimization with Large Language Models

Yimeng Zeng, Natalie Maus, Haydn Thomas Jones, Jeffrey Tao, Fangping Wan, Marcelo Der Torossian Torres, Cesar de la Fuente-Nunez, Ryan Marcus

Abstract

In multi-task Bayesian optimization, the goal is to leverage experience from optimizing existing tasks to improve the efficiency of optimizing new ones. While approaches using multi-task Gaussian processes or deep kernel transfer exist, the performance improvement is marginal when scaling beyond a moderate number of tasks. We introduce **BOLT**, an initialization-only transfer strategy that distills prior BO runs into an LLM which proposes candidates for new tasks, while the surrogate at test time remains single-task. The LLM is periodically fine-tuned on top solutions from completed runs, creating a closed loop where better BO outputs yield better initializations over time. This decoupled design scales to roughly 1500 tasks without the saturation observed for shared-surrogate MTBO and adds only a small, amortized overhead relative to the BO inner loops. We evaluate on two domains: database query optimization and antimicrobial peptide design. We demonstrate that LLM-generated initializations steadily improve and accelerate BO, and with sufficient fine-tuning, a few LLM samples often match or surpass full ''from-scratch'' BO with far fewer oracle calls.

Bayesian optimizationlarge language modelsprotein designmeta learningscientific discovery
BibTeX
@inproceedings{
zeng2026scaling,
title={Scaling Multi-Task Bayesian Optimization with Large Language Models},
author={Yimeng Zeng and Natalie Maus and Haydn Thomas Jones and Jeffrey Tao and Fangping Wan and Marcelo Der Torossian Torres and Cesar de la Fuente-Nunez and Ryan Marcus and Osbert Bastani and Jacob R. Gardner},
booktitle={The Fourteenth International Conference on Learning Representations},
year={2026},
url={https://openreview.net/forum?id=ekmUkRYnkN}
}
Scaling Multi-Task Bayesian Optimization with Large Language Models · ICLR 2026