arXiv:2505.21528cs.CVcs.AI2025-05TPAMI被引 1

提出统一且快速采样的扩散桥框架,提升图像修复细节质量。

A Unified and Fast-Sampling Diffusion Bridge Framework via Stochastic Optimal Control

  • 基于随机最优控制重构扩散桥问题,统一现有方法
  • 可调终端惩罚系数显著改善图像细节保留效果
  • 无需训练的闭式解加速采样,适合低步数生成

近期扩散桥模型利用Doob's $h$-transform在分布间建立固定端点,已在图像翻译与修复任务中表现良好。然而这些方法常导致图像模糊或过度平滑,且缺乏解释其局限性的完整理论基础。为此,我们提出UniDB,一种基于随机最优控制(SOC)的统一且快速采样的扩散桥框架。通过将问题重构为SOC优化,我们证明采用Doob's $h$-transform的现有方法是当终端惩罚系数趋于无穷时的特例。引入可调终端惩罚系数,使控制代价与终端约束达到最优平衡,显著提升细节保留与输出质量。为避免迭代欧拉采样带来的计算开销,我们推导出反向SDE的精确闭式解,设计无需训练的加速算法;同时以更稳定的数据预测替代传统噪声预测,并引入SDE-Corrector机制,在低步数下维持感知质量,有效减少误差累积。大量实验验证了该框架在多样化图像修复任务中的优越性与适应性。代码已公开于https://github.com/2769433owo/UniDB-plusplus。

原文摘要 · Abstract (English)

Recent advances in diffusion bridge models leverage Doob's $h$-transform to establish fixed endpoints between distributions, demonstrating promising results in image translation and restoration tasks. However, these approaches often produce blurred or excessively smoothed image details and lack a comprehensive theoretical foundation to explain these shortcomings. To address these limitations, we propose UniDB, a unified and fast-sampling framework for diffusion bridges based on Stochastic Optimal Control (SOC). We reformulate the problem through an SOC-based optimization, proving that existing diffusion bridges employing Doob's $h$-transform constitute a special case, emerging when the terminal penalty coefficient in the SOC cost function tends to infinity. By incorporating a tunable terminal penalty coefficient, UniDB achieves an optimal balance between control costs and terminal penalties, substantially improving detail preservation and output quality. To avoid computationally expensive costs of iterative Euler sampling methods in UniDB, we design a training-free accelerated algorithm by deriving exact closed-form solutions for UniDB's reverse-time SDE. It is further complemented by replacing conventional noise prediction with a more stable data prediction model, along with an SDE-Corrector mechanism that maintains perceptual quality for low-step regimes, effectively reducing error accumulation. Extensive experiments across diverse image restoration tasks validate the superiority and adaptability of the proposed framework, bridging the gap between theoretical generality and practical efficiency. Our code is available online https://github.com/2769433owo/UniDB-plusplus.

扩散模型图像修复随机控制快速采样

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