arXiv:2605.01568cs.CV2026-05

统一了图像增强的随机过程,揭示不同方法的本质差异。

Unifying Deep Stochastic Processes for Image Enhancement

论文配图:Unifying Deep Stochastic Processes for Image Enhancement
图 1 · 摘自论文原文
  • 从同一随机微分方程出发,分类三类连续时间过程。
  • 实验证明无方法始终最优,设计选择影响性能显著。
  • 开源ItoVision库,支持快速原型与公平对比。

深度随机过程已成为图像增强的核心范式,许多方法显式地将随机轨迹条件于退化输入。然而,这些条件过程与标准扩散模型之间的关系尚不清晰。本文通过将近期方法归类为三类连续时间过程——无条件扩散模型、Ornstein-Uhlenbeck (OU) 过程和扩散桥,提出了一个统一的视角。我们证明,所有方法均可由同一随机微分方程(SDE)推导而来。该框架明确指出,看似不同的方法主要区别在于漂移项、扩散项、终端分布和边界条件,而调度器与采样器是独立的设计选择。基于此统一框架,我们在多个图像增强任务上采用相同架构和训练协议进行了受控实验。结果表明,不存在始终占优的方法;我们成功识别并解耦了对性能影响最显著的设计因素。最后,我们发布了 ItoVision,一个模块化的 PyTorch 库,实现了该统一框架,支持快速原型设计与方法间的公平比较。

原文摘要 · Abstract (English)

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.

图像增强随机过程扩散模型PyTorch

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。