English

Language Model Embeddings Can Be Sufficient for Bayesian Optimization

Machine Learning 2025-10-10 v3 Artificial Intelligence

Abstract

Bayesian Optimization is ubiquitous in experimental design and black-box optimization for improving search efficiency. However, most existing approaches rely on regression models which are limited to fixed search spaces and structured, tabular input features. This paper explores the use of LLM embeddings over string inputs for in-context regression in Bayesian Optimization. Our results show that representing inputs as strings enables general-purpose regression across diverse domains, including synthetic, combinatorial, and hyperparameter optimization. Furthermore, our approach achieves optimization performance comparable to state-of-the-art Gaussian Process-based methods such as Google Vizier, and demonstrates potential for broader and more flexible applications.

Keywords

Cite

@article{arxiv.2410.10190,
  title  = {Language Model Embeddings Can Be Sufficient for Bayesian Optimization},
  author = {Tung Nguyen and Qiuyi Zhang and Bangding Yang and Chansoo Lee and Jorg Bornschein and Yingjie Miao and Sagi Perel and Yutian Chen and Xingyou Song},
  journal= {arXiv preprint arXiv:2410.10190},
  year   = {2025}
}

Comments

Code can be found in https://github.com/google-research/optformer/tree/main/optformer/embed_then_regress

R2 v1 2026-06-28T19:20:04.835Z