Accelerating Bayesian Optimization for Biological Sequence Design with Denoising Autoencoders
Not on the map. Why?
Abstract (arXiv)
Bayesian optimization (BayesOpt) is a gold standard for query-efficient continuous optimization. However, its adoption for drug design has been hindered by the discrete, high-dimensional nature of the decision variables. We develop a new approach (LaMBO) which jointly trains a denoising autoencoder with a discriminative multi-task Gaussian process head, allowing gradient-based optimization of multi-objective acquisition functions in the latent space of the autoencoder. These acquisition functions allow LaMBO to balance the explore-exploit tradeoff over multiple design rounds, and to balance objective tradeoffs by optimizing sequences at many different points on the Pareto frontier. We evaluate LaMBO on two small-molecule design tasks, and introduce new tasks optimizing in silico and in vitro properties of large-molecule fluorescent proteins. In our experiments LaMBO outperforms genetic optimizers and does not require a large pretraining corpus, demonstrating that BayesOpt is practical and effective for biological sequence design.
Key facts. Cited by 99 papers in the map (data through 7 Oct 2026). Builds on 43 of its 90 references that are in the map. First posted on arXiv 23 Mar 2022.
Builds on
43 of 90 references.
References · 43
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Automatic chemical design using a data-driven continuous representation of moleculescited by 762
- 2018A Tutorial on Bayesian Optimizationcited by 734
- 2018Junction Tree Variational Autoencoder for Molecular Graph Generationcited by 647
- 2018GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Accelerationcited by 603
- 2015Deep Kernel Learningcited by 440
- 2013Gaussian Processes for Big Datacited by 377
- 2019Scalable Global Optimization via Local Bayesian Optimizationcited by 359
- 2020ChemBERTa: Large-Scale Self-Supervised Pretraining for Molecular Property Predictioncited by 332
- 2021Therapeutics Data Commons: Machine Learning Datasets and Tasks for Drug Discovery and Developmentcited by 258
- 2019Evaluating Protein Transfer Learning with TAPEcited by 229
- 2020pymoo: Multi-objective Optimization in Pythoncited by 196
31 more
- 2018Deterministic Non-Autoregressive Neural Sequence Modeling by Iterative Refinementcited by 194
- 2013Bayesian Optimization in a Billion Dimensions via Random Embeddingscited by 178
- 2020Differentiable Expected Hypervolume Improvement for Parallel Multi-Objective Bayesian Optimizationcited by 176
- 2021Bayesian Optimization is Superior to Random Search for Machine Learning Hyperparameter Tuning: Analysis of the Black-Box Optimization Challenge 2020cited by 146
- 2020Sample-Efficient Optimization in the Latent Space of Deep Generative Models via Weighted Retrainingcited by 121
- 2017Multi-fidelity Bayesian Optimisation with Continuous Approximationscited by 120
- 2021Parallel Bayesian Optimization of Multiple Noisy Objectives with Expected Hypervolume Improvementcited by 115
- 2017Constrained Bayesian Optimization with Noisy Experimentscited by 103
- 2016Stochastic Variational Deep Kernel Learningcited by 92
- 2019Practical Multi-fidelity Bayesian Optimization for Hyperparameter Tuningcited by 91
- 2021Multi-Objective Bayesian Optimization over High-Dimensional Search Spacescited by 87
- 2021Iterative Refinement Graph Neural Network for Antibody Sequence-Structure Co-designcited by 82
- 2017The Uncertainty Bellman Equation and Explorationcited by 81
- 2022Local Latent Space Bayesian Optimization over Structured Inputscited by 78
- 2019Augmenting Genetic Algorithms with Deep Neural Networks for Exploring the Chemical Spacecited by 66
- 2019Latent-Variable Non-Autoregressive Neural Machine Translation with Deterministic Inference Using a Delta Posteriorcited by 58
- 2021High-Dimensional Bayesian Optimisation with Variational Autoencoders and Deep Metric Learningcited by 54
- 2021Deciphering antibody affinity maturation with language models and weakly supervised learningcited by 52
- 2020MIMOSA: Multi-constraint Molecule Sampling for Molecule Optimizationcited by 48
- 2021Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spacescited by 45
- 2021Amortized Tree Generation for Bottom-up Synthesis Planning and Synthesizable Molecular Designcited by 44
- 2020Efficient Nonmyopic Bayesian Optimization via One-Shot Multi-Step Treescited by 41
- 2020Bayesian Optimization of Risk Measurescited by 35
- 2021Bayesian Optimization with High-Dimensional Outputscited by 32
- 2020Population-Based Black-Box Optimization for Biological Sequence Designcited by 29
- 2020Autonomous discovery in the chemical sciences part II: Outlookcited by 26
- 2021Conditioning Sparse Variational Gaussian Processes for Online Decision-makingcited by 23
- 2022AntBO: Towards Real-World Automated Antibody Design with Combinatorial Bayesian Optimisationcited by 14
- 2015Unbounded Bayesian Optimization via Regularizationcited by 13
- 2021Optimizing High-Dimensional Physics Simulations via Composite Bayesian Optimizationcited by 6
- 2019Batched Stochastic Bayesian Optimization via Combinatorial Constraints Designcited by 4
Built on by
Citing papers per year: 2022: 9; 2023: 16; 2024: 35; 2025: 27; 2026: 12.
99 papers.
Sort:
By area:
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 2022Multi-Objective GFlowNetscited by 100
- 2024Fine-Tuned Language Models Generate Stable Inorganic Materials as Textcited by 92
- 2024Position: Bayesian Deep Learning is Needed in the Age of Large-Scale AIcited by 60
- 2023A Study of Bayesian Neural Network Surrogates for Bayesian Optimizationcited by 50
- 2024A Sober Look at LLMs for Material Discovery: Are They Actually Good for Bayesian Optimization Over Molecules?cited by 46
- 2022GAUCHE: A Library for Gaussian Processes in Chemistrycited by 40
- 2023AbDiffuser: Full-Atom Generation of in vitro Functioning Antibodiescited by 35
- 2024Full-Atom Peptide Design based on Multi-modal Flow Matchingcited by 34
- 2022Discovering Many Diverse Solutions with Bayesian Optimizationcited by 29
- 2022Bayesian Optimization with Conformal Prediction Setscited by 27
- 2024Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)cited by 26
38 more
- 2024A survey and benchmark of high-dimensional Bayesian optimization of discrete sequencescited by 22
- 2023Improving Protein Optimization with Smoothed Fitness Landscapescited by 20
- 2023The Behavior and Convergence of Local Bayesian Optimizationcited by 19
- 2025Latent Bayesian Optimization via Autoregressive Normalizing Flowscited by 19
- 2023Advancing Bayesian Optimization via Learning Correlated Latent Spacecited by 18
- 2023Multi-Fidelity Active Learning with GFlowNetscited by 15
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generationcited by 15
- 2024Concept Bottleneck Language Models For protein designcited by 15
- 2024Offline Multi-Objective Optimizationcited by 14
- 2025Offline Model-Based Optimization: Comprehensive Reviewcited by 14
- 2023Order-Preserving GFlowNetscited by 13
- 2024Robust Optimization in Protein Fitness Landscapes Using Reinforcement Learning in Latent Spacecited by 13
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Designcited by 13
- 2024Offline Model-Based Optimization by Learning to Rankcited by 13
- 2025Return of the Latent Space COWBOYS: Re-thinking the use of VAEs for Bayesian Optimisation of Structured Spacescited by 13
- 2024TAGMol: Target-Aware Gradient-guided Molecule Generationcited by 12
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimizationcited by 12
- 2022Generalizing Bayesian Optimization with Decision-theoretic Entropiescited by 11
- 2024Annealing Flow Generative Models Towards Sampling High-Dimensional and Multi-Modal Distributionscited by 11
- 2022Graph Neural Network Banditscited by 10
- 2024Bayesian Optimization of Antibodies Informed by a Generative Model of Evolving Sequencescited by 10
- 2025Large language models as uncertainty-calibrated optimizers for experimental discoverycited by 10
- 2023PG-LBO: Enhancing High-Dimensional Bayesian Optimization with Pseudo-Label and Gaussian Process Guidancecited by 9
- 2024Hotspot-Driven Peptide Design via Multi-Fragment Autoregressive Extensioncited by 9
- 2022PropertyDAG: Multi-objective Bayesian optimization of partially ordered, mixed-variable properties for biological sequence designcited by 8
- 2024Generative AI for Controllable Protein Sequence Design: A Surveycited by 8
- 2024Approximation-Aware Bayesian Optimizationcited by 8
- 2024Generalists vs. Specialists: Evaluating LLMs on Highly-Constrained Biophysical Sequence Optimization Taskscited by 8
- 2024Closed-Form Test Functions for Biophysical Sequence Optimization Algorithmscited by 7
- 2025Steering Generative Models with Experimental Data for Protein Fitness Optimizationcited by 7
- 2025Steering Protein Language Modelscited by 7
- 2023Biological Sequence Kernels with Guaranteed Flexibilitycited by 6
- 2024Variational Search Distributionscited by 6
- 2025Towards Universal Offline Black-Box Optimization via Learning Language Model Embeddingscited by 6
- 2023Joint Composite Latent Space Bayesian Optimizationcited by 5
- 2024Bayesian Optimization through Gaussian Cox Process Models for Spatio-temporal Datacited by 5
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomoleculescited by 5
- 2025AReUReDi: Annealed Rectified Updates for Refining Discrete Flows with Multi-Objective Guidancecited by 5
50 most cited of 99 All 99 (JSON)
- 2026pCoMole: Pareto-Constrained Molecule Editing with Discrete Flows1 Oct 2026
- 2026Co-PiLOT: Constrained Physics-Informed Latent Optimization for Target-Driven Inverse Design29 Sep 2026
- 2026Finding the Needle in a Haystack: Test-Time Analog Circuit Representation Adaptation for Bayesian Optimization13 Aug 2026
- 2026In-Context Learning for Latent Space Bayesian Optimization8 Jun 2026
- 2026AgentPLM: Agentic Protein Language Models with Reasoning-Augmented Decoding for Protein Sequence Design1 Jun 2026
- 2026Online Sharp-Calibrated Bayesian Optimization11 May 2026
- 2026BOAT: Navigating the Sea of In Silico Predictors for Antibody Design via Multi-Objective Bayesian Optimization15 Apr 2026
- 2026SciDesignBench: Benchmarking and Improving Language Models for Scientific Inverse Design13 Mar 2026
- 2026Few-Shot Design Optimization by Exploiting Auxiliary Information12 Feb 2026
- 2026Pareto-Conditioned Diffusion Models for Offline Multi-Objective Optimization31 Jan 2026
- 2026Regime-Adaptive Bayesian Optimization via Dirichlet Process Mixtures of Gaussian Processes27 Jan 2026
- 2026Sampling via Stochastic Interpolants by Langevin-based Velocity and Initialization Estimation in Flow ODEs13 Jan 2026
38 more
- 2025Trust Region-Based Bayesian Optimisation to Discover Diverse Solutions2 Nov 2025
- 2025Generative Bayesian Optimization: Generative Models as Acquisition Functions29 Oct 2025
- 2025Informed Initialization for Bayesian Optimization and Active Learning27 Oct 2025
- 2025Amortized Active Generation of Pareto Sets23 Oct 2025
- 2025RNAGenScape: Property-Guided, Optimized Generation of mRNA Sequences with Manifold Langevin Dynamics14 Oct 2025
- 2025AReUReDi: Annealed Rectified Updates for Refining Discrete Flows with Multi-Objective Guidance30 Sep 2025
- 2025Is Sequence Information All You Need for Bayesian Optimization of Antibodies?29 Sep 2025
- 2025Response to Promises and Pitfalls of Deep Kernel Learning25 Sep 2025
- 2025BioBO: Biology-informed Bayesian Optimization for Perturbation Design24 Sep 2025
- 2025Guided Sequence-Structure Generative Modeling for Iterative Antibody Optimization19 Sep 2025
- 2025Return of the Latent Space COWBOYS: Re-thinking the use of VAEs for Bayesian Optimisation of Structured Spaces5 Jul 2025
- 2025Guided Generation for Developable Antibodies3 Jul 2025
- 2025Steering Protein Language Models1 Jul 2025
- 2025DualEquiNet: A Dual-Space Hierarchical Equivariant Network for Large Biomolecules10 Jun 2025
- 2025Towards Universal Offline Black-Box Optimization via Learning Language Model Embeddings8 Jun 2025
- 2025Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization6 Jun 2025
- 2025ProSpero: Active Learning for Robust Protein Design Beyond Wild-Type Neighborhoods28 May 2025
- 2025Robust and Computation-Aware Gaussian Processes27 May 2025
- 2025Steering Generative Models with Experimental Data for Protein Fitness Optimization21 May 2025
- 2025Latent Bayesian Optimization via Autoregressive Normalizing Flows21 Apr 2025
- 2025Large language models as uncertainty-calibrated optimizers for experimental discovery8 Apr 2025
- 2025Global-Order GFlowNets3 Apr 2025
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimization21 Mar 2025
- 2025Offline Model-Based Optimization: Comprehensive Review21 Mar 2025
- 2025Large Scale Multi-Task Bayesian Optimization with Large Language Models11 Mar 2025
- 2025Covering Multiple Objectives with a Small Set of Solutions Using Bayesian Optimization31 Jan 2025
- 2025Bayesian Optimization with Preference Exploration using a Monotonic Neural Network Ensemble30 Jan 2025
- 2024Indirect Query Bayesian Optimization with Integrated Feedback18 Dec 2024
- 2024Bayesian Optimization of Antibodies Informed by a Generative Model of Evolving Sequences10 Dec 2024
- 2024Hotspot-Driven Peptide Design via Multi-Fragment Autoregressive Extension26 Nov 2024
- 2024GROOT: Effective Design of Biological Sequences with Limited Experimental Data18 Nov 2024
- 2024Concept Bottleneck Language Models For protein design9 Nov 2024
- 2024Respecting the limit:Bayesian optimization with a bound on the optimal value7 Nov 2024
- 2024MutaPLM: Protein Language Modeling for Mutation Explanation and Engineering30 Oct 2024
- 2024Generalists vs. Specialists: Evaluating LLMs on Highly-Constrained Biophysical Sequence Optimization Tasks29 Oct 2024
- 2024Offline Model-Based Optimization by Learning to Rank15 Oct 2024
- 2024Batched Energy-Entropy acquisition for Bayesian Optimization11 Oct 2024
- 2024Improved Off-policy Reinforcement Learning in Biological Sequence Design6 Oct 2024
50 newest of 99 All 99 (JSON)
- 2023Protein Design with Guided Discrete Diffusioncited by 154
- 2023AbDiffuser: Full-Atom Generation of in vitro Functioning Antibodiescited by 35
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generationcited by 15
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimizationcited by 12
- 2025Steering Generative Models with Experimental Data for Protein Fitness Optimizationcited by 7
- 2026Pareto-Conditioned Diffusion Models for Offline Multi-Objective Optimizationcited by 2
- 2022Conditional Diffusion with Less Explicit Guidance via Model Predictive Controlcited by 1
- 2025Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimizationcited by 1
- 2026Co-PiLOT: Constrained Physics-Informed Latent Optimization for Target-Driven Inverse Designcited by 0
- 2026Co-PiLOT: Constrained Physics-Informed Latent Optimization for Target-Driven Inverse Design29 Sep 2026
- 2026Pareto-Conditioned Diffusion Models for Offline Multi-Objective Optimization31 Jan 2026
- 2025Learning Design-Score Manifold to Guide Diffusion Models for Offline Optimization6 Jun 2025
- 2025Steering Generative Models with Experimental Data for Protein Fitness Optimization21 May 2025
- 2025Preference-Guided Diffusion for Multi-Objective Offline Optimization21 Mar 2025
- 2024PROUD: PaRetO-gUided Diffusion Model for Multi-objective Generation5 Jul 2024
- 2023AbDiffuser: Full-Atom Generation of in vitro Functioning Antibodies28 Jul 2023
- 2023Protein Design with Guided Discrete Diffusion31 May 2023
- 2022Conditional Diffusion with Less Explicit Guidance via Model Predictive Control21 Oct 2022
- 2022Multi-Objective GFlowNetscited by 100
- 2024Fine-Tuned Language Models Generate Stable Inorganic Materials as Textcited by 92
- 2024Position: Bayesian Deep Learning is Needed in the Age of Large-Scale AIcited by 60
- 2023A Study of Bayesian Neural Network Surrogates for Bayesian Optimizationcited by 50
- 2024A Sober Look at LLMs for Material Discovery: Are They Actually Good for Bayesian Optimization Over Molecules?cited by 46
- 2022GAUCHE: A Library for Gaussian Processes in Chemistrycited by 40
- 2024Full-Atom Peptide Design based on Multi-modal Flow Matchingcited by 34
- 2022Discovering Many Diverse Solutions with Bayesian Optimizationcited by 29
- 2022Bayesian Optimization with Conformal Prediction Setscited by 27
- 2024Conformal Validity Guarantees Exist for Any Data Distribution (and How to Find Them)cited by 26
- 2024A survey and benchmark of high-dimensional Bayesian optimization of discrete sequencescited by 22
- 2023Improving Protein Optimization with Smoothed Fitness Landscapescited by 20
12 most cited of 89 All 89 (JSON)
- 2026pCoMole: Pareto-Constrained Molecule Editing with Discrete Flows1 Oct 2026
- 2026Finding the Needle in a Haystack: Test-Time Analog Circuit Representation Adaptation for Bayesian Optimization13 Aug 2026
- 2026In-Context Learning for Latent Space Bayesian Optimization8 Jun 2026
- 2026Online Sharp-Calibrated Bayesian Optimization11 May 2026
- 2026BOAT: Navigating the Sea of In Silico Predictors for Antibody Design via Multi-Objective Bayesian Optimization15 Apr 2026
- 2026SciDesignBench: Benchmarking and Improving Language Models for Scientific Inverse Design13 Mar 2026
- 2026Few-Shot Design Optimization by Exploiting Auxiliary Information12 Feb 2026
- 2026Regime-Adaptive Bayesian Optimization via Dirichlet Process Mixtures of Gaussian Processes27 Jan 2026
- 2026Sampling via Stochastic Interpolants by Langevin-based Velocity and Initialization Estimation in Flow ODEs13 Jan 2026
- 2025Trust Region-Based Bayesian Optimisation to Discover Diverse Solutions2 Nov 2025
- 2025Generative Bayesian Optimization: Generative Models as Acquisition Functions29 Oct 2025
- 2025Informed Initialization for Bayesian Optimization and Active Learning27 Oct 2025
12 newest of 89 All 89 (JSON)