Return of the latent space cowboys: rethinking the use of VAEs in Bayesian optimisation over structured spaces

Date:

January 21, 2025

Author:

Abstract 

Bayesian optimization in the latent space of generative models has become a powerful method for exploring structured search spaces, such as molecular design. Instead of directly optimizing over complex, high-dimensional discrete structures, this approach maps structured inputs into a fixed-size latent space. Within this space, standard surrogate models and gradient-based optimization routines can be effectively applied. However, because these generative models are not trained with specific downstream tasks in mind, latent space optimization can exhibit problematic behaviours. In this talk, we explore an alternative method that decouples the surrogate and generative models. Instead of tightly integrating Gaussian Processes with Variational Autoencoders, our method trains these models separately and combines them through a simple Bayesian update. The aim is to develop a sampling strategy that efficiently identifies candidate structures with a high likelihood of achieving the target objective.

Notes

Personal website can be found here.

Share on social media

Share on social media

Share on social media

Related Seminars

Linear combinations of latents in generative models: subspaces and beyond

Erik Bodin - University of Cambridge

Mar 13, 2025

Linear combinations of latents in generative models: subspaces and beyond

Erik Bodin - University of Cambridge

Mar 13, 2025

Return of the latent space cowboys: rethinking the use of VAEs in Bayesian optimisation over structured spaces

Henry Moss - University of Cambridge, Lancaster University

Jan 21, 2025

Return of the latent space cowboys: rethinking the use of VAEs in Bayesian optimisation over structured spaces

Henry Moss - University of Cambridge, Lancaster University

Jan 21, 2025

Advancing sequential decision-making: efficient querying in clustering and best of both worlds for contextual bandits

Yuko Kuroki - CENTAI Institute

Oct 10, 2024

Advancing sequential decision-making: efficient querying in clustering and best of both worlds for contextual bandits

Yuko Kuroki - CENTAI Institute

Oct 10, 2024

AI in drug discovery - from model to process, from academic publication to decision-making

Andreas Bender - University of Cambridge

Sep 19, 2024

AI in drug discovery - from model to process, from academic publication to decision-making

Andreas Bender - University of Cambridge

Sep 19, 2024