mlsanityData to 14 Oct 2026

Theoretical guidelines for annealed Langevin dynamics in compositional simulation-based inference

arXiv 2605.21253 ↗First posted 20 May 2026stat.ML cs.LGcited by 1

Not on the map. Why?

Abstract (arXiv)

Compositional score-based approaches to simulation-based inference (SBI) approximate the posterior over a shared parameter given n independent observations by aggregating individually learned posterior scores: currently, there are two main propositions of such methods (Geffner et al. (2023), Linhart et al. (2026)). As the resulting composite score does not correspond to the score of any distribution along the forward diffusion path of the true multi-observation posterior, sampling from it via a reverse SDE leads to an irreducible bias. Annealed Langevin dynamics provides a principled alternative: it treats the composite score as the genuine score of a sequence of tractable bridging densities and samples from them in succession. When properly tuned, it could lead to a controllable bias. However, its hyperparameters, namely step sizes, the number of steps per level, and the number of annealing levels, have so far been chosen empirically. We derive Wasserstein bounds for annealed Langevin with approximate scores and translate them into explicit decision rules for these hyperparameters that guarantee a prescribed sampling accuracy, while highlighting different theoretical aspects of each composite score formulation. In the Gaussian setting, we obtain closed-form expressions for all relevant quantities and prove that the bridging densities of Linhart et al. (2026) consistently admit larger step sizes and require fewer total Langevin steps than those of Geffner et al. (2023). Furthermore, we show empirically that the tuning obtained in the Gaussian setting generalizes to more complex problems, thus providing a well-understood and theoretically grounded starting point for practitioners using compositional score-based approaches.

Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 22 of its 29 references that are in the map. First posted on arXiv 20 May 2026.

Builds on

22 of 29 references.

References · 22

  1. 2020Denoising Diffusion Probabilistic Modelscited by 21,076
  2. 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
  3. 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
  4. 2020Improved Techniques for Training Score-Based Generative Modelscited by 1,043
  5. 2019The frontier of simulation-based inferencecited by 502
  6. 2019Rapid Convergence of the Unadjusted Langevin Algorithm: Isoperimetry Sufficescited by 227
  7. 2023Reduce, Reuse, Recycle: Compositional Generation with Energy-Based Diffusion Models and MCMCcited by 220
  8. 2015Non-asymptotic convergence analysis for the Unadjusted Langevin Algorithmcited by 197
  9. 2019Composable Effects for Flexible and Accelerated Probabilistic Programming in NumPyrocited by 191
  10. 2021Benchmarking Simulation-Based Inferencecited by 162
  11. 2020BayesFlow: Learning complex stochastic models with invertible neural networkscited by 144
  12. 2017User-friendly guarantees for the Langevin Monte Carlo with inaccurate gradientcited by 116
10 more
  1. 2022Sequential Neural Score Estimation: Likelihood-Free Inference with Conditional Score Based Diffusion Modelscited by 50
  2. 2022Compositional Score Modeling for Simulation-based Inferencecited by 48
  3. 2025Simulation-Based Inference: A Practical Guidecited by 32
  4. 2017The Tamed Unadjusted Langevin Algorithmcited by 31
  5. 2025Beyond Log-Concavity and Score Regularity: Improved Convergence Bounds for Score-Based Generative Models in W2-distancecited by 29
  6. 2021HNPE: Leveraging Global Parameters for Neural Posterior Estimationcited by 19
  7. 2024Diffusion posterior sampling for simulation-based inference in tall data settingscited by 19
  8. 2023Gaussian Interpolation Flowscited by 17
  9. 2024Compositional simulation-based inference for time seriescited by 9
  10. 2025Error analysis of a compositional score-based algorithm for simulation-based inferencecited by 1

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 1.

1 paper.