Grey-Box Bayesian Optimization Improves Efficiency with Separability

Joshua E. Hammond, Tyler A. Soderstrom, Brian A. Korgel, Michael Baldea· August 5, 2026 View original

Key takeaways

  • Separating optimization variables into black-box and white-box components can significantly improve efficiency.
  • Bilevel Bayesian optimization offers a robust framework for grey-box problems.
  • The method avoids approximations, leading to more precise constraint satisfaction.
  • It consistently outperforms baselines in terms of regret, iterations, and time.

Who benefits

ManufacturingChemical EngineeringDrug DiscoveryFinancial ModelingLogistics

Summary

This research introduces a bilevel Bayesian optimization approach that leverages the separability of decision variables into black-box and white-box components. By optimizing the black-box variables with an outer Bayesian optimization and solving the white-box subproblem globally, the method achieves better performance and efficiency.

This paper presents a novel approach to grey-box optimization problems, where decision variables can be naturally divided into two categories: black-box variables that feed into an expensive, opaque function, and white-box variables governed by explicit, closed-form equations. The core idea is a bilevel optimization strategy. An outer Bayesian optimization process handles the black-box variables, aiming to optimize the overall objective. Concurrently, an inner global optimization problem addresses the white-box variables, ensuring their constraints are precisely met. This method avoids common pitfalls like penalty functions or approximations. The proposed bilevel Bayesian optimization framework demonstrates superior performance across a suite of 13 benchmark problems. It consistently achieves lower regret, requires fewer iterations, and reduces overall wall-clock time compared to existing methods. The advantages hold true regardless of the initial dataset size, exploration parameters, or the specific inner-solver chosen, highlighting its robustness and efficiency in complex optimization scenarios.

Why it matters

Professionals dealing with complex systems where some parameters are well-understood (white-box) and others are opaque (black-box) can use this method to optimize processes more efficiently and accurately. It offers a robust way to improve outcomes in engineering design, scientific discovery, and resource allocation.

How to implement this in your domain

  1. 1Identify separable variables in your optimization problem, distinguishing between black-box and white-box components.
  2. 2Implement an outer Bayesian optimization loop for the black-box variables, using a Gaussian process surrogate.
  3. 3Integrate an inner global optimizer to solve the white-box subproblem, ensuring exact constraint satisfaction.
  4. 4Test the bilevel optimization framework on your specific problem, comparing its performance against current methods.
  5. 5Adjust exploration parameters and inner-solver choices to fine-tune the optimization for your application.

Original post by Joshua E. Hammond, Tyler A. Soderstrom, Brian A. Korgel, Michael Baldea

"arXiv:2608.03045v1 Announce Type: new Abstract: We consider grey-box optimization problems where the decision variables naturally partition into black-box variables (as arguments to an expensive black-box function) and white-box variables, governed by a set of explicit, closed-fo…"

View on X

Originally posted by Joshua E. Hammond, Tyler A. Soderstrom, Brian A. Korgel, Michael Baldea on X · view source

Want to go deeper?

Turn these trends into skills with Learnijoy's hands-on AI & tech courses.

Explore courses