用变分自编码器降维,提升贝叶斯优化在高维黑箱问题中的效率
Dimensionality Reduction Techniques for Global Bayesian Optimisation
- 通过变分自编码器构建结构化低维隐空间,实现高效贝叶斯优化
- 在多个测试任务上,基于结构化隐空间的优化性能显著优于传统方法
- 首次将序列域缩减策略引入隐空间贝叶斯优化,支持GPU加速
贝叶斯优化(BO)是处理无导数信息、样本代价高的黑箱优化问题的先进方法。然而其全局可扩展性仍面临挑战。本文研究隐空间贝叶斯优化(LSBO),通过维度约简在低维子空间中执行优化。相比早期使用随机投影的方法,本文采用变分自编码器(VAE)以应对更复杂的数据结构和通用降维任务。基于Grosnit等(2021)的工作,我们分析了基于VAE的LSBO框架,重点改进了VAE重训练与深度度量损失的设计,针对分子生成等任务原方案存在的问题进行修正,并重新设计算法以适用于更广泛的优化场景。数值实验表明,结构化隐流形能显著提升优化性能。此外,本文探讨了在该框架下使用Matérn-5/2核函数的高斯过程表现。还首次将序列域缩减(SDR)这一标准优化加速策略集成至基于BOtch的GPU环境中,分别应用于原始空间与VAE生成的隐空间。
原文摘要 · Abstract (English)
Bayesian Optimisation (BO) is a state-of-the-art global optimisation technique for black-box problems where derivative information is unavailable, and sample efficiency is crucial. However, improving the general scalability of BO has proved challenging. Here, we explore Latent Space Bayesian Optimisation (LSBO), that applies dimensionality reduction to perform BO in a reduced-dimensional subspace. While early LSBO methods used (linear) random projections (Wang et al., 2013), we employ Variational Autoencoders (VAEs) to manage more complex data structures and general DR tasks. Building on Grosnit et. al. (2021), we analyse the VAE-based LSBO framework, focusing on VAE retraining and deep metric loss. We suggest a few key corrections in their implementation, originally designed for tasks such as molecule generation, and reformulate the algorithm for broader optimisation purposes. Our numerical results show that structured latent manifolds improve BO performance. Additionally, we examine the use of the Matérn-$\frac{5}{2}$ kernel for Gaussian Processes in this LSBO context. We also integrate Sequential Domain Reduction (SDR), a standard global optimization efficiency strategy, into BO. SDR is included in a GPU-based environment using \textit{BoTorch}, both in the original and VAE-generated latent spaces, marking the first application of SDR within LSBO.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。