Differentiating Metropolis-Hastings to Optimize Intractable Densities
Not on the map. Why?
Abstract (arXiv)
We develop an algorithm for automatic differentiation of Metropolis-Hastings samplers, allowing us to differentiate through probabilistic inference, even if the model has discrete components within it. Our approach fuses recent advances in stochastic automatic differentiation with traditional Markov chain coupling schemes, providing an unbiased and low-variance gradient estimator. This allows us to apply gradient-based optimization to objectives expressed as expectations over intractable target densities. We demonstrate our approach by finding an ambiguous observation in a Gaussian mixture model and by maximizing the specific heat in an Ising model.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 10 of its 24 references that are in the map. First posted on arXiv 13 Jun 2023.
Builds on
10 of 24 references.
References · 10
- 2020Improved Contrastive Divergence Training of Energy Based Modelscited by 130
- 2022Score-Based Diffusion meets Annealed Importance Samplingcited by 51
- 2021Monte Carlo Variational Auto-Encoderscited by 36
- 2021Differentiable Annealed Importance Sampling and the Perils of Gradient Noisecited by 36
- 2022Automatic Differentiation of Programs with Discrete Randomnesscited by 29
- 2022Designing Perceptual Puzzles by Differentiating Probabilistic Programscited by 12
- 2022ADEV: Sound Automatic Differentiation of Expected Values of Probabilistic Programscited by 12
- 2021Storchastic: A Framework for General Stochastic Automatic Differentiationcited by 11
- 2020Maximal couplings of the Metropolis-Hastings algorithmcited by 7
- 2023Acting as Inverse Inverse Planningcited by 5
Built on by
Citing papers per year: 2023: 2; 2024: 0; 2025: 0; 2026: 4.
6 papers.
- 2023Branches of a Tree: Taking Derivatives of Programs with Discrete and Branching Randomness in High Energy Physicscited by 6
- 2023On Feynman--Kac training of partial Bayesian neural networkscited by 4
- 2026A Blueprint for Equilibrium-Based Differentiable Continuous-Variable Thermodynamic Computingcited by 2
- 2026Gradient estimators for parameter inference in discrete stochastic kinetic modelscited by 1
- 2026Refined Gradient-Based Temperature Optimization for the Replica-Exchange Monte-Carlo Methodcited by 0
- 2026Learning Graph Energies for Finite-Time Generationcited by 0
- 2026Learning Graph Energies for Finite-Time Generation14 Oct 2026
- 2026A Blueprint for Equilibrium-Based Differentiable Continuous-Variable Thermodynamic Computing17 Jul 2026
- 2026Gradient estimators for parameter inference in discrete stochastic kinetic models2 Apr 2026
- 2026Refined Gradient-Based Temperature Optimization for the Replica-Exchange Monte-Carlo Method20 Jan 2026
- 2023On Feynman--Kac training of partial Bayesian neural networks30 Oct 2023
- 2023Branches of a Tree: Taking Derivatives of Programs with Discrete and Branching Randomness in High Energy Physics31 Aug 2023