mlsanityData to 7 Oct 2026

Quantifying Epistemic Uncertainty in Diffusion Models

arXiv 2602.09170 ↗First posted 9 Feb 2026stat.ML cs.AI cs.LGcited by 2

On the map: Diffusion models → Diffusion foundations and sampling

Abstract (arXiv)

To ensure high quality outputs, it is important to quantify the epistemic uncertainty of diffusion models. Existing methods are often unreliable because they mix epistemic and aleatoric uncertainty. We introduce a method based on Fisher information that explicitly isolates epistemic variance, producing more reliable plausibility scores for generated data. To make this approach scalable, we propose FLARE (Fisher-Laplace Randomized Estimator), which approximates the Fisher information using a uniformly random subset of model parameters. Empirically, FLARE improves uncertainty estimation in synthetic time-series generation tasks, achieving more accurate and reliable filtering than other methods. Theoretically, we bound the convergence rate of our randomized approximation and provide analytic and empirical evidence that last-layer Laplace approximations are insufficient for this task.

Key facts. Cited by 2 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 34 of its 40 references that are in the map. First posted on arXiv 9 Feb 2026.

Builds on

34 of 40 references.

Closest ancestors · 15

Same direction, newest first

  1. 2025DDPM Score Matching and Distribution Learning7 Apr
  2. 2025Generative Uncertainty in Diffusion Models28 Feb
  3. 2024Diffusion Model Guided Sampling with Pixel-Wise Aleatoric Uncertainty Estimation29 Nov
  4. 2024Zero-Shot Uncertainty Quantification using Diffusion Probabilistic Models8 Aug
  5. 2024Shedding Light on Large Generative Networks: Estimating Epistemic Uncertainty in Diffusion Models5 Jun
  6. 2024Scaling Rectified Flow Transformers for High-Resolution Image Synthesis5 Mar
  7. 2024Estimating Epistemic and Aleatoric Uncertainty with a Single Model5 Feb
  8. 2023BayesDiff: Estimating Pixel-wise Uncertainty in Diffusion via Bayesian Inference17 Oct
  9. 2023Efficient Diffusion Training via Min-SNR Weighting Strategy16 Mar
  10. 2022Sampling is as easy as learning the score: theory for diffusion models with minimal data assumptions22 Sep
  11. 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
  12. 2021Improved Denoising Diffusion Probabilistic Models18 Feb
3 more
  1. 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
  2. 2020Denoising Diffusion Implicit Models6 Oct
  3. 2020Denoising Diffusion Probabilistic Models19 Jun

Other references · 19

  1. 2022Flow Matching for Generative Modelingcited by 5,426
  2. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  3. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
  4. 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
  5. 2016Improved Techniques for Training GANscited by 2,938
  6. 2017What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision?cited by 2,335
  7. 2019Normalizing Flows for Probabilistic Modeling and Inferencecited by 1,315
  8. 2019Aleatoric and Epistemic Uncertainty in Machine Learning: An Introduction to Concepts and Methodscited by 903
  9. 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 555
  10. 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
  11. 2024Flow Matching Guide and Codecited by 291
  12. 2014Sketching as a Tool for Numerical Linear Algebracited by 285
7 more
  1. 2024Understanding Hallucinations in Diffusion Models through Mode Interpolationcited by 87
  2. 2020Uncertainty Quantification Using Neural Networks for Molecular Property Predictioncited by 66
  3. 2023Randomized Numerical Linear Algebra : A Perspective on the Field With an Eye to Softwarecited by 38
  4. 2017Lectures on Randomized Numerical Linear Algebracited by 19
  5. 2019Uncertainty Quantification with Generative Modelscited by 17
  6. 2022UQGAN: A Unified Model for Uncertainty Quantification of Deep Classifiers trained via Conditional GANscited by 16
  7. 2016Randomized Matrix Decompositions using Rcited by 13

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2026: 2.

2 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2023User-defined Event Sampling and Uncertainty Quantification in Diffusion Models for Physical Dynamical Systems
  2. 2025Diffusion Transformers for Imputation: Statistical Efficiency and Uncertainty Quantification
  3. 2025A Review of Diffusion-based Simulation-Based Inference: Foundations and Applications in Non-Ideal Data Scenarios
  4. 2026DiffPTS: Rethinking Diffusion ELBO for Probabilistic Time Series Forecasting
  5. 2025Diffusion Models in Simulation-Based Inference: A Tutorial Review
  6. 2024Towards a mathematical theory for consistency training in diffusion models
  7. 2025On the Collapse Errors Induced by the Deterministic Sampler for Diffusion Models
  8. 2026When Denoising Hurts: Rethinking the Terminal Step of Diffusion Time Series Forecasters -- Extended Version