Unifying Deep Stochastic Processes for Image Enhancement
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Deep stochastic processes have recently become a central paradigm for image enhancement, with many methods explicitly conditioning the stochastic trajectory on the degraded input. However, the relationship between these conditional processes and standard diffusion models remains unclear. In this work, we introduce a unified perspective on stochastic image enhancement by classifying recent methods into three families of continuous-time processes: unconditional diffusion models, Ornstein-Uhlenbeck (OU) processes, and diffusion bridges. We show that all of these approaches arise from a common stochastic differential equation (SDE) formulation. This framework makes explicit that seemingly disparate methods differ primarily in their drift and diffusion terms, terminal distributions, and boundary conditions, while schedulers and samplers constitute orthogonal design choices. Leveraging this unification, we conduct a controlled empirical study across multiple image enhancement tasks using identical architectures and training protocols. Our results reveal no consistently dominant method; instead, we identify and disentangle the specific design choices that most strongly influence performance. Finally, we release ItoVision, a modular PyTorch library that implements the unified framework and enables rapid prototyping and fair comparison of stochastic image enhancement methods.
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 49 of its 63 references that are in the map. First posted on arXiv 2 May 2026.
Builds on
49 of 63 references.
Closest ancestors · 13
Same direction, newest first
- 2024Schr"odinger Bridge Flow for Unpaired Data Translation14 Sep
- 2023Denoising Diffusion Bridge Models29 Sep
- 2023Unpaired Image-to-Image Translation via Neural Schr"odinger Bridge24 May
- 2023I2SB: Image-to-Image Schr"odinger Bridge12 Feb
- 2022Cold Diffusion: Inverting Arbitrary Image Transforms Without Noise19 Aug
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 2021Diffusion Normalizing Flow14 Oct
- 2021Diffusion Schr"odinger Bridge with Applications to Score-Based Generative Modeling1 Jun
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
1 more
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 36
- 2018The Unreasonable Effectiveness of Deep Features as a Perceptual Metriccited by 11,029
- 2014ImageNet Large Scale Visual Recognition Challengecited by 8,441
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2018A Style-Based Generator Architecture for Generative Adversarial Networkscited by 5,986
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2016Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Networkcited by 1,898
- 2021Image Super-Resolution via Iterative Refinementcited by 1,506
- 2021Palette: Image-to-Image Diffusion Modelscited by 1,296
- 2022Denoising Diffusion Restoration Modelscited by 951
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
24 more
- 2023Improving and generalizing flow-based generative models with minibatch optimal transportcited by 835
- 2018Deep Retinex Decomposition for Low-Light Enhancementcited by 720
- 2022Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Modelcited by 574
- 2025Mean Flows for One-step Generative Modelingcited by 524
- 2022Improving Diffusion Models for Inverse Problems using Manifold Constraintscited by 501
- 2023ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shiftingcited by 360
- 2019Kindling the Darkness: A Practical Low-light Image Enhancercited by 355
- 2017Are GANs Created Equal? A Large-Scale Studycited by 267
- 2023Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancementcited by 228
- 2022BBDM: Image-to-image Translation with Brownian Bridge Diffusion Modelscited by 222
- 2021Likelihood Training of Schr"odinger Bridge using Forward-Backward SDEs Theorycited by 216
- 2022Dual Diffusion Implicit Bridges for Image-to-Image Translationcited by 212
- 2021SNIPS: Solving Noisy Inverse Problems Stochasticallycited by 191
- 2023Image Restoration with Mean-Reverting Stochastic Differential Equationscited by 182
- 2023Inversion by Direct Iteration: An Alternative to Denoising Diffusion for Image Restorationcited by 164
- 2023Diffusion Models for Image Restoration and Enhancement: A Comprehensive Surveycited by 108
- 2023Global Structure-Aware Diffusion Process for Low-Light Image Enhancementcited by 92
- 2023Refusion: Enabling Large-Size Realistic Image Restoration with Latent-Space Diffusion Modelscited by 86
- 2023Pyramid Diffusion Models For Low-light Image Enhancementcited by 67
- 2023PGDiff: Guiding Diffusion Models for Versatile Face Restoration via Partial Guidancecited by 49
- 2023Image Restoration Through Generalized Ornstein-Uhlenbeck Bridgecited by 41
- 2025UniDB: A Unified Diffusion Bridge Framework via Stochastic Optimal Controlcited by 18
- 2024Implicit Image-to-Image Schrodinger Bridge for Image Restorationcited by 13
- 2025Diffusion Models are Secretly Exchangeable: Parallelizing DDPMs via Autospeculationcited by 11
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023A Unified Conditional Framework for Diffusion-based Image Restoration
- 2022Diffusion Models in Vision: A Survey
- 2023Exploring the Optimal Choice for Generative Processes in Diffusion Models: Ordinary vs Stochastic Differential Equations
- 2023The Blessing of Randomness: SDE Beats ODE in General Diffusion-based Image Editing
- 2023The Missing U for Efficient Diffusion Models
- 2026A Tutorial on Diffusion Theory: From Differential Equations to Diffusion Models
- 2025Warm Diffusion: Recipe for Blur-Noise Mixture Diffusion Models
- 2023Generalised Diffusion Probabilistic Scale-Spaces