Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Design
Not on the map. Why?
Abstract (arXiv)
We introduce Deep Adaptive Design (DAD), a method for amortizing the cost of adaptive Bayesian experimental design that allows experiments to be run in real-time. Traditional sequential Bayesian optimal experimental design approaches require substantial computation at each stage of the experiment. This makes them unsuitable for most real-world applications, where decisions must typically be made quickly. DAD addresses this restriction by learning an amortized design network upfront and then using this to rapidly run (multiple) adaptive experiments at deployment time. This network represents a design policy which takes as input the data from previous steps, and outputs the next design using a single forward pass; these design decisions can be made in milliseconds during the live experiment. To train the network, we introduce contrastive information bounds that are suitable objectives for the sequential setting, and propose a customized network architecture that exploits key symmetries. We demonstrate that DAD successfully amortizes the process of experimental design, outperforming alternative strategies on a number of problems.
Key facts. Cited by 103 papers in the map (data through 7 Oct 2026). Builds on 25 of its 66 references that are in the map. First posted on arXiv 3 Mar 2021.
Builds on
25 of 66 references.
References · 25
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2013Auto-Encoding Variational Bayescited by 3,980
- 2017Deep Setscited by 1,394
- 2015Automatic differentiation in machine learning: a surveycited by 846
- 2019On Variational Bounds of Mutual Informationcited by 550
- 2018Conditional Neural Processescited by 421
- 2018Pyro: Deep Universal Probabilistic Programmingcited by 379
- 2020Planning to Explore via Self-Supervised World Modelscited by 365
- 2018Neural Processescited by 307
- 2019Monte Carlo Gradient Estimation in Machine Learningcited by 281
13 more
- 2018Inference Suboptimality in Variational Autoencoderscited by 131
- 2015Gradient Estimation Using Stochastic Computation Graphscited by 120
- 2016Towards a Neural Statisticiancited by 119
- 2019Variational Bayesian Optimal Experimental Designcited by 116
- 2017On Nesting Monte Carlo Estimatorscited by 82
- 2017REBAR: Low-variance, unbiased gradient estimates for discrete latent variable modelscited by 78
- 2019A Unified Stochastic Gradient Approach to Designing Bayesian-Optimal Experimentscited by 58
- 2020Bayesian Experimental Design for Implicit Models by Mutual Information Neural Estimationcited by 56
- 2015GLASSES: Relieving The Myopia Of Bayesian Optimisationcited by 38
- 2019BINOCULARS for Efficient, Nonmyopic Sequential Experimental Designcited by 35
- 2016Sequential Bayesian optimal experimental design via approximate dynamic programmingcited by 27
- 2020Sequential Bayesian Experimental Design for Implicit Models via Mutual Informationcited by 23
- 2020Unbiased MLMC stochastic gradient-based optimization of Bayesian experimental designscited by 19
Built on by
Citing papers per year: 2021: 6; 2022: 11; 2023: 13; 2024: 14; 2025: 25; 2026: 34.
103 papers.
Sort:
By area:
- 2023Modern Bayesian Experimental Designcited by 121
- 2021Active Testing: Sample-Efficient Model Evaluationcited by 82
- 2023GFlowNets for AI-Driven Scientific Discoverycited by 63
- 2021Implicit Deep Adaptive Design: Policy-Based Experimental Design without Likelihoodscited by 61
- 2022Interventions, Where and How? Experimental Design for Causal Models at Scalecited by 49
- 2022Optimizing Sequential Experimental Design with Deep Reinforcement Learningcited by 45
- 2024The Essential Role of Causality in Foundation World Models for Embodied AIcited by 32
- 2021Gradient-based Bayesian Experimental Design for Implicit Models using Mutual Information Lower Boundscited by 26
- 2022Active Surrogate Estimators: An Active Learning Approach to Label-Efficient Model Evaluationcited by 26
- 2025BED-LLM: Intelligent Information Gathering with LLMs and Bayesian Experimental Designcited by 21
- 2024Amortized Bayesian Experimental Design for Decision-Makingcited by 20
- 2022Active Exploration via Experiment Design in Markov Chainscited by 19
38 more
- 2021Tight Mutual Information Estimation With Contrastive Fenchel-Legendre Optimizationcited by 18
- 2022Combining Multi-Fidelity Modelling and Asynchronous Batch Bayesian Optimizationcited by 16
- 2022Policy-Based Bayesian Experimental Design for Non-Differentiable Implicit Modelscited by 15
- 2023Differentiable Multi-Target Causal Bayesian Experimental Designcited by 15
- 2025BoxingGym: Benchmarking Progress in Automated Experimental Design and Model Discoverycited by 15
- 2021Active Learning in Bayesian Neural Networks with Balanced Entropy Learning Principlecited by 14
- 2023Leveraging Self-Consistency for Data-Efficient Amortized Bayesian Inferencecited by 14
- 2023Black-Box Batch Active Learning for Regressioncited by 12
- 2024Nesting Particle Filters for Experimental Design in Dynamical Systemscited by 12
- 2025Step-DAD: Semi-Amortized Policy-Based Bayesian Experimental Designcited by 12
- 2024Amortized Active Causal Induction with Deep Reinforcement Learningcited by 10
- 2024Bayesian Experimental Design via Contrastive Diffusionscited by 9
- 2023Variational Sequential Optimal Experimental Design using Reinforcement Learningcited by 8
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisitioncited by 8
- 2021Deep Bayesian Active Learning for Accelerating Stochastic Simulationcited by 7
- 2022Design Amortization for Bayesian Optimal Experimental Designcited by 7
- 2024Advancing Deep Active Learning & Data Subset Selection: Unifying Principles with Information-Theory Intuitionscited by 7
- 2025PABBO: Preferential Amortized Black-Box Optimizationcited by 7
- 2024Unraveling the Interplay between Carryover Effects and Reward Autocorrelations in Switchback Experimentscited by 6
- 2026Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world modelscited by 6
- 2023Statistically Efficient Bayesian Sequential Experiment Design via Reinforcement Learning with Cross-Entropy Estimatorscited by 5
- 2024PASOA- PArticle baSed Bayesian Optimal Adaptive designcited by 5
- 2025Active Learning of Model Discrepancy with Bayesian Experimental Designcited by 5
- 2025Balancing Interference and Correlation in Spatial Experimental Designs: A Causal Graph Cut Approachcited by 5
- 2025In-Context Multi-Objective Optimizationcited by 5
- 2022Bayesian Active Learning for Discrete Latent Variable Modelscited by 4
- 2023CO-BED: Information-Theoretic Contextual Optimization via Bayesian Experimental Designcited by 4
- 2023Online simulator-based experimental design for cognitive model selectioncited by 3
- 2023Experimentation Platforms Meet Reinforcement Learning: Bayesian Sequential Decision-Making for Continuous Monitoringcited by 3
- 2024Recursive Nested Filtering for Efficient Amortized Bayesian Experimental Designcited by 3
- 2025Amortized Safe Active Learning for Real-Time Data Acquisition: Pretrained Neural Policies From Simulated Nonparametric Functionscited by 3
- 2025Performance Comparisons of Reinforcement Learning Algorithms for Sequential Experimental Designcited by 3
- 2026ATLAS: Active Theory Learning for Automated Sciencecited by 3
- 2023Adaptive Experimentation at Scale: A Computational Framework for Flexible Batchescited by 2
- 2024Amortized nonmyopic active search via deep imitation learningcited by 2
- 2025PIED: Physics-Informed Experimental Design for Inverse Problemscited by 2
- 2025On Efficient Bayesian Exploration in Model-Based Reinforcement Learningcited by 2
- 2025FraPPE: Fast and Efficient Preference-based Pure Explorationcited by 2
50 most cited of 103 All 103 (JSON)
- 2026MECHVAR: Variance-Guided Mechanism Discrimination for Autonomous Machine Learning Experiment Selection1 Oct 2026
- 2026Information-theoretic receding-horizon active learning of nonlinear dynamical systems29 Sep 2026
- 2026NeuronSifter: Intervention Planning in CNS Microenvironments28 Sep 2026
- 2026GPARA: Graph-Posterior-Aligned Refinement and Active Acquisition for Grounding Diffusion Priors28 Sep 2026
- 2026Active Data Acquisition with Side Information via Discrete Diffusion Priors26 Sep 2026
- 2026LAVOIR: Teaching a Single-Pass Decision Encoder When and What to Ask with Amortized Value of Information25 Sep 2026
- 2026WhatWorkedBench: Benchmarking Experimental Understanding in AI Agents23 Sep 2026
- 2026New Evidence, Same Choice: Testing Physical Experiment Selection in Vision Language Models10 Sep 2026
- 2026Wide Learning: Learning to Reach Evidence30 Aug 2026
- 2026Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models10 Aug 2026
- 2026Capability-Gated Planning: Cost-to-Goal Discovery and the Limits of Myopic Experiment Selection5 Aug 2026
- 2026Identifying Informative Environments for Cognition Parameter Inference via Bayesian Experimental Design30 Jul 2026
38 more
- 2026Deep Adaptive Bayesian Screening18 Jul 2026
- 2026Robust Bayesian Decision Making under Adversarial Uncertainty9 Jul 2026
- 2026Bayesian Experimental Design via Score Matching9 Jul 2026
- 2026Efficient Adaptive Data Acquisition via Pretrained Belief Representations23 Jun 2026
- 2026FairBED: A Bayesian Experimental Design Approach to Gathering Fairer Data22 Jun 2026
- 2026SymQNet: Amortized Acquisition for Low-Latency Adaptive Hamiltonian Learning11 Jun 2026
- 2026ATLAS: Active Theory Learning for Automated Science10 Jun 2026
- 2026When Should an AI Scientist Stop? Verifiable Experiment Steering and Refusal for Autonomous Discovery26 May 2026
- 2026Constrained Bayesian Experimental Design via Online Planning26 May 2026
- 2026Goal-driven Bayesian Optimal Experimental Design for Robust Decision-Making Under Model Uncertainty25 May 2026
- 2026In-Context Black-Box Optimization with Unreliable Feedback7 May 2026
- 2026Adaptive Querying with AI Persona Priors1 May 2026
- 2026Beyond Expected Information Gain: Stable Bayesian Optimal Experimental Design with Integral Probability Metrics and Plug-and-Play Extensions23 Apr 2026
- 2026Loss-Driven Bayesian Active Learning13 Apr 2026
- 2026Deep Adaptive Model-Based Design of Experiments17 Mar 2026
- 2026Maximin Robust Bayesian Experimental Design14 Mar 2026
- 2026Cold-Start Personalization via Training-Free Priors from Structured World Models16 Feb 2026
- 2026Whom to Query for What: Adaptive Group Elicitation via Multi-Turn LLM Interactions15 Feb 2026
- 2026Supercharging Simulation-Based Inference for Bayesian Optimal Experimental Design6 Feb 2026
- 2026Designing Time Series Experiments in A/B Testing with Transformer Reinforcement Learning2 Feb 2026
- 2026Neural Nonmyopic Bayesian Optimization in Dynamic Cost Settings10 Jan 2026
- 2026Sequential Bayesian Optimal Experimental Design in Infinite Dimensions via Policy Gradient Reinforcement Learning9 Jan 2026
- 2025JADAI: Jointly Amortizing Adaptive Design and Bayesian Inference28 Dec 2025
- 2025In-Context Multi-Objective Optimization11 Dec 2025
- 2025Online Bayesian Experimental Design for Partially Observed Dynamical Systems6 Nov 2025
- 2025A Geometric Approach to Optimal Experimental Design16 Oct 2025
- 2025BALLAST: Bayesian Active Learning with Look-ahead Amendment for Sea-drifter Trajectories under Spatio-Temporal Vector Fields30 Sep 2025
- 2025BED-LLM: Intelligent Information Gathering with LLMs and Bayesian Experimental Design28 Aug 2025
- 2025FraPPE: Fast and Efficient Preference-based Pure Exploration22 Aug 2025
- 2025Step-DAD: Semi-Amortized Policy-Based Bayesian Experimental Design18 Jul 2025
- 2025Goal-Oriented Sequential Bayesian Experimental Design for Causal Learning10 Jul 2025
- 2025Direct Regret Optimization in Bayesian Optimization9 Jul 2025
- 2025On Efficient Bayesian Exploration in Model-Based Reinforcement Learning3 Jul 2025
- 2025Active MRI Acquisition with Diffusion Guided Bayesian Experimental Design19 Jun 2025
- 2025Representative, Informative, and De-Amplifying: Requirements for Robust Bayesian Active Learning under Model Misspecification9 Jun 2025
- 2025ALINE: Joint Amortization for Bayesian Inference and Active Data Acquisition8 Jun 2025
- 2025Optimization of Epsilon-Greedy Exploration3 Jun 2025
- 2025Balancing Interference and Correlation in Spatial Experimental Designs: A Causal Graph Cut Approach26 May 2025
50 newest of 103 All 103 (JSON)
- 2024The Essential Role of Causality in Foundation World Models for Embodied AIcited by 32
- 2026Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world modelscited by 6
- 2026Cold-Start Personalization via Training-Free Priors from Structured World Modelscited by 2
- 2026Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models10 Aug 2026
- 2026Cold-Start Personalization via Training-Free Priors from Structured World Models16 Feb 2026
- 2024The Essential Role of Causality in Foundation World Models for Embodied AI6 Feb 2024
- 2024Amortized nonmyopic active search via deep imitation learningcited by 2
- 2025Direct Regret Optimization in Bayesian Optimizationcited by 0
- 2025Direct Regret Optimization in Bayesian Optimization9 Jul 2025
- 2024Amortized nonmyopic active search via deep imitation learning23 May 2024
- 2023Modern Bayesian Experimental Designcited by 121
- 2021Active Testing: Sample-Efficient Model Evaluationcited by 82
- 2023GFlowNets for AI-Driven Scientific Discoverycited by 63
- 2021Implicit Deep Adaptive Design: Policy-Based Experimental Design without Likelihoodscited by 61
- 2022Interventions, Where and How? Experimental Design for Causal Models at Scalecited by 49
- 2022Optimizing Sequential Experimental Design with Deep Reinforcement Learningcited by 45
- 2021Gradient-based Bayesian Experimental Design for Implicit Models using Mutual Information Lower Boundscited by 26
- 2022Active Surrogate Estimators: An Active Learning Approach to Label-Efficient Model Evaluationcited by 26
- 2025BED-LLM: Intelligent Information Gathering with LLMs and Bayesian Experimental Designcited by 21
- 2024Amortized Bayesian Experimental Design for Decision-Makingcited by 20
- 2022Active Exploration via Experiment Design in Markov Chainscited by 19
- 2021Tight Mutual Information Estimation With Contrastive Fenchel-Legendre Optimizationcited by 18
12 most cited of 93 All 93 (JSON)
- 2026MECHVAR: Variance-Guided Mechanism Discrimination for Autonomous Machine Learning Experiment Selection1 Oct 2026
- 2026Information-theoretic receding-horizon active learning of nonlinear dynamical systems29 Sep 2026
- 2026NeuronSifter: Intervention Planning in CNS Microenvironments28 Sep 2026
- 2026LAVOIR: Teaching a Single-Pass Decision Encoder When and What to Ask with Amortized Value of Information25 Sep 2026
- 2026Wide Learning: Learning to Reach Evidence30 Aug 2026
- 2026Capability-Gated Planning: Cost-to-Goal Discovery and the Limits of Myopic Experiment Selection5 Aug 2026
- 2026Identifying Informative Environments for Cognition Parameter Inference via Bayesian Experimental Design30 Jul 2026
- 2026Deep Adaptive Bayesian Screening18 Jul 2026
- 2026Robust Bayesian Decision Making under Adversarial Uncertainty9 Jul 2026
- 2026Bayesian Experimental Design via Score Matching9 Jul 2026
- 2026Efficient Adaptive Data Acquisition via Pretrained Belief Representations23 Jun 2026
- 2026FairBED: A Bayesian Experimental Design Approach to Gathering Fairer Data22 Jun 2026
12 newest of 93 All 93 (JSON)