Monte Carlo Gradient Estimation in Machine Learning
Not on the map. Why?
Cited by 24 in Diffusion models, but placed by its own abstract, which didn’t fit.
Abstract (arXiv)
This paper is a broad and accessible survey of the methods we have at our disposal for Monte Carlo gradient estimation in machine learning and across the statistical sciences: the problem of computing the gradient of an expectation of a function with respect to parameters defining the distribution that is integrated; the problem of sensitivity analysis. In machine learning research, this gradient problem lies at the core of many learning problems, in supervised, unsupervised and reinforcement learning. We will generally seek to rewrite such gradients in a form that allows for Monte Carlo estimation, allowing them to be easily and efficiently used and analysed. We explore three strategies--the pathwise, score function, and measure-valued gradient estimators--exploring their historical development, derivation, and underlying assumptions. We describe their use in other fields, show how they are related and can be combined, and expand on their possible generalisations. Wherever Monte Carlo gradient estimators have been derived and deployed in the past, important advances have followed. A deeper and more widely-held understanding of this problem will lead to further advances, and it is these advances that we wish to support.
Key facts. Cited by 281 papers in the map (data through 7 Oct 2026). First posted on arXiv 25 Jun 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 42; 2022: 43; 2023: 52; 2024: 52; 2025: 55; 2026: 37.
281 papers.
Sort:
By area:
- 2023Training Diffusion Models with Reinforcement Learningcited by 939
- 2023ReMax: A Simple, Effective, and Efficient Reinforcement Learning Method for Aligning Large Language Modelscited by 184
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
- 2021Neural Temporal Point Processes: A Reviewcited by 113
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 105
- 2021Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Designcited by 103
- 2022Tutorial on amortized optimizationcited by 83
- 2022Do Differentiable Simulators Give Better Policy Gradients?cited by 83
- 2025Pass@K Policy Optimization: Solving Harder Reinforcement Learning Problemscited by 83
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2021Gradients are Not All You Needcited by 82
- 2023Optimizing DDPM Sampling with Shortcut Fine-Tuningcited by 81
38 more
- 2024Large-scale Reinforcement Learning for Diffusion Modelscited by 78
- 2022Gradients without Backpropagationcited by 73
- 2021BCD Nets: Scalable Variational Approaches for Bayesian Causal Discoverycited by 68
- 2024Variational Bayesian Last Layerscited by 67
- 2021Risk-Averse Offline Reinforcement Learningcited by 65
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2021Implicit Deep Adaptive Design: Policy-Based Experimental Design without Likelihoodscited by 61
- 2022Constrained Policy Optimization via Bayesian World Modelscited by 55
- 2022A-NeSI: A Scalable Approximate Method for Probabilistic Neurosymbolic Inferencecited by 50
- 2021Variational Neural Annealingcited by 45
- 2024Discrete Probabilistic Inference as Control in Multi-path Environmentscited by 40
- 2021Challenges and Opportunities in High-dimensional Variational Inferencecited by 35
- 2022Bayesian Optimization over Discrete and Mixed Spaces via Probabilistic Reparameterizationcited by 35
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 34
- 2021Order Matters: Probabilistic Modeling of Node Sequence for Graph Generationcited by 32
- 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuningcited by 30
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewardscited by 29
- 2025Learning Unmasking Policies for Diffusion Language Modelscited by 29
- 2022Automatic Differentiation of Programs with Discrete Randomnesscited by 28
- 2021Gradient-based Bayesian Experimental Design for Implicit Models using Mutual Information Lower Boundscited by 26
- 2021Pathfinder: Parallel quasi-Newton variational inferencecited by 26
- 2023Graph Deep Learning for Time Series Forecastingcited by 26
- 2022Gradient Estimation with Discrete Stein Operatorscited by 22
- 2022An Introduction to Quantum Machine Learning for Engineerscited by 22
- 2022Sparse Graph Learning from Spatiotemporal Time Seriescited by 22
- 2021Autoregressive neural-network wavefunctions for ab initio quantum chemistrycited by 21
- 2022A Differential Entropy Estimator for Training Neural Networkscited by 20
- 2024On the limits of agency in agent-based modelscited by 20
- 2025Rethinking Diffusion Posterior Sampling: From Conditional Score Estimator to Maximizing a Posteriorcited by 20
- 2025Guiding Generative Models to Uncover Diverse and Novel Crystals via Reinforcement Learningcited by 19
- 2024Adaptive Horizon Actor-Critic for Policy Learning in Contact-Rich Differentiable Simulationcited by 18
- 2022Efficient Deep Visual and Inertial Odometry with Adaptive Visual Modality Selectioncited by 17
- 2023Black Box Variational Inference with a Deterministic Objective: Faster, More Accurate, and Even More Black Boxcited by 17
- 2024PWM: Policy Learning with Multi-Task World Modelscited by 17
- 2021Ab-initio study of interacting fermions at finite temperature with neural canonical transformationcited by 16
- 2022Amortized Variational Inference: A Systematic Reviewcited by 16
- 2021Generalized Doubly Reparameterized Gradient Estimatorscited by 14
- 2021Combining Pessimism with Optimism for Robust and Efficient Model-Based Deep Reinforcement Learningcited by 14
50 most cited of 281 All 281 (JSON)
- 2026OpenJev-RLCD: A Working RLCD Implementation30 Sep 2026
- 2026The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learning29 Sep 2026
- 2026Beyond Verbalized Confidence: Calibrating Reasoners with Differentiable Readouts28 Sep 2026
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Learning Collective Dynamics with Differentiable Gaussian Representations23 Sep 2026
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026Physical policy gradient theorem for in situ stochastic-adjoint training5 Sep 2026
- 2026On efficiency gains via augmenting a tiny sample with a massive auxiliary sample27 Aug 2026
- 2026Accelerating Evolutionary Strategy via Rao-Blackwellizing Realization of Uncertain Input3 Aug 2026
- 2026Variational-Ising-Attention:Tailored Attention Matters for Science26 Jul 2026
- 2026GradInf: Gradient Estimation as Probabilistic Inference8 Jul 2026
- 2026Inverting the Bellman Equation: From Q-Values to World Models19 Jun 2026
38 more
- 2026Learning a Sampling-Free Variational DNN Plugin from Tiny Training Sets to Refine OOD Segmentation With Uncertainty Estimation14 Jun 2026
- 2026Low-variance estimators overcome the phase-gradient bottleneck in complex-valued neural quantum states11 Jun 2026
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2026OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimation4 Jun 2026
- 2026Policy Optimization in Hybrid Discrete-Continuous Action Spaces via Mixed Gradients14 May 2026
- 2026Learning U-Statistics with Active Inference12 May 2026
- 2026Revisiting Mixture Policies in Entropy-Regularized Actor-Critic9 May 2026
- 2026Non-Myopic Active Feature Acquisition via Pathwise Policy Gradients6 May 2026
- 2026How to Approximate Inference with Subtractive Mixture Models17 Apr 2026
- 2026Gradient estimators for parameter inference in discrete stochastic kinetic models2 Apr 2026
- 2026Rdm: Re-conceptualizing Distribution Matching as a Reward for Diffusion Distillation30 Mar 2026
- 2026Optimal low-rank stochastic gradient estimation for LLM training21 Mar 2026
- 2026Goal-oriented learning of stochastic differential equations using error bounds on path-space observables20 Mar 2026
- 2026Contextual Preference Distribution Learning17 Mar 2026
- 2026Align Forward, Adapt Backward: Closing the Discretization Gap in Logic Gate Networks14 Mar 2026
- 2026A Stein Identity for q-Gaussians with Bounded Support4 Mar 2026
- 2026Exact Discrete Stochastic Simulation with Deep-Learning-Scale Gradient Optimization23 Feb 2026
- 2026Stochastic Gradient Variational Inference with Price's Gradient Estimator from Bures-Wasserstein to Parameter Space21 Feb 2026
- 2026MDP Planning as Policy Inference19 Feb 2026
- 2026Supercharging Simulation-Based Inference for Bayesian Optimal Experimental Design6 Feb 2026
- 2026Efficient Online Variational Estimation via Monte Carlo Sampling6 Feb 2026
- 2026Reparameterization Flow Policy Optimization3 Feb 2026
- 2026Refined Gradient-Based Temperature Optimization for the Replica-Exchange Monte-Carlo Method20 Jan 2026
- 2026Neural Nonmyopic Bayesian Optimization in Dynamic Cost Settings10 Jan 2026
- 2026Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion6 Jan 2026
- 2025Dichotomous Diffusion Policy Optimization31 Dec 2025
- 2025ReDiF: Reinforced Distillation for Few Step Diffusion28 Dec 2025
- 2025Learning Unmasking Policies for Diffusion Language Models9 Dec 2025
- 2025Scalable Decision Focused Learning via Online Trainable Surrogates3 Dec 2025
- 2025Internalizing Tools as Morphisms in Graded Transformers21 Nov 2025
- 2025Guiding Generative Models to Uncover Diverse and Novel Crystals via Reinforcement Learning10 Nov 2025
- 2025Fine-Tuning Diffusion-Based Recommender Systems via Reinforcement Learning with Reward Function Optimization10 Nov 2025
- 2025Malliavin Calculus as Stochastic Backpropogation2 Nov 2025
- 2025PyDPF: A Python Package for Differentiable Particle Filtering29 Oct 2025
- 2025Amortized Active Generation of Pareto Sets23 Oct 2025
- 2025Learning-To-Measure: In-Context Active Feature Acquisition14 Oct 2025
- 2025Diffusion-DFL: Decision-focused Diffusion Models for Stochastic Optimization13 Oct 2025
- 2025Understanding Sampler Stochasticity in Training Diffusion Models for RLHF12 Oct 2025
50 newest of 281 All 281 (JSON)
- 2023Training Diffusion Models with Reinforcement Learningcited by 939
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inferencecited by 173
- 2023Optimizing DDPM Sampling with Shortcut Fine-Tuningcited by 81
- 2024Large-scale Reinforcement Learning for Diffusion Modelscited by 78
- 2024Alignment of Diffusion Models: Fundamentals, Challenges, and Futurecited by 34
- 2025A Simple and Effective Reinforcement Learning Method for Text-to-Image Diffusion Fine-tuningcited by 30
- 2025Towards Better Alignment: Training Diffusion Models with Reinforcement Learning Against Sparse Rewardscited by 29
- 2025Learning Unmasking Policies for Diffusion Language Modelscited by 29
- 2025Rethinking Diffusion Posterior Sampling: From Conditional Score Estimator to Maximizing a Posteriorcited by 20
- 2024DLPO: Diffusion Model Loss-Guided Reinforcement Learning for Fine-Tuning Text-to-Speech Diffusion Modelscited by 7
- 2025Inference-Time Alignment Control for Diffusion Models with Reinforcement Learning Guidancecited by 7
- 2025Understanding Sampler Stochasticity in Training Diffusion Models for RLHFcited by 7
12 most cited of 23 All 23 (JSON)
- 2026Rdm: Re-conceptualizing Distribution Matching as a Reward for Diffusion Distillation30 Mar 2026
- 2026Critic-Guided Reinforcement Unlearning in Text-to-Image Diffusion6 Jan 2026
- 2025Dichotomous Diffusion Policy Optimization31 Dec 2025
- 2025ReDiF: Reinforced Distillation for Few Step Diffusion28 Dec 2025
- 2025Learning Unmasking Policies for Diffusion Language Models9 Dec 2025
- 2025Fine-Tuning Diffusion-Based Recommender Systems via Reinforcement Learning with Reward Function Optimization10 Nov 2025
- 2025Diffusion-DFL: Decision-focused Diffusion Models for Stochastic Optimization13 Oct 2025
- 2025Understanding Sampler Stochasticity in Training Diffusion Models for RLHF12 Oct 2025
- 2025Inference-Time Alignment Control for Diffusion Models with Reinforcement Learning Guidance28 Aug 2025
- 2025Fine-Tuning Text-to-Speech Diffusion Models Using Reinforcement Learning with Human Feedback5 Aug 2025
- 2025Seed Diffusion: A Large-Scale Diffusion Language Model with High-Speed Inference4 Aug 2025
- 2025Path Gradients after Flow Matching15 May 2025
12 newest of 23 All 23 (JSON)
- 2022A-NeSI: A Scalable Approximate Method for Probabilistic Neurosymbolic Inferencecited by 50
- 2025Neurosymbolic Diffusion Modelscited by 12
- 2024On the Hardness of Probabilistic Neurosymbolic Learningcited by 9
- 2026Beyond Verbalized Confidence: Calibrating Reasoners with Differentiable Readoutscited by 0
- 2026OpenJev-RLCD: A Working RLCD Implementationcited by 0
- 2026The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learningcited by 0
- 2026OpenJev-RLCD: A Working RLCD Implementation30 Sep 2026
- 2026The Weakest Link: Distilling LLM Reasoning with Worst-Case Constrained Reinforcement Learning29 Sep 2026
- 2026Beyond Verbalized Confidence: Calibrating Reasoners with Differentiable Readouts28 Sep 2026
- 2025Neurosymbolic Diffusion Models19 May 2025
- 2024On the Hardness of Probabilistic Neurosymbolic Learning6 Jun 2024
- 2022A-NeSI: A Scalable Approximate Method for Probabilistic Neurosymbolic Inference23 Dec 2022
- 2022A Survey on Model-based Reinforcement Learningcited by 65
- 2022Constrained Policy Optimization via Bayesian World Modelscited by 55
- 2024PWM: Policy Learning with Multi-Task World Modelscited by 17
- 2026Inverting the Bellman Equation: From Q-Values to World Modelscited by 1
- 2026Inverting the Bellman Equation: From Q-Values to World Models19 Jun 2026
- 2024PWM: Policy Learning with Multi-Task World Models2 Jul 2024
- 2022A Survey on Model-based Reinforcement Learning19 Jun 2022
- 2022Constrained Policy Optimization via Bayesian World Models24 Jan 2022
- 2024On the limits of agency in agent-based modelscited by 20
- 2024On the limits of agency in agent-based models14 Sep 2024
- 2021Neural Temporal Point Processes: A Reviewcited by 113
- 2021DiBS: Differentiable Bayesian Structure Learningcited by 105
- 2021Deep Adaptive Design: Amortizing Sequential Bayesian Experimental Designcited by 103
- 2022Tutorial on amortized optimizationcited by 83
- 2022Do Differentiable Simulators Give Better Policy Gradients?cited by 83
- 2025Pass@K Policy Optimization: Solving Harder Reinforcement Learning Problemscited by 83
- 2021A Review of the Gumbel-max Trick and its Extensions for Discrete Stochasticity in Machine Learningcited by 82
- 2021Gradients are Not All You Needcited by 82
- 2022Gradients without Backpropagationcited by 73
- 2021BCD Nets: Scalable Variational Approaches for Bayesian Causal Discoverycited by 68
- 2024Variational Bayesian Last Layerscited by 67
- 2021Implicit Deep Adaptive Design: Policy-Based Experimental Design without Likelihoodscited by 61
12 most cited of 242 All 242 (JSON)
- 2026Brenier Meets Adversarial Training: Optimal Transport Geometry for Robust Learning25 Sep 2026
- 2026Learning Collective Dynamics with Differentiable Gaussian Representations23 Sep 2026
- 2026Gradient-estimator design overcomes trainability barriers in neural-network-based variational optimization16 Sep 2026
- 2026Physical policy gradient theorem for in situ stochastic-adjoint training5 Sep 2026
- 2026On efficiency gains via augmenting a tiny sample with a massive auxiliary sample27 Aug 2026
- 2026Accelerating Evolutionary Strategy via Rao-Blackwellizing Realization of Uncertain Input3 Aug 2026
- 2026Variational-Ising-Attention:Tailored Attention Matters for Science26 Jul 2026
- 2026GradInf: Gradient Estimation as Probabilistic Inference8 Jul 2026
- 2026Learning a Sampling-Free Variational DNN Plugin from Tiny Training Sets to Refine OOD Segmentation With Uncertainty Estimation14 Jun 2026
- 2026Low-variance estimators overcome the phase-gradient bottleneck in complex-valued neural quantum states11 Jun 2026
- 2026Large-scale empirical tuning and comparison of default optimizers for variational inference5 Jun 2026
- 2026OrderGrad: Optimizing Beyond the Mean with Order-Statistic Policy Gradient Estimation4 Jun 2026
12 newest of 242 All 242 (JSON)