Global Optimisation of Black-Box Functions with Generative Models in the Wasserstein Space

Abstract

We propose a new uncertainty estimator for gradient-free optimisation of black-box simulators using deep generative surrogate models. Optimisation of these simulators is especially challenging for stochastic simulators and higher dimensions. To address these issues, we utilise a deep generative surrogate approach to model the black box response for the entire parameter space. We then leverage this knowledge to estimate the proposed uncertainty based on the Wasserstein distance - the Wasserstein uncertainty. This approach is employed in a posterior agnostic gradient-free optimisation algorithm that minimises regret over the entire parameter space. A series of tests were conducted to demonstrate that our method is more robust to the shape of both the black box function and the stochastic response of the black box than state-of-the-art methods, such as efficient global optimisation with a deep Gaussian process surrogate.

European Conference on Artificial Intelligence 2024 Main Track

References in corpus (10)

• Practical Bayesian Optimization of Machine Learning Algorithms
• A Brief Introduction to PYTHIA 8.1
• Convergence rates of efficient global optimization algorithms
• Accelerating Science with Generative Adversarial Networks: An Application to 3D Particle Showers in Multi-Layer Calorimeters
• Deep Gaussian Processes
• CosmoGAN: creating high-fidelity weak lensing convergence maps using Generative Adversarial Networks
• The active muon shield in the SHiP experiment
• Active Learning of Linear Embeddings for Gaussian Processes
• Black-Box Optimization with Local Generative Surrogates
• Diffusion Models for Black-Box Optimization