Sticking the Landing: Simple, Lower-Variance Gradient Estimators for Variational Inference
Not on the map. Why?
Abstract (arXiv)
We propose a simple and general variant of the standard reparameterized gradient estimator for the variational evidence lower bound. Specifically, we remove a part of the total derivative with respect to the variational parameters that corresponds to the score function. Removing this term produces an unbiased gradient estimator whose variance approaches zero as the approximate posterior approaches the exact posterior. We analyze the behavior of this gradient estimator theoretically and empirically, and generalize it to more complex variational distributions such as mixtures and importance-weighted posteriors.
Key facts. Cited by 97 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 Mar 2017.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2019: 1; 2020: 0; 2021: 13; 2022: 16; 2023: 25; 2024: 23; 2025: 10; 2026: 9.
97 papers.
Sort:
By area:
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,006
- 2019Variational Mixture-of-Experts Autoencoders for Multi-Modal Deep Generative Modelscited by 177
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Modelscited by 169
- 2022An optimal control perspective on diffusion-based generative modelingcited by 145
- 2023Improved sampling via learned diffusionscited by 103
- 2023Generative Time Series Forecasting with Diffusion, Denoise, and Disentanglementcited by 89
- 2024TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolutioncited by 83
- 2021BCD Nets: Scalable Variational Approaches for Bayesian Causal Discoverycited by 68
- 2023Score Regularized Policy Optimization through Diffusion Behaviorcited by 67
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2021Bayesian Learning via Neural Schr"odinger-F"ollmer Flowscited by 56
- 2023Training Chain-of-Thought via Latent-Variable Inferencecited by 46
38 more
- 2022Variational methods for simulation-based inferencecited by 43
- 2024Score Distillation via Reparametrized DDIMcited by 41
- 2023Stochastic Optimal Control Matchingcited by 39
- 2021Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equationscited by 38
- 2021Challenges and Opportunities in High-dimensional Variational Inferencecited by 35
- 2022Mitigating Modality Collapse in Multimodal VAEs via Impartial Optimizationcited by 32
- 2023Provable convergence guarantees for black-box variational inferencecited by 29
- 2023SteinDreamer: Variance Reduction for Text-to-3D Score Distillation via Stein Identitycited by 23
- 2022Gradients should stay on Path: Better Estimators of the Reverse- and Forward KL Divergence for Normalizing Flowscited by 22
- 2022A Unified Perspective on Natural Gradient Variational Inference with Gaussian Mixture Modelscited by 22
- 2023On the Convergence of Black-Box Variational Inferencecited by 22
- 2023Transport meets Variational Inference: Controlled Monte Carlo Diffusionscited by 21
- 2021Amortized Variational Inference for Simple Hierarchical Modelscited by 19
- 2021Generalized Doubly Reparameterized Gradient Estimatorscited by 14
- 2021Nested Variational Inferencecited by 14
- 2024Scalable Bayesian Learning with posteriorscited by 13
- 2022ProbNeRF: Uncertainty-Aware Inference of 3D Shapes from 2D Imagescited by 12
- 2023MonoFlow: Rethinking Divergence GANs via the Perspective of Wasserstein Gradient Flowscited by 12
- 2024A Taxonomy of Loss Functions for Stochastic Optimal Controlcited by 11
- 2021Empirical Evaluation of Biased Methods for Alpha Divergence Minimizationcited by 10
- 2022Path-Gradient Estimators for Continuous Normalizing Flowscited by 10
- 2023Linear Convergence of Black-Box Variational Inference: Should We Stick the Landing?cited by 10
- 2023Bridging the Gap Between Variational Inference and Wasserstein Gradient Flowscited by 8
- 2025Model Immunization from a Condition Number Perspectivecited by 8
- 2024Particle Semi-Implicit Variational Inferencecited by 7
- 2021A variational approximate posterior for the deep Wishart processcited by 6
- 2022Variational Sparse Coding with Learned Thresholdingcited by 6
- 2022Markov Chain Score Ascent: A Unifying Framework of Variational Inference with Markovian Gradientscited by 6
- 2022Cooperation in the Latent Space: The Benefits of Adding Mixture Components in Variational Autoencoderscited by 6
- 2023Practical and Matching Gradient Variance Bounds for Black-Box Variational Bayesian Inferencecited by 6
- 2023Sample Average Approximation for Black-Box VIcited by 6
- 2024Partially Stochastic Infinitely Deep Bayesian Neural Networkscited by 6
- 2024On Divergence Measures for Training GFlowNetscited by 6
- 2025Control Consistency Losses for Diffusion Bridgescited by 6
- 2022Bayesian Nonparametrics for Offline Skill Discoverycited by 5
- 2022Multi-Sample Training for Neural Image Compressioncited by 5
- 2024Stable Training of Normalizing Flows for High-dimensional Variational Inferencecited by 5
- 2024Fast and Unified Path Gradient Estimators for Normalizing Flowscited by 5
50 most cited of 97 All 97 (JSON)
- 2026How I learned to stop worrying and love StopGrads: Stationarity, Convergence, and a case study on Flow Map Learning14 Sep 2026
- 2026Causal Gaussian Processes for Robust Treatment Effect Evaluation with Unobserved Confounding20 Jun 2026
- 2026Structured Nonparametric Variational Inference for Dependent Latent Modeling13 Jun 2026
- 2026H"older++: Improving the Quality-Coherence Trade-off in Multimodal VAEs11 Jun 2026
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2026How Fast Should a Model Commit to Supervision? Training Reasoning Models on the Tsallis Loss Continuum28 Apr 2026
- 2026Variational Learning of Fractional Posteriors29 Mar 2026
- 2026Deep probabilistic model synthesis enables unified modeling of whole-brain neural activity across individual subjects15 Mar 2026
- 2026Preference Score Distillation: Leveraging 2D Rewards to Align Text-to-3D Generation with Human Preference2 Mar 2026
- 2025Energy-Tweedie: Score meets Score, Energy meets Energy29 Dec 2025
- 2025Control Consistency Losses for Diffusion Bridges4 Dec 2025
- 2025Stick-Breaking Mixture Normalizing Flows with Component-Wise Tail Adaptation for Variational Inference9 Oct 2025
38 more
- 2025Rao-Blackwellised Reparameterisation Gradients9 Jun 2025
- 2025Semi-Implicit Variational Inference via Kernelized Path Gradient Descent5 Jun 2025
- 2025Revisiting Unbiased Implicit Variational Inference4 Jun 2025
- 2025Model Immunization from a Condition Number Perspective29 May 2025
- 2025Nearly Dimension-Independent Convergence of Mean-Field Black-Box Variational Inference27 May 2025
- 2025AutoSGD: Automatic Learning Rate Selection for Stochastic Gradient Descent27 May 2025
- 2025Path Gradients after Flow Matching15 May 2025
- 2024TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolution27 Nov 2024
- 2024Recursive Learning of Asymptotic Variational Objectives4 Nov 2024
- 2024On Divergence Measures for Training GFlowNets12 Oct 2024
- 2024Stochastic variance-reduced Gaussian variational inference on the Bures-Wasserstein manifold3 Oct 2024
- 2024A Taxonomy of Loss Functions for Stochastic Optimal Control1 Oct 2024
- 2024Alpha-VI DeepONet: A prior-robust variational Bayesian approach for enhancing DeepONets with uncertainty quantification1 Aug 2024
- 2024SoftCVI: Contrastive variational inference with self-generated soft labels22 Jul 2024
- 2024Particle Semi-Implicit Variational Inference30 Jun 2024
- 2024Efficient Mixture Learning in Black-Box Variational Inference11 Jun 2024
- 2024Dream-in-Style: Text-to-3D Generation Using Stylized Score Distillation5 Jun 2024
- 2024Scalable Bayesian Learning with posteriors31 May 2024
- 2024Understanding and mitigating difficulties in posterior predictive evaluation30 May 2024
- 2024Federated Learning for Non-factorizable Models using Deep Generative Prior Approximations25 May 2024
- 2024Score Distillation via Reparametrized DDIM24 May 2024
- 2024Scalable Vertical Federated Learning via Data Augmentation and Amortized Inference7 May 2024
- 2024MVEB: Self-Supervised Learning with Multi-View Entropy Bottleneck28 Mar 2024
- 2024Fast and Unified Path Gradient Estimators for Normalizing Flows23 Mar 2024
- 2024Towards Model-Agnostic Posterior Approximation for Fast and Accurate Variational Autoencoders13 Mar 2024
- 2024Nonparametric Automatic Differentiation Variational Inference with Spline Approximation10 Mar 2024
- 2024Improving Variational Autoencoder Estimation from Incomplete Data with Mixture Variational Families5 Mar 2024
- 2024Stable Training of Normalizing Flows for High-dimensional Variational Inference26 Feb 2024
- 2024Partially Stochastic Infinitely Deep Bayesian Neural Networks5 Feb 2024
- 2024Robust Inverse Graphics via Probabilistic Inference2 Feb 2024
- 2023SteinDreamer: Variance Reduction for Text-to-3D Score Distillation via Stein Identity31 Dec 2023
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Models21 Dec 2023
- 2023Online Variational Sequential Monte Carlo19 Dec 2023
- 2023Stochastic Optimal Control Matching4 Dec 2023
- 2023Training Chain-of-Thought via Latent-Variable Inference28 Nov 2023
- 2023Bridging the Gap Between Variational Inference and Wasserstein Gradient Flows31 Oct 2023
- 2023On permutation symmetries in Bayesian neural network posteriors: a variational perspective16 Oct 2023
- 2023Score Regularized Policy Optimization through Diffusion Behavior11 Oct 2023
50 newest of 97 All 97 (JSON)
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Modelscited by 169
- 2022An optimal control perspective on diffusion-based generative modelingcited by 145
- 2024TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolutioncited by 83
- 2023Score Regularized Policy Optimization through Diffusion Behaviorcited by 67
- 2024Score Distillation via Reparametrized DDIMcited by 41
- 2025Path Gradients after Flow Matchingcited by 2
- 2026Preference Score Distillation: Leveraging 2D Rewards to Align Text-to-3D Generation with Human Preferencecited by 2
- 2026Preference Score Distillation: Leveraging 2D Rewards to Align Text-to-3D Generation with Human Preference2 Mar 2026
- 2025Path Gradients after Flow Matching15 May 2025
- 2024TSD-SR: One-Step Diffusion with Target Score Distillation for Real-World Image Super-Resolution27 Nov 2024
- 2024Score Distillation via Reparametrized DDIM24 May 2024
- 2023Align Your Gaussians: Text-to-4D with Dynamic 3D Gaussians and Composed Diffusion Models21 Dec 2023
- 2023Score Regularized Policy Optimization through Diffusion Behavior11 Oct 2023
- 2022An optimal control perspective on diffusion-based generative modeling2 Nov 2022
- 2022Bayesian Nonparametrics for Offline Skill Discoverycited by 5
- 2022Bayesian Nonparametrics for Offline Skill Discovery9 Feb 2022
- 2022DreamFusion: Text-to-3D using 2D Diffusioncited by 3,006
- 2022DreamFusion: Text-to-3D using 2D Diffusion29 Sep 2022
- 2023Improved sampling via learned diffusionscited by 103
- 2023Generative Time Series Forecasting with Diffusion, Denoise, and Disentanglementcited by 89
- 2021BCD Nets: Scalable Variational Approaches for Bayesian Causal Discoverycited by 68
- 2023Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimizationcited by 57
- 2021Bayesian Learning via Neural Schr"odinger-F"ollmer Flowscited by 56
- 2022Variational methods for simulation-based inferencecited by 43
- 2023Stochastic Optimal Control Matchingcited by 39
- 2021Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equationscited by 38
- 2021Challenges and Opportunities in High-dimensional Variational Inferencecited by 35
- 2022Mitigating Modality Collapse in Multimodal VAEs via Impartial Optimizationcited by 32
- 2023Provable convergence guarantees for black-box variational inferencecited by 29
- 2023SteinDreamer: Variance Reduction for Text-to-3D Score Distillation via Stein Identitycited by 23
12 most cited of 84 All 84 (JSON)
- 2026How I learned to stop worrying and love StopGrads: Stationarity, Convergence, and a case study on Flow Map Learning14 Sep 2026
- 2026Causal Gaussian Processes for Robust Treatment Effect Evaluation with Unobserved Confounding20 Jun 2026
- 2026Structured Nonparametric Variational Inference for Dependent Latent Modeling13 Jun 2026
- 2026H"older++: Improving the Quality-Coherence Trade-off in Multimodal VAEs11 Jun 2026
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2026Variational Learning of Fractional Posteriors29 Mar 2026
- 2026Deep probabilistic model synthesis enables unified modeling of whole-brain neural activity across individual subjects15 Mar 2026
- 2025Energy-Tweedie: Score meets Score, Energy meets Energy29 Dec 2025
- 2025Control Consistency Losses for Diffusion Bridges4 Dec 2025
- 2025Stick-Breaking Mixture Normalizing Flows with Component-Wise Tail Adaptation for Variational Inference9 Oct 2025
- 2025Rao-Blackwellised Reparameterisation Gradients9 Jun 2025
- 2025Semi-Implicit Variational Inference via Kernelized Path Gradient Descent5 Jun 2025
12 newest of 84 All 84 (JSON)