Compactness Matters: Improving Bayesian Optimization Efficiency of Materials Formulations through Invariant Search Spaces

03 February 2023, Version 3
This content is a preprint and has not undergone peer review at the time of posting.

Abstract

Would you rather search for a line inside a cube or a point inside a square? Physics-based simulations and wet-lab experiments often have symmetries (degeneracies) that allow reducing problem dimensionality or search space, but constraining these degeneracies is often unsupported or difficult to implement in many optimization packages, requiring additional time and expertise. So, are the possible improvements in efficiency worth the cost of implementation? We demonstrate that the compactness of a search space (to what extent and how degenerate solutions and non-solutions are removed) affects Bayesian optimization search efficiency. Here, we use the Adaptive Experimentation (Ax) Platform by Meta and a physics-based particle packing simulation with eight or nine tunable parameters, depending on the search space compactness. These parameters represent three truncated log-normal distributions of particle sizes which exhibit compositional-invariance and permutation-invariance characteristic of formulation problems (e.g., chemical formulas, composite materials, alloys). We assess a total of four search space types which range from none up to both constraint types imposed simultaneously. In general, the removal of degeneracy through problem reformulation (as seen by the optimizers surrogate model) improves optimization efficiency. We recommend that optimization practitioners in the physical sciences carefully consider the trade-off between implementation cost and search efficiency before running expensive optimization campaigns.

Keywords

Bayesian optimization
sequential learning
adaptive design
active learning
concurrency
AxSearch
RayTune
Ax platform
Python
optimization
constrained optimization
constrained materials optimization
constrained Bayesian optimization

Supplementary materials

Title
Description
Actions
Title
Compactness Matters: Improving Bayesian Optimization Efficiency of Materials Formulations through Invariant Search Spaces: Supporting Information
Description
Supporting information including approximating particle size distributions as log-normal distributions, size invariance, simulation variation, solutions visualized as summed distributions, feature importances, best objective vs. iteration plots, cross-validation results, 2D contours through model parameter space, and Ax SearchSpace objects.
Actions

Supplementary weblinks

Comments

Comments are not moderated before they are posted, but they can be removed by the site moderators if they are found to be in contravention of our Commenting Policy [opens in a new tab] - please read this policy before you post. Comments should be used for scholarly discussion of the content in question. You can find more information about how to use the commenting feature here [opens in a new tab] .
This site is protected by reCAPTCHA and the Google Privacy Policy [opens in a new tab] and Terms of Service [opens in a new tab] apply.