mlsanityData to 7 Oct 2026

Error Bounds for a Diffusion Model-Based Drift Estimator

arXiv 2606.02115 ↗First posted 1 Jun 2026stat.ML cs.LGcited by 0

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

Parameter estimation in stochastic differential equations is a classical statistical problem of much importance in many scientific fields. Recent work of Tapia Costa et al. (2026) introduced a novel technique for estimating the drift when the diffusion parameter is known, using discrete samples from multiple trajectories. Their method treats drift estimation as a denoising problem, and leverages tools from (conditional) score-matching diffusion models. Although their experiments showed promising results across different drift classes, the question of theoretical guarantees for their estimator was left unanswered. In this note, we address this gap by exploiting techniques from diffusion model theory. More concretely, we derive an explicit risk bound for the time-averaged mean-squared error of said drift estimator. Our bound decomposes the risk into the (i) Euler-Maruyama discretization, (ii) score/denoiser approximation, (iii) noise initialization, and (iv) sampling variance, revealing the trade-offs between the different hyperparameters and sources of error in the estimator.

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 12 of its 29 references that are in the map. First posted on arXiv 1 Jun 2026.

Builds on

12 of 29 references.

Closest ancestors · 9

Same direction, newest first

  1. 2026Drift Estimation for Stochastic Differential Equations with Denoising Diffusion Models19 Feb
  2. 2025The Principles of Diffusion Models24 Oct
  3. 2024A Training-Free Conditional Diffusion Model for Learning Stochastic Dynamical Systems4 Oct
  4. 2024Score-based Diffusion Models via Stochastic Differential Equations -- a Technical Tutorial12 Feb
  5. 2023A Note on the Convergence of Denoising Diffusion Probabilistic Models10 Dec
  6. 2023Wasserstein Convergence Guarantees for a General Class of Score-Based Generative Models18 Nov
  7. 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptions22 Sep
  8. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  9. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 3

  1. 2022Action Matching: Learning Stochastic Dynamics from Samplescited by 94
  2. 2024Identifying Drift, Diffusion, and Causal Structure from Temporal Snapshotscited by 10
  3. 2025Drift Estimation for Diffusion Processes Using Neural Networks Based on Discretely Observed Independent Pathscited by 3

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2023A Geometric Perspective on Diffusion Models
  2. 2025Diffusion Models under Alternative Noise: Simplified Analysis and Sensitivity
  3. 2026Connections between the F"ollmer process and the denoising diffusion probabilistic model
  4. 2024Non-asymptotic bounds for forward processes in denoising diffusions: Ornstein-Uhlenbeck is hard to beat
  5. 2023ERA-Solver: Error-Robust Adams Solver for Fast Sampling of Diffusion Probabilistic Models
  6. 2026Geometry-Aware Discretization Error of Diffusion Models
  7. 2026Wasserstein bounds for denoising diffusion probabilistic models via the F"ollmer process
  8. 2023Error Bounds for Flow Matching Methods