Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distance
Not on the map. Why?
Abstract (arXiv)
Designing experiments that systematically gather data from complex physical systems is central to accelerating scientific discovery. While Bayesian experimental design (BED) provides a principled, information-based framework that integrates experimental planning with probabilistic inference, the selection of utility functions in BED is a long-standing and active topic, where different criteria emphasize different notions of information. Although Kullback--Leibler (KL) divergence has been one of the most common choices, recent studies have proposed Wasserstein distance as an alternative. In this work, we first employ a toy example to illustrate an issue of Wasserstein distance - the value of Wasserstein distance of a fixed-shape posterior depends on the relative position of its main mass within the support and can exhibit false rewards unrelated to information gain, especially with a non-informative prior (e.g., uniform distribution). We then further provide a systematic comparison between these two criteria through a classical source inversion problem in the BED literature, revealing that the KL divergence tends to lead to faster convergence in the absence of model discrepancy, while Wasserstein metrics provide more robust sequential BED results if model discrepancy is non-negligible. These findings clarify the trade-offs between KL divergence and Wasserstein metrics for the utility function and provide guidelines for selecting suitable criteria in practical BED applications.
Key facts. No citing papers in the map yet (data through 13 Oct 2026). Builds on 57 of its 154 references that are in the map. First posted on arXiv 23 Jan 2026.
Builds on
57 of 154 references.
References · 57
- 2013Auto-Encoding Variational Bayescited by 4,024
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,857
- 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,989
- 2016Variational Inference: A Review for Statisticianscited by 1,469
- 2017Deep Bayesian Active Learning with Image Datacited by 816
- 2018Computational Optimal Transportcited by 772
- 2019On Variational Bounds of Mutual Informationcited by 551
- 2021Machine learning accelerated computational fluid dynamicscited by 465
- 2019Machine Learning for Fluid Mechanicscited by 381
- 2017Wasserstein Auto-Encoderscited by 372
- 2017Certifying Some Distributional Robustness with Principled Adversarial Trainingcited by 336
- 2013Black Box Variational Inferencecited by 319
38 more
- 2018Statistical Aspects of Wasserstein Distancescited by 202
- 2016Structured Inference Networks for Nonlinear State Space Modelscited by 182
- 2023Modern Bayesian Experimental Designcited by 123
- 2019Variational Bayesian Optimal Experimental Designcited by 116
- 2014Inconsistency of Bayesian Inference for Misspecified Linear Models, and a Proposal for Repairing Itcited by 93
- 2017On Nesting Monte Carlo Estimatorscited by 82
- 2017REBAR: Low-variance, unbiased gradient estimates for discrete latent variable modelscited by 79
- 2019A Unified Stochastic Gradient Approach to Designing Bayesian-Optimal Experimentscited by 58
- 2017Assigning a value to a power likelihood in a general Bayesian modelcited by 46
- 2024Optimal experimental design: Formulations and computationscited by 45
- 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
- 2021A Framework for Machine Learning of Model Error in Dynamical Systemscited by 38
- 2018Ensemble Kalman Inversion: A Derivative-Free Technique For Machine Learning Taskscited by 37
- 2018Efficient Bayesian Experimental Design for Implicit Modelscited by 34
- 2021Auto-differentiable Ensemble Kalman Filterscited by 28
- 2020A comparison of learning rate selection methods in generalized Bayesian inferencecited by 23
- 2021Bayesian Sequential Optimal Experimental Design for Nonlinear Models Using Policy Gradient Reinforcement Learningcited by 23
- 2022Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfecited by 22
- 2020Automatic structured variational inferencecited by 21
- 2019Importance Weighted Hierarchical Variational Inferencecited by 20
- 2022Ensemble Kalman Methods: A Mean Field Perspectivecited by 20
- 2020Unbiased MLMC stochastic gradient-based optimization of Bayesian experimental designscited by 19
- 2017The Ensemble Kalman Filter: A Signal Processing Perspectivecited by 17
- 2018Wasserstein Variational Inferencecited by 17
- 2022Large-scale Bayesian optimal experimental design with derivative-informed projected neural networkcited by 17
- 2022Sliced Wasserstein Variational Inferencecited by 17
- 2021Learning High-Dimensional Parametric Maps via Reduced Basis Adaptive Residual Networkscited by 16
- 2022Discrepancy Modeling Framework: Learning missing physics, modeling systematic residuals, and disambiguating between deterministic and random effectscited by 15
- 2023Neural Dynamical Operator: Continuous Spatial-Temporal Model with Gradient-Based and Derivative-Free Optimization Methodscited by 14
- 2020Iterative Ensemble Kalman Methods: A Unified Perspective with Some New Variantscited by 13
- 2023Learning About Structural Errors in Models of Complex Dynamical Systemscited by 13
- 2024Variational Bayesian Optimal Experimental Design with Normalizing Flowscited by 13
- 2022Robust Expected Information Gain for Optimal Bayesian Experimental Design Using Ambiguity Setscited by 12
- 2024Data-Driven Stochastic Closure Modeling via Conditional Diffusion Model and Neural Operatorcited by 12
- 2023Bayesian inference for misspecified generative modelscited by 11
- 2023Accurate, scalable, and efficient Bayesian optimal experimental design with derivative-informed neural operatorscited by 10
- 2020A fast and scalable computational framework for large-scale and high-dimensional Bayesian optimal experimental designcited by 9
- 2024Sequential infinite-dimensional Bayesian optimal experimental design with derivative-informed latent attention neural operatorcited by 9
50 most cited of 57 All 57 (JSON)
Built on by
Not cited yet.