Tractable Expected Information Gains for Exponential Family Posteriors
Rik Knowles, Tom Rainforth
Abstract
We investigate which models admit a collapse of the expected information gain (EIG) and its derivative from a doubly intractable to a singly intractable expression. We prove that a sufficient condition is that the posterior distribution belongs to an exponential family (EF) and depends on the experimental design and data only through its natural parameters, and derive corresponding singly intractable and unbiased estimators for the and its (reparameterized) gradient. We further show that this is achieved when using a likelihood of an analogous form and any arbitrary prior. This is complemented by a theoretical analysis of certain degenerate behaviors that may arise when optimizing the for EF-modeled experiments. Finally, we empirically demonstrate the benefits of using our singly intractable estimators, showing superior convergence rates, and substantial performance gains for sequential design problems compared to using standard nested estimators.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on7
- Deep Adaptive Design: Amortizing Sequential Bayesian Experimental DesignAdam Foster, Desi R. Ivanova, Ilyas Malik, Tom RainforthICML 2021 · 119 citations
- Fast and Flexible Temporal Point Processes with Triangular MapsOleksandr Shchur, Nicholas Gao, Marin Bilos, Stephan GünnemannNeurIPS 2020 · 43 citations
- Bayesian Online Natural Gradient (BONG)Matt Jones, Peter G. Chang, Kevin P. MurphyNeurIPS 2024 · 20 citations
- On Estimating the Gradient of the Expected Information Gain in Bayesian Experimental DesignZiqiao Ao, Jinglai LiAAAI 2024 · 4 citations
- Bayesian Experimental Design Via Contrastive DiffusionsJacopo Iollo, Christophe Heinkelé, Pierre Alliez, Florence ForbesICLR 2025
Related papers
- Reverse-Annealed Sequential Monte Carlo for Efficient Bayesian Optimal Experiment DesignJake Callahan, Andrew Chin, Jason Pacheco, Tommie A. CatanachNeurIPS 2025 · 1 citation
- Online Bayesian Experimental Design for Partially Observed Dynamical SystemsSara Pérez-Vieites, Sahel Iqbal, Simo Särkkä, Dominik BaumannICML 2026
- Design Amortization for Bayesian Optimal Experimental DesignNoble Kennamer, Steven Walton, Alexander IhlerAAAI 2023 · 7 citations
- PASOA- PArticle baSed Bayesian Optimal Adaptive designJacopo Iollo, Christophe Heinkelé, Pierre Alliez, Florence ForbesICML 2024 · 4 citations
- Improving Unsupervised Hierarchical Representation With Reinforcement LearningRuyi An, Yewen Li, Xu He, Pengjie Gu et al.CVPR 2024
