Collaborative Contextual Bayesian Optimization

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chang, Chih-Yu, Chen, Qiyuan, Gao, Tianhan, Fenning, David, Okwudire, Chinedum, Dasgupta, Neil, Lu, Wei, Kontar, Raed Al
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915947040735232
author Chang, Chih-Yu
Chen, Qiyuan
Gao, Tianhan
Fenning, David
Okwudire, Chinedum
Dasgupta, Neil
Lu, Wei
Kontar, Raed Al
author_facet Chang, Chih-Yu
Chen, Qiyuan
Gao, Tianhan
Fenning, David
Okwudire, Chinedum
Dasgupta, Neil
Lu, Wei
Kontar, Raed Al
contents Discovering optimal designs through sequential data collection is essential in many real-world applications. While Bayesian Optimization (BO) has achieved remarkable success in this setting, growing attention has recently turned to context-specific optimal design, formalized as Contextual Bayesian Optimization (CBO). Unlike BO, CBO is inherently more challenging as it must approximate an entire mapping from the context space to its corresponding optimal design, requiring simultaneous exploration across contexts and exploitation within each. In many modern applications, such tasks arise across multiple potentially heterogeneous but related clients, where collaboration can significantly improve learning efficiency. We propose CCBO, Collaborative Contextual Bayesian Optimization, a unified framework enabling multiple clients to jointly perform CBO with controllable contexts, supporting both online collaboration and offline initialization from peers' historical beliefs, with an optional privacy-preserving communication mechanism. We establish sublinear regret guarantees and demonstrate, through extensive simulations and a real-world hot rolling application, that CCBO achieves substantial improvements over existing approaches even under client heterogeneity. The code to reproduce the results can be found at https://github.com/cchihyu/Collaborative-Contextual-Bayesian-Optimization
format Preprint
id arxiv_https___arxiv_org_abs_2604_18912
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Collaborative Contextual Bayesian Optimization
Chang, Chih-Yu
Chen, Qiyuan
Gao, Tianhan
Fenning, David
Okwudire, Chinedum
Dasgupta, Neil
Lu, Wei
Kontar, Raed Al
Machine Learning
Methodology
Discovering optimal designs through sequential data collection is essential in many real-world applications. While Bayesian Optimization (BO) has achieved remarkable success in this setting, growing attention has recently turned to context-specific optimal design, formalized as Contextual Bayesian Optimization (CBO). Unlike BO, CBO is inherently more challenging as it must approximate an entire mapping from the context space to its corresponding optimal design, requiring simultaneous exploration across contexts and exploitation within each. In many modern applications, such tasks arise across multiple potentially heterogeneous but related clients, where collaboration can significantly improve learning efficiency. We propose CCBO, Collaborative Contextual Bayesian Optimization, a unified framework enabling multiple clients to jointly perform CBO with controllable contexts, supporting both online collaboration and offline initialization from peers' historical beliefs, with an optional privacy-preserving communication mechanism. We establish sublinear regret guarantees and demonstrate, through extensive simulations and a real-world hot rolling application, that CCBO achieves substantial improvements over existing approaches even under client heterogeneity. The code to reproduce the results can be found at https://github.com/cchihyu/Collaborative-Contextual-Bayesian-Optimization
title Collaborative Contextual Bayesian Optimization
topic Machine Learning
Methodology
url https://arxiv.org/abs/2604.18912