mlsanityData to 13 Oct 2026

Bayesian Experimental Design for Model Discrepancy Calibration: A Rivalry between Kullback--Leibler Divergence and Wasserstein Distance

arXiv 2601.16425 ↗First posted 23 Jan 2026cs.LGcited by 0

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

  1. 2013Auto-Encoding Variational Bayescited by 4,024
  2. 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,857
  3. 2019DeepONet: Learning nonlinear operators for identifying differential equations based on the universal approximation theorem of operatorscited by 1,989
  4. 2016Variational Inference: A Review for Statisticianscited by 1,469
  5. 2017Deep Bayesian Active Learning with Image Datacited by 816
  6. 2018Computational Optimal Transportcited by 772
  7. 2019On Variational Bounds of Mutual Informationcited by 551
  8. 2021Machine learning accelerated computational fluid dynamicscited by 465
  9. 2019Machine Learning for Fluid Mechanicscited by 381
  10. 2017Wasserstein Auto-Encoderscited by 372
  11. 2017Certifying Some Distributional Robustness with Principled Adversarial Trainingcited by 336
  12. 2013Black Box Variational Inferencecited by 319
38 more
  1. 2018Statistical Aspects of Wasserstein Distancescited by 202
  2. 2016Structured Inference Networks for Nonlinear State Space Modelscited by 182
  3. 2023Modern Bayesian Experimental Designcited by 123
  4. 2019Variational Bayesian Optimal Experimental Designcited by 116
  5. 2014Inconsistency of Bayesian Inference for Misspecified Linear Models, and a Proposal for Repairing Itcited by 93
  6. 2017On Nesting Monte Carlo Estimatorscited by 82
  7. 2017REBAR: Low-variance, unbiased gradient estimates for discrete latent variable modelscited by 79
  8. 2019A Unified Stochastic Gradient Approach to Designing Bayesian-Optimal Experimentscited by 58
  9. 2017Assigning a value to a power likelihood in a general Bayesian modelcited by 46
  10. 2024Optimal experimental design: Formulations and computationscited by 45
  11. 2021Tighter expected generalization error bounds via Wasserstein distancecited by 43
  12. 2021A Framework for Machine Learning of Model Error in Dynamical Systemscited by 38
  13. 2018Ensemble Kalman Inversion: A Derivative-Free Technique For Machine Learning Taskscited by 37
  14. 2018Efficient Bayesian Experimental Design for Implicit Modelscited by 34
  15. 2021Auto-differentiable Ensemble Kalman Filterscited by 28
  16. 2020A comparison of learning rate selection methods in generalized Bayesian inferencecited by 23
  17. 2021Bayesian Sequential Optimal Experimental Design for Nonlinear Models Using Policy Gradient Reinforcement Learningcited by 23
  18. 2022Faster Unbalanced Optimal Transport: Translation invariant Sinkhorn and 1-D Frank-Wolfecited by 22
  19. 2020Automatic structured variational inferencecited by 21
  20. 2019Importance Weighted Hierarchical Variational Inferencecited by 20
  21. 2022Ensemble Kalman Methods: A Mean Field Perspectivecited by 20
  22. 2020Unbiased MLMC stochastic gradient-based optimization of Bayesian experimental designscited by 19
  23. 2017The Ensemble Kalman Filter: A Signal Processing Perspectivecited by 17
  24. 2018Wasserstein Variational Inferencecited by 17
  25. 2022Large-scale Bayesian optimal experimental design with derivative-informed projected neural networkcited by 17
  26. 2022Sliced Wasserstein Variational Inferencecited by 17
  27. 2021Learning High-Dimensional Parametric Maps via Reduced Basis Adaptive Residual Networkscited by 16
  28. 2022Discrepancy Modeling Framework: Learning missing physics, modeling systematic residuals, and disambiguating between deterministic and random effectscited by 15
  29. 2023Neural Dynamical Operator: Continuous Spatial-Temporal Model with Gradient-Based and Derivative-Free Optimization Methodscited by 14
  30. 2020Iterative Ensemble Kalman Methods: A Unified Perspective with Some New Variantscited by 13
  31. 2023Learning About Structural Errors in Models of Complex Dynamical Systemscited by 13
  32. 2024Variational Bayesian Optimal Experimental Design with Normalizing Flowscited by 13
  33. 2022Robust Expected Information Gain for Optimal Bayesian Experimental Design Using Ambiguity Setscited by 12
  34. 2024Data-Driven Stochastic Closure Modeling via Conditional Diffusion Model and Neural Operatorcited by 12
  35. 2023Bayesian inference for misspecified generative modelscited by 11
  36. 2023Accurate, scalable, and efficient Bayesian optimal experimental design with derivative-informed neural operatorscited by 10
  37. 2020A fast and scalable computational framework for large-scale and high-dimensional Bayesian optimal experimental designcited by 9
  38. 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.