Robust Bayesian Optimization via Localized Online Conformal Prediction

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Kim, Dongwon, Zecchin, Matteo, Park, Sangwoo, Kang, Joonhyuk, Simeone, Osvaldo
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866915220175192064
author Kim, Dongwon
Zecchin, Matteo
Park, Sangwoo
Kang, Joonhyuk
Simeone, Osvaldo
author_facet Kim, Dongwon
Zecchin, Matteo
Park, Sangwoo
Kang, Joonhyuk
Simeone, Osvaldo
contents Bayesian optimization (BO) is a sequential approach for optimizing black-box objective functions using zeroth-order noisy observations. In BO, Gaussian processes (GPs) are employed as probabilistic surrogate models to estimate the objective function based on past observations, guiding the selection of future queries to maximize utility. However, the performance of BO heavily relies on the quality of these probabilistic estimates, which can deteriorate significantly under model misspecification. To address this issue, we introduce localized online conformal prediction-based Bayesian optimization (LOCBO), a BO algorithm that calibrates the GP model through localized online conformal prediction (CP). LOCBO corrects the GP likelihood based on predictive sets produced by LOCBO, and the corrected GP likelihood is then denoised to obtain a calibrated posterior distribution on the objective function. The likelihood calibration step leverages an input-dependent calibration threshold to tailor coverage guarantees to different regions of the input space. Under minimal noise assumptions, we provide theoretical performance guarantees for LOCBO's iterates that hold for the unobserved objective function. These theoretical findings are validated through experiments on synthetic and real-world optimization tasks, demonstrating that LOCBO consistently outperforms state-of-the-art BO algorithms in the presence of model misspecification.
format Preprint
id arxiv_https___arxiv_org_abs_2411_17387
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Robust Bayesian Optimization via Localized Online Conformal Prediction
Kim, Dongwon
Zecchin, Matteo
Park, Sangwoo
Kang, Joonhyuk
Simeone, Osvaldo
Machine Learning
Signal Processing
Bayesian optimization (BO) is a sequential approach for optimizing black-box objective functions using zeroth-order noisy observations. In BO, Gaussian processes (GPs) are employed as probabilistic surrogate models to estimate the objective function based on past observations, guiding the selection of future queries to maximize utility. However, the performance of BO heavily relies on the quality of these probabilistic estimates, which can deteriorate significantly under model misspecification. To address this issue, we introduce localized online conformal prediction-based Bayesian optimization (LOCBO), a BO algorithm that calibrates the GP model through localized online conformal prediction (CP). LOCBO corrects the GP likelihood based on predictive sets produced by LOCBO, and the corrected GP likelihood is then denoised to obtain a calibrated posterior distribution on the objective function. The likelihood calibration step leverages an input-dependent calibration threshold to tailor coverage guarantees to different regions of the input space. Under minimal noise assumptions, we provide theoretical performance guarantees for LOCBO's iterates that hold for the unobserved objective function. These theoretical findings are validated through experiments on synthetic and real-world optimization tasks, demonstrating that LOCBO consistently outperforms state-of-the-art BO algorithms in the presence of model misspecification.
title Robust Bayesian Optimization via Localized Online Conformal Prediction
topic Machine Learning
Signal Processing
url https://arxiv.org/abs/2411.17387