mlsanityData to 7 Oct 2026

Malliavin Calculus as Stochastic Backpropogation

arXiv 2602.17013 ↗First posted 2 Nov 2025cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

We establish a rigorous connection between pathwise (reparameterization) and score-function (Malliavin) gradient estimators by showing that both arise from the Malliavin integration-by-parts identity. Building on this equivalence, we introduce a unified and variance-aware hybrid estimator that adaptively combines pathwise and Malliavin gradients using their empirical covariance structure. The resulting formulation provides a principled understanding of stochastic backpropagation and achieves minimum variance among all unbiased linear combinations, with closed-form finite-sample convergence bounds. We demonstrate 9% variance reduction on VAEs (CIFAR-10) and up to 35% on strongly-coupled synthetic problems. Exploratory policy gradient experiments reveal that non-stationary optimization landscapes present challenges for the hybrid approach, highlighting important directions for future work. Overall, this work positions Malliavin calculus as a conceptually unifying and practically interpretable framework for stochastic gradient estimation, clarifying when hybrid approaches provide tangible benefits and when they face inherent limitations.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 11 of its 37 references that are in the map. First posted on arXiv 2 Nov 2025.

Builds on

11 of 37 references.

References · 11

  1. 2020Denoising Diffusion Probabilistic Modelscited by 20,851
  2. 2018Neural Ordinary Differential Equationscited by 3,733
  3. 2016Categorical Reparameterization with Gumbel-Softmaxcited by 2,708
  4. 2016The Concrete Distribution: A Continuous Relaxation of Discrete Random Variablescited by 1,135
  5. 2016Stein Variational Gradient Descent: A General Purpose Bayesian Inference Algorithmcited by 498
  6. 2013Black Box Variational Inferencecited by 317
  7. 2019Monte Carlo Gradient Estimation in Machine Learningcited by 281
  8. 2017Backpropagation through the Void: Optimizing control variates for black-box gradient estimationcited by 104
  9. 2013The Optimal Reward Baseline for Gradient-Based Reinforcement Learningcited by 84
  10. 2017REBAR: Low-variance, unbiased gradient estimates for discrete latent variable modelscited by 78
  11. 2021Efficient and Accurate Gradients for Neural SDEscited by 68

Built on by

Not cited yet.