Soft Diffusion: Score Matching for General Corruptions
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
We define a broader family of corruption processes that generalizes previously known diffusion models. To reverse these general diffusions, we propose a new objective called Soft Score Matching that provably learns the score function for any linear corruption process and yields state of the art results for CelebA. Soft Score Matching incorporates the degradation process in the network. Our new loss trains the model to predict a clean image, textitthat after corruption, matches the diffused observation. We show that our objective learns the gradient of the likelihood under suitable regularity conditions for a family of corruption processes. We further develop a principled way to select the corruption levels for general diffusion processes and a novel sampling method that we call Momentum Sampler. We show experimentally that our framework works for general linear corruption processes, such as Gaussian blur and masking. We achieve state-of-the-art FID score 1.85 on CelebA-64, outperforming all previous linear diffusion models. We also show significant computational benefits compared to vanilla denoising diffusion.
Key facts. Cited by 102 papers in the map (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 47 of its 57 references that are in the map. First posted on arXiv 12 Sep 2022.
Builds on
47 of 57 references.
Closest ancestors · 23
Same direction, newest first
- 2022Diffusion Models in Vision: A Survey10 Sep
- 2022Diffusion Models: A Comprehensive Survey of Methods and Applications2 Sep
- 2022Cold Diffusion: Inverting Arbitrary Image Transforms Without Noise19 Aug
- 2022Pyramidal Denoising Diffusion Probabilistic Models3 Aug
- 2022Progressive Deblurring of Diffusion Models for Coarse-to-Fine Image Synthesis16 Jul
- 2022Accelerating Score-based Generative Models for High-Resolution Image Synthesis8 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2022Pseudo Numerical Methods for Diffusion Models on Manifolds20 Feb
- 2022Analytic-DPM: an Analytic Estimate of the Optimal Reverse Variance in Diffusion Probabilistic Models17 Jan
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Heavy-tailed denoising score matching17 Dec
- 2021Tackling the Generative Learning Trilemma with Denoising Diffusion GANs15 Dec
11 more
- 2021Denoising Diffusion Gamma Models10 Oct
- 2021Variational Diffusion Models1 Jul
- 2021Soft Truncation: A Universal Training Technique of Score-based Diffusion Model for High Precision Score Estimation10 Jun
- 2021Cascaded Diffusion Models for High Fidelity Image Generation30 May
- 2021Gotta Go Fast When Generating Data with Score-Based Models28 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Improved Denoising Diffusion Probabilistic Models18 Feb
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
- 2020Improved Techniques for Training Score-Based Generative Models16 Jun
Other references · 24
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Hierarchical Text-Conditional Image Generation with CLIP Latentscited by 6,585
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,355
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2018Neural Ordinary Differential Equationscited by 3,779
- 2022Video Diffusion Modelscited by 2,305
- 2021Image Super-Resolution via Iterative Refinementcited by 1,511
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,354
- 2022Denoising Diffusion Restoration Modelscited by 963
- 2021Blended Diffusion for Text-driven Editing of Natural Imagescited by 927
- 2022Equivariant Diffusion for Molecule Generation in 3Dcited by 677
- 2022Improving Diffusion Models for Inverse Problems using Manifold Constraintscited by 508
12 more
- 2021Robust Compressed Sensing MRI with Deep Generative Priorscited by 286
- 2021Deblurring via Stochastic Refinementcited by 249
- 2021Autoregressive Diffusion Modelscited by 210
- 2022Speech Enhancement and Dereverberation with Diffusion-based Generative Modelscited by 197
- 2021SNIPS: Solving Noisy Inverse Problems Stochasticallycited by 191
- 2022Generative Modelling With Inverse Heat Dissipationcited by 167
- 2022Optimal Transport Tools (OTT): A JAX Toolbox for all things Wassersteincited by 101
- 2021Bayesian imaging using Plug & Play priors: when Langevin meets Tweediecited by 97
- 2020Interacting particle solutions of Fokker-Planck equations through gradient-log-density estimationcited by 88
- 2022Universal Speech Enhancement with Score-based Diffusioncited by 75
- 2021Beyond In-Place Corruption: Insertion and Deletion In Denoising Probabilistic Modelscited by 23
- 2022Score-Guided Intermediate Layer Optimization: Fast Langevin Mixing for Inverse Problemscited by 11
Built on by
Citing papers per year: 2022: 11; 2023: 44; 2024: 19; 2025: 16; 2026: 12.
102 papers.
Sort:
By area:
- 2023InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generationcited by 395
- 2022DiffRF: Rendering-Guided 3D Radiance Field Diffusioncited by 204
- 2024A Survey on Diffusion Models for Inverse Problemscited by 180
- 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 165
- 2023Solving Linear Inverse Problems Provably via Posterior Sampling with Latent Diffusion Modelscited by 157
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Modelscited by 133
- 2023Ambient Diffusion: Learning Clean Distributions from Corrupted Datacited by 111
- 2023DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal Forecastingcited by 109
- 2022How to Backdoor Diffusion Models?cited by 104
- 2022Refining Generative Process with Discriminator Guidance in Score-based Diffusion Modelscited by 100
- 2023Stochastic interpolants with data-dependent couplingscited by 100
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
38 more
- 2023Refusion: Enabling Large-Size Realistic Image Restoration with Latent-Space Diffusion Modelscited by 86
- 2024Variational Flow Matching for Graph Generationcited by 83
- 2023Beyond First-Order Tweedie: Solving Inverse Problems using Latent Diffusioncited by 78
- 2022Multiscale Structure Guided Diffusion for Image Deblurringcited by 77
- 2024A Survey on Diffusion Models for Time Series and Spatio-Temporal Datacited by 67
- 2024Diffusion Models, Image Super-Resolution And Everything: A Surveycited by 64
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 60
- 2023Consistent Diffusion Models: Mitigating Sampling Drift by Learning to be Consistentcited by 58
- 2022Efficient Diffusion Models for Vision: A Surveycited by 51
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modellingcited by 45
- 2022Multiresolution Textual Inversioncited by 34
- 2024The Stronger the Diffusion Model, the Easier the Backdoor: Data Poisoning to Induce Copyright Breaches Without Adjusting Finetuning Pipelinecited by 33
- 2022f-DM: A Multi-stage Diffusion Model via Progressive Signal Transformationcited by 32
- 2024Self-Consistent Recursive Diffusion Bridge for Medical Image Translationcited by 26
- 2022Spectral Diffusion Processescited by 25
- 2023Graph Signal Diffusion Model for Collaborative Filteringcited by 25
- 2023The role of noise in denoising models for anomaly detection in medical imagescited by 23
- 2023Learning multi-scale local conditional probability models of imagescited by 21
- 2023Neural Diffusion Modelscited by 21
- 2024Warped Diffusion: Solving Video Inverse Problems with Image Diffusion Modelscited by 21
- 2024Deep MMD Gradient Flow without adversarial trainingcited by 19
- 2023Learning to Jump: Thinning and Thickening Latent Counts for Generative Modelingcited by 18
- 2025Ambient Diffusion Omni: Training Good Models with Bad Datacited by 18
- 2023Beta Diffusioncited by 17
- 2023DiffEnc: Variational Diffusion with a Learned Encodercited by 17
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusioncited by 17
- 2022Boomerang: Local sampling on image manifolds using diffusion modelscited by 16
- 2023Enhancing Diffusion Models with 3D Perspective Geometry Constraintscited by 16
- 2022DriftRec: Adapting diffusion models to blind JPEG restorationcited by 15
- 2023DiracDiffusion: Denoising and Incremental Reconstruction with Assured Data-Consistencycited by 15
- 2023Denoising Diffusion Semantic Segmentation with Mask Prior Modelingcited by 15
- 2023Removing Structured Noise with Diffusion Modelscited by 14
- 2023Learning Fourier-Constrained Diffusion Bridges for MRI Reconstructioncited by 14
- 2023Star-Shaped Denoising Diffusion Probabilistic Modelscited by 13
- 2023Generative Fractional Diffusion Modelscited by 12
- 2025Nonisotropic Gaussian Diffusion for Realistic 3D Human Motion Predictioncited by 12
- 2025FlexiDiT: Your Diffusion Transformer Can Easily Generate High-Quality Samples with Less Computecited by 12
- 2023Masked Diffusion as Self-supervised Representation Learnercited by 11
50 most cited of 102 All 102 (JSON)
- 2026Colored Graph Signal Diffusion Models12 Oct 2026
- 2026Mean Velocity Matching: Rethinking Generative Dynamics in Diffusion Models21 Sep 2026
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modeling24 Aug 2026
- 2026ARCHER: Amortized cross-specimen pose estimation for cryo-electron microscopy22 Aug 2026
- 2026Differencing the Diffusion Trajectory toward Uncertain Components for Time Series Forecasting12 Jun 2026
- 2026Optimal Transport Flow Matching by Design2 Jun 2026
- 2026Let EEG Models Learn EEG20 May 2026
- 2026Learning Normalized Energy Models for Linear Inverse Problems15 May 2026
- 2026Is Monotonic Sampling Necessary in Diffusion Models?12 May 2026
- 2026Can We Change the Stroke Size for Easier Diffusion?25 Mar 2026
- 2026Selective Denoising Diffusion Model for Time Series Anomaly Detection27 Feb 2026
- 2026Insertion Based Sequence Generation with Learnable Order Dynamics21 Feb 2026
38 more
- 2025Warm Diffusion: Recipe for Blur-Noise Mixture Diffusion Models21 Nov 2025
- 2025Fractional Diffusion Bridge Models3 Nov 2025
- 2025DistillKac: Few-Step Image Generation via Damped Wave Equations25 Sep 2025
- 2025Beyond Blur: A Fluid Perspective on Generative Diffusion Models20 Jun 2025
- 2025Ambient Diffusion Omni: Training Good Models with Bad Data10 Jun 2025
- 2025Diffusion Models under Alternative Noise: Simplified Analysis and Sensitivity10 Jun 2025
- 2025What is Adversarial Training for Diffusion Models?27 May 2025
- 2025Whitened Score Diffusion: A Structured Prior for Imaging Inverse Problems15 May 2025
- 2025Efficient Flow Matching using Latent Variables7 May 2025
- 2025Learning Energy-Based Generative Models via Potential Flow: A Variational Principle Approach to Probability Density Homotopy Matching22 Apr 2025
- 2025MirrorVerse: Pushing Diffusion Models to Realistically Reflect the World21 Apr 2025
- 2025GIViC: Generative Implicit Video Compression25 Mar 2025
- 2025Your Text Encoder Can Be An Object-Level Watermarking Controller15 Mar 2025
- 2025Does Generation Require Memorization? Creative Diffusion Models using Ambient Diffusion28 Feb 2025
- 2025FlexiDiT: Your Diffusion Transformer Can Easily Generate High-Quality Samples with Less Compute27 Feb 2025
- 2025Nonisotropic Gaussian Diffusion for Realistic 3D Human Motion Prediction10 Jan 2025
- 2024S-Diff: An Anisotropic Diffusion Model for Collaborative Filtering in Spectral Domain31 Dec 2024
- 2024PQD: Post-training Quantization for Efficient Diffusion Models30 Dec 2024
- 2024Equivariant Blurring Diffusion for Hierarchical Molecular Conformer Generation26 Oct 2024
- 2024Warped Diffusion: Solving Video Inverse Problems with Image Diffusion Models21 Oct 2024
- 2024Edge-preserving noise for diffusion models2 Oct 2024
- 2024A Survey on Diffusion Models for Inverse Problems30 Sep 2024
- 2024CPSample: Classifier Protected Sampling for Guarding Training Data During Diffusion11 Sep 2024
- 2024Sequential Posterior Sampling with Diffusion Models9 Sep 2024
- 2024Particle-Filtering-based Latent Diffusion for Inverse Problems25 Aug 2024
- 2024Bayesian Conditioned Diffusion Models for Inverse Problems14 Jun 2024
- 2024Variational Flow Matching for Graph Generation7 Jun 2024
- 2024Self-Consistent Recursive Diffusion Bridge for Medical Image Translation10 May 2024
- 2024Deep MMD Gradient Flow without adversarial training10 May 2024
- 2024A Survey on Diffusion Models for Time Series and Spatio-Temporal Data29 Apr 2024
- 2024Neural Flow Diffusion Models: Learnable Forward Process for Improved Diffusion Modelling19 Apr 2024
- 2024Efficient Denoising using Score Embedding in Score-based Diffusion Models10 Apr 2024
- 2024Structure-Guided Adversarial Training of Diffusion Models27 Feb 2024
- 2024The Stronger the Diffusion Model, the Easier the Backdoor: Data Poisoning to Induce Copyright Breaches Without Adjusting Finetuning Pipeline7 Jan 2024
- 2024Diffusion Models, Image Super-Resolution And Everything: A Survey1 Jan 2024
- 2023Diffusion-C: Unveiling the Generative Challenges of Diffusion Models through Corrupted Data14 Dec 2023
- 2023Compensation Sampling for Improved Convergence in Diffusion Models11 Dec 2023
- 2023Diffusion Reflectance Map: Single-Image Stochastic Inverse Rendering of Illumination and Reflectance7 Dec 2023
50 newest of 102 All 102 (JSON)
- 2023InstaFlow: One Step is Enough for High-Quality Diffusion-Based Text-to-Image Generationcited by 395
- 2022DiffRF: Rendering-Guided 3D Radiance Field Diffusioncited by 204
- 2024A Survey on Diffusion Models for Inverse Problemscited by 180
- 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 165
- 2023Solving Linear Inverse Problems Provably via Posterior Sampling with Latent Diffusion Modelscited by 157
- 2023Patch Diffusion: Faster and More Data-Efficient Training of Diffusion Modelscited by 133
- 2023DYffusion: A Dynamics-informed Diffusion Model for Spatiotemporal Forecastingcited by 109
- 2022How to Backdoor Diffusion Models?cited by 104
- 2022Refining Generative Process with Discriminator Guidance in Score-based Diffusion Modelscited by 100
- 2023Minimizing Trajectory Curvature of ODE-based Generative Modelscited by 89
- 2023Refusion: Enabling Large-Size Realistic Image Restoration with Latent-Space Diffusion Modelscited by 86
- 2024Variational Flow Matching for Graph Generationcited by 83
12 most cited of 82 All 82 (JSON)
- 2026Colored Graph Signal Diffusion Models12 Oct 2026
- 2026Mean Velocity Matching: Rethinking Generative Dynamics in Diffusion Models21 Sep 2026
- 2026Renormalization Group Flow Matching for Scalable Local Generative Modeling24 Aug 2026
- 2026Differencing the Diffusion Trajectory toward Uncertain Components for Time Series Forecasting12 Jun 2026
- 2026Optimal Transport Flow Matching by Design2 Jun 2026
- 2026Learning Normalized Energy Models for Linear Inverse Problems15 May 2026
- 2026Is Monotonic Sampling Necessary in Diffusion Models?12 May 2026
- 2026Can We Change the Stroke Size for Easier Diffusion?25 Mar 2026
- 2026Selective Denoising Diffusion Model for Time Series Anomaly Detection27 Feb 2026
- 2026Insertion Based Sequence Generation with Learnable Order Dynamics21 Feb 2026
- 2025Warm Diffusion: Recipe for Blur-Noise Mixture Diffusion Models21 Nov 2025
- 2025Beyond Blur: A Fluid Perspective on Generative Diffusion Models20 Jun 2025
12 newest of 82 All 82 (JSON)
- 2023Ambient Diffusion: Learning Clean Distributions from Corrupted Datacited by 111
- 2023Stochastic interpolants with data-dependent couplingscited by 100
- 2022Multiresolution Textual Inversioncited by 34
- 2023Learning multi-scale local conditional probability models of imagescited by 21
- 2024Deep MMD Gradient Flow without adversarial trainingcited by 19
- 2023Learning to Jump: Thinning and Thickening Latent Counts for Generative Modelingcited by 18
- 2025Nonisotropic Gaussian Diffusion for Realistic 3D Human Motion Predictioncited by 12
- 2023Diffusion Reflectance Map: Single-Image Stochastic Inverse Rendering of Illumination and Reflectancecited by 11
- 2023Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D Diffusioncited by 10
- 2025GIViC: Generative Implicit Video Compressioncited by 10
- 2026Let EEG Models Learn EEGcited by 7
- 2023Deep Classifier Mimicry without Data Accesscited by 6
12 most cited of 19 All 19 (JSON)
- 2026ARCHER: Amortized cross-specimen pose estimation for cryo-electron microscopy22 Aug 2026
- 2026Let EEG Models Learn EEG20 May 2026
- 2025Fractional Diffusion Bridge Models3 Nov 2025
- 2025DistillKac: Few-Step Image Generation via Damped Wave Equations25 Sep 2025
- 2025Learning Energy-Based Generative Models via Potential Flow: A Variational Principle Approach to Probability Density Homotopy Matching22 Apr 2025
- 2025GIViC: Generative Implicit Video Compression25 Mar 2025
- 2025Nonisotropic Gaussian Diffusion for Realistic 3D Human Motion Prediction10 Jan 2025
- 2024Deep MMD Gradient Flow without adversarial training10 May 2024
- 2023Diffusion Reflectance Map: Single-Image Stochastic Inverse Rendering of Illumination and Reflectance7 Dec 2023
- 2023Molecule Joint Auto-Encoding: Trajectory Pretraining with 2D and 3D Diffusion6 Dec 2023
- 2023Stochastic interpolants with data-dependent couplings5 Oct 2023
- 2023Physics-Informed DeepMRI: Bridging the Gap from Heat Diffusion to k-Space Interpolation30 Aug 2023
12 newest of 19 All 19 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2024From optimal score matching to optimal sampling
- 2024Improving Diffusion Models's Data-Corruption Resistance using Scheduled Pseudo-Huber Loss
- 2024Random Walks with Tweedie: A Unified View of Score-Based Diffusion Models
- 2023Reflected Diffusion Models
- 2021Maximum Likelihood Training of Score-Based Diffusion Models
- 2025Denoising Score Distillation: From Noisy Diffusion Pretraining to One-Step High-Quality Generation
- 2025Denoising Score Matching with Random Features: Insights on Diffusion Models from Precise Learning Curves
- 2024Slight Corruption in Pre-training Data Makes Better Diffusion Models