Consensus-Based Bayesian Optimization
- Consensus-Based Bayesian Optimization is a framework where multiple agents collaboratively optimize black-box functions via consensus protocols and distributed experimentation.
- It employs adaptive acquisition functions and consensus protocols, such as uniform-decay and leader-driven scheduling, to balance collective learning and individual personalization.
- Empirical evaluations and regret analyses demonstrate that CBO achieves accelerated convergence, robust scalability, and efficient optimization in heterogeneous settings.
Consensus-Based Bayesian Optimization (CBO), an extension of classical Bayesian Optimization (BO), is a framework where multiple agents collaboratively optimize black-box objective functions by reaching consensus on experimental decisions. The methodology leverages distributed experimentation, information sharing via consensus protocols, and transitional collaboration–personalization schemes to accelerate optimal design in both homogeneous and heterogeneous settings. CBO techniques, including the Collaborative Bayesian Optimization via Consensus (CBOC) and Social Bayesian Optimization (SBO) variants, provide rigorous theoretical guarantees and have demonstrated empirical efficacy in complex optimization domains (Yue et al., 2023, Adachi et al., 11 Feb 2025).
1. Problem Formulation and Notation
Consensus-based Bayesian optimization addresses multi-agent scenarios with clients, each with a latent black-box objective . The goal for each client is to identify given a strictly bounded experimental budget .
Each client maintains its local dataset , where after rounds . Learning and inference proceed via Gaussian Process (GP) surrogates for each , with GP kernel, noise, and hyperparameters learned or specified a priori:
- where and .
Given , the GP posterior mean and variance at are:
2. Consensus Protocols and Collaboration Mechanisms
Consensus in CBO is achieved by clients exchanging sampling proposals and aggregating these collaboratively. At each round, every client maximizes a utility/acquisition function (e.g., Expected Improvement, EI) to propose a local sampling point . The collection of proposals forms .
The core consensus mechanism employs a doubly-stochastic matrix (, ), yielding the consensus update:
so that
for each agent (Yue et al., 2023).
Transitional weighting schedules interpolate from full collaboration to independence, allowing strong early-stage knowledge sharing and late-stage personalization:
- Uniform-decay: , followed by diag/off-diag updates such that .
- Leader-driven: At each , one client is the leader (highest reward), with reweighted accordingly, always preserving doubly-stochasticity (Yue et al., 2023).
3. Acquisition and Sampling Strategies
CBO methods are acquisition-agnostic. The canonical example is Expected Improvement (EI):
where and (Yue et al., 2023).
Consensus is integrated at the sampling stage:
There is flexibility to use other acquisition functions including UCB, KG, PI, or qEI. The consensus protocol operates independently of the acquisition function (Yue et al., 2023).
4. Algorithmic Workflow and Pseudocode
A prototypical CBOC iteration is:
- Consensus step: Compute by aggregating local proposals via .
- Experiment: Query and update .
- GP update: Update surrogate using .
- Acquisition maximization: Find .
- Consensus update: Adjust following the uniform-decay or leader-driven schedule.
All clients perform these steps in parallel. Acquisition optimization remains the principal computational bottleneck; the consensus step itself scales as per agent (Yue et al., 2023).
5. Regret Analysis and Theoretical Guarantees
In the homogeneous case (), under squared-exponential kernels and any sequence of doubly-stochastic , CBO maintains classical Bayesian Optimization regret bounds:
where is client 's cumulative regret (Yue et al., 2023). The bound stems from GP posterior concentration, acquisition function analysis, and information-theoretic arguments. The consensus step does not harm these rates as long as is properly controlled and client iterates remain bounded.
For Social Bayesian Optimization, in collective decision-making with feedback corrupted by social influence, regret and the number of expensive "private" votes required are sublinear (in ) provided the graph is estimable and appropriate decay is set for private vote acquisition. The social graph can be estimated as , and utility estimation converges as (Adachi et al., 11 Feb 2025).
6. Empirical Validation
Simulation Benchmarks
CBOC was evaluated on Levy, Shekel, Branin, Ackley, and Hartmann functions with both homogeneous and heterogeneous agent objectives. The primary baseline comparisons were Individual BO and federated BO (FedBO):
| Task | K | CBOC-L Gap | Individual Gap | FedBO Gap |
|---|---|---|---|---|
| Levy-2 (homog.) | 5 | ≈0.993 | 0.931 | 0.990 |
| Levy-4 (heterog.) | 10 | ≈0.984 | 0.933 | 0.940 |
| Shekel-10 (homog.) | 20 | ≈0.592 | 0.335 | 0.535 |
CBOC demonstrated uniform acceleration, with performance gain amplifying as increased. The method rapidly directed clients into their optimal basins, especially in heterogeneous settings (Yue et al., 2023).
Real-World Case Study
In a collaborative sensor design task using three FEA simulators, CBOC-L achieved higher mean () and faster convergence compared to non-collaborative runs (), with about 30% reduction in convergence time and lower variance (Yue et al., 2023).
Social Consensus Applications
SBO was validated across tasks including thermal comfort, team-building, hotel selection, and energy trading. Notably, sublinear growth in private votes (often terminating after $10$–$20$ rounds) was observed, and regret metrics improved upon both private-only and oracle-A baselines. SBO demonstrated robustness to aggregator parameters and various social graph structures (Adachi et al., 11 Feb 2025).
7. Practical Implementation Considerations
- Consensus matrix (): Use uniform-decay if no prior information, or leader-driven if trust/reliability differences exist. Doubly-stochasticity is critical.
- Communication frequency: Minimal consensus update per experiment is sufficient; skipping communication between some steps is permissible if varies slowly.
- Scalability: Storage overhead is ; use sparse communications if needed.
- Privacy: Only share proposals ; retain private reward/observed data and GP hyperparameters.
- Acquisition: Any acquisition function integrates directly; consensus operates agnostic to the choice.
- Stopping: Global budget or local EI threshold.
- Reproducibility: Source code is available in the referenced open-source repository (Yue et al., 2023).
8. Extensions: Social Influence and Collective Consensus
The SBO framework generalizes consensus-based BO to settings with social influence, capturing both true and observed (graph-convolved) utilities. A dual-voting system (public and private votes) enables identification and debiasing of the social graph, enabling efficient social-influence-free consensus optimization with provable regret and sample complexity guarantees (Adachi et al., 11 Feb 2025).
SBO employs kernelized GP surrogates, upper-confidence–bound acquisitions, and decouples costly private queries as the social graph becomes well-estimated, lowering the practical burden of high-fidelity feedback.
Comparison of CBOC and SBO Approaches
| Feature | CBOC (Yue et al., 2023) | SBO (Adachi et al., 11 Feb 2025) |
|---|---|---|
| Collaboration protocol | Doubly-stochastic consensus | Dual-voting, social-graph estimation |
| Objective(s) | Personalized per client | Aggregated social consensus |
| Surrogate model | Local GPs | GP over utilities (with graph) |
| Acquisition | EI (agnostic) | UCB-style over social aggregator |
| Empirical context | Simulation, sensor design | Decision-making, voting tasks |
CBO methods formalize a principled class of distributed, collaborative BO techniques. They allow efficient transfer of sampling knowledge, adaptively balance knowledge sharing and personalization, and are extensible to consensus under nontrivial social dynamics, all supported by information-theoretic regret analysis and scalable empirical evidence (Yue et al., 2023, Adachi et al., 11 Feb 2025).