arXiv:2604.10983cs.CV2026-04中稿 · ICLR被引 1

用能量优化的扩散桥实现高效图像修复,单步即可高质量恢复。

Energy-oriented Diffusion Bridge for Image Restoration with Foundational Diffusion Models

论文配图:Energy-oriented Diffusion Bridge for Image Restoration with Foundational Diffusion Models
图 1 · 摘自论文原文
  • 设计短时程能量优化路径,逆过程从噪声混合态启动
  • 单步映射实现快速生成,支持不同退化任务自适应调节
  • 在去噪与超分辨率上均达顶尖效果,采样效率显著提升

扩散桥模型通过显式连接清晰与退化图像分布,在图像修复中表现优异。然而,其常依赖复杂高成本的轨迹,限制了采样效率与修复质量。为此,本文提出能量导向的扩散桥(E-Bridge)框架,通过设计一组低成本流形测地线轨迹来提升性能。方法上,采用更短时间跨度的桥接过程,并使逆过程从熵正则化点(退化图像与高斯噪声混合)出发,理论上降低所需轨迹能量。为高效求解,借鉴一致性模型思想,学习单步映射函数,通过专为该轨迹设计的连续时间一致性目标进行优化,可解析地将任意轨迹状态映射至目标图像。值得注意的是,轨迹长度成为可调的任务自适应参数,使模型能根据不同退化程度(如去噪与超分辨率)灵活权衡信息保留与生成能力。大量实验表明,E-Bridge 在多种图像修复任务中均达到当前最优性能,且仅需单步或更少采样步骤即可实现高质量恢复。

原文摘要 · Abstract (English)

Diffusion bridge models have shown great promise in image restoration by explicitly connecting clean and degraded image distributions. However, they often rely on complex and high-cost trajectories, which limit both sampling efficiency and final restoration quality. To address this, we propose an Energy-oriented diffusion Bridge (E-Bridge) framework to approximate a set of low-cost manifold geodesic trajectories to boost the performance of the proposed method. We achieve this by designing a novel bridge process that evolves over a shorter time horizon and makes the reverse process start from an entropy-regularized point that mixes the degraded image and Gaussian noise, which theoretically reduces the required trajectory energy. To solve this process efficiently, we draw inspiration from consistency models to learn a single-step mapping function, optimized via a continuous-time consistency objective tailored for our trajectory, so as to analytically map any state on the trajectory to the target image. Notably, the trajectory length in our framework becomes a tunable task-adaptive knob, allowing the model to adaptively balance information preservation against generative power for tasks of varying degradation, such as denoising versus super-resolution. Extensive experiments demonstrate that our E-Bridge achieves state-of-the-art performance across various image restoration tasks while enabling high-quality recovery with a single or fewer sampling steps. Our project page is https://jinnh.github.io/E-Bridge/.

图像修复扩散模型一致性模型高效生成

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