Variational Search Distributions
Not on the map. Why?
Abstract (arXiv)
We develop VSD, a method for conditioning a generative model of discrete, combinatorial designs on a rare desired class by efficiently evaluating a black-box (e.g. experiment, simulation) in a batch sequential manner. We call this task active generation; we formalize active generation's requirements and desiderata, and formulate a solution via variational inference. VSD uses off-the-shelf gradient based optimization routines, can learn powerful generative models for desirable designs, and can take advantage of scalable predictive models. We derive asymptotic convergence rates for learning the true conditional generative distribution of designs with certain configurations of our method. After illustrating the generative model on images, we empirically demonstrate that VSD can outperform existing baseline methods on a set of real sequence-design problems in various protein and DNA/RNA engineering tasks.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Builds on 49 of its 93 references that are in the map. First posted on arXiv 10 Sep 2024.
Builds on
49 of 93 references.
References · 49
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 793
- 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 762
- 2019Wide Neural Networks of Any Depth Evolve as Linear Models Under Gradient Descentcited by 732
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 649
- 2018Machine Learning for Combinatorial Optimization: a Methodological Tour d'Horizoncited by 612
- 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
- 2017On Kernelized Multi-armed Banditscited by 348
- 2013Black Box Variational Inferencecited by 317
- 2019Monte Carlo Gradient Estimation in Machine Learningcited by 281
- 2021On the Origin of Implicit Regularization in Stochastic Gradient Descentcited by 183
37 more
- 2022Biological Sequence Design with GFlowNetscited by 172
- 2020On Information Gain and Regret Bounds in Gaussian Process Banditscited by 156
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 2020Implicit Gradient Regularizationcited by 145
- 2020On the linearity of large non-linear models: when and why the tangent kernel is constantcited by 129
- 2022A Survey of Controllable Text Generation using Transformer-based Pre-trained Language Modelscited by 128
- 2020Sample-Efficient Optimization in the Latent Space of Deep Generative Models via Weighted Retrainingcited by 121
- 2020Infinite attention: NNGP and NTK for deep attention networkscited by 117
- 2019Conditioning by adaptive sampling for robust designcited by 110
- 2022Accelerating Bayesian Optimization for Biological Sequence Design with Denoising Autoencoderscited by 99
- 2022Design-Bench: Benchmarks for Data-Driven Offline Model-Based Optimizationcited by 96
- 2021Conservative Objective Models for Effective Offline Model-Based Optimizationcited by 87
- 2020Deep Neural Tangent Kernel and Laplace Kernel Have the Same RKHScited by 74
- 2022Formal Algorithms for Transformerscited by 70
- 2020AdaLead: A simple and robust adaptive greedy search algorithm for sequence designcited by 63
- 2017The reparameterization trick for acquisition functionscited by 61
- 2016Truncated Variance Reduction: A Unified Approach to Bayesian Optimization and Level-Set Estimationcited by 40
- 2022Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic Reparameterizationcited by 35
- 2023BayesDAG: Gradient-Based Posterior Inference for Causal Discoverycited by 33
- 2018Design by adaptive samplingcited by 31
- 2021BORE: Bayesian Optimization by Density-Ratio Estimationcited by 26
- 2022Characterizing the Spectrum of the NTK via a Power Series Expansioncited by 22
- 2024A survey and benchmark of high-dimensional Bayesian optimization of discrete sequencescited by 22
- 2023Kernelized Reinforcement Learning with Order Optimal Regret Boundscited by 20
- 2023Improving Protein Optimization with Smoothed Fitness Landscapescited by 20
- 2022A General Recipe for Likelihood-free Bayesian Optimizationcited by 19
- 2021Non-asymptotic approximations of neural networks by Gaussian processescited by 18
- 2017Streaming kernel regression with provably adaptive mean, variance, and regularizationcited by 15
- 2023Posterior Sampling-Based Bayesian Optimization with Tighter Bayesian Regret Boundscited by 15
- 2018Singular Value Decomposition of Operators on Reproducing Kernel Hilbert Spacescited by 14
- 2022Batch Bayesian optimisation via density-ratio estimation with guaranteescited by 9
- 2023Learning Regions of Interest for Bayesian Optimization with Adaptive Level-Set Estimationcited by 8
- 2024Closed-Form Test Functions for Biophysical Sequence Optimization Algorithmscited by 7
- 2015Discovering Valuable Items from Massive Datacited by 6
- 2018Stochastic Variational Optimizationcited by 5
- 2019A view of Estimation of Distribution Algorithms through the lens of Expectation-Maximizationcited by 5
- 2024A Continuous Relaxation for Discrete Bayesian Optimizationcited by 4
Built on by
Citing papers per year: 2024: 1; 2025: 4; 2026: 1.
6 papers.
- 2024Generalists vs. Specialists: Evaluating LLMs on Highly-Constrained Biophysical Sequence Optimization Taskscited by 8
- 2025Variational Transdimensional Inferencecited by 2
- 2025Generative Bayesian Optimization: Generative Models as Acquisition Functionscited by 2
- 2025Amortized Active Generation of Pareto Setscited by 1
- 2026Kernel-based guarantees for nonlinear parametric models in Bayesian optimizationcited by 1
- 2025Sample-Efficient Optimisation over the Outputs of Generative Modelscited by 0
- 2026Kernel-based guarantees for nonlinear parametric models in Bayesian optimization13 May 2026
- 2025Generative Bayesian Optimization: Generative Models as Acquisition Functions29 Oct 2025
- 2025Amortized Active Generation of Pareto Sets23 Oct 2025
- 2025Sample-Efficient Optimisation over the Outputs of Generative Models28 Sep 2025
- 2025Variational Transdimensional Inference5 Jun 2025
- 2024Generalists vs. Specialists: Evaluating LLMs on Highly-Constrained Biophysical Sequence Optimization Tasks29 Oct 2024