arXiv:2507.18534cs.CVcs.LG2025-07中稿 · CVPR

提出可处理任意噪声的扩散模型EDA,提升图像修复泛化能力

Elucidating the Design Space of Arbitrary-Noise-Based Diffusion Models

  • 构建统一框架EDA,支持多种噪声模式且不增加计算开销
  • 仅5步采样即达专业方法水平,跨医学与自然图像任务表现优
  • 适合图像修复领域研究者,尤其关注噪声建模与泛化性能

尽管EDM旨在统一扩散模型设计空间,但其依赖固定高斯噪声,难以解释新兴的基于流的方法(这些方法可处理任意噪声)。我们的研究表明,EDM强制注入高斯噪声会损害图像修复任务,因它污染退化图像、延长修复距离并增加任务复杂度。为在统一理论框架下解释不同噪声模式的处理方法,并最小化修复距离,我们提出EDA(Elucidating the Design space of Arbitrary-noise diffusion models)。理论上,EDA在保持EDM模块化的同时扩展了噪声模式灵活性,严格证明:噪声复杂度提升不会带来额外计算开销。EDA在三个代表性任务中验证:磁共振成像偏置场校正(全局平滑噪声)、CT金属伪影去除(全局尖锐噪声)和自然图像阴影去除(局部边界感知噪声)。仅用5次采样,即在医学与自然图像任务上超越专用方法,展现强大泛化能力。代码已开源:https://github.com/PerceptionComputingLab/EDA。

原文摘要 · Abstract (English)

Although EDM aims to unify the design space of diffusion models, its reliance on fixed Gaussian noise prevents it from explaining emerging flow-based methods that diffuse arbitrary noise. Moreover, our study reveals that EDM's forcible injection of Gaussian noise has adverse effects on image restoration task, as it corrupts the degraded images, overextends the restoration distance, and increases the task's complexity. To interpret diverse methods for handling distinct noise patterns within a unified theoretical framework and to minimize the restoration distance, we propose EDA, which Elucidates the Design space of Arbitrary-noise diffusion models. Theoretically, EDA expands noise pattern flexibility while preserving EDM's modularity, with rigorous proof that increased noise complexity introduces no additional computational overhead during restoration. EDA is validated on three representative medical image denoising and natural image restoration tasks: MRI bias field correction (global smooth noise), CT metal artifact removal (global sharp noise) and natural image shadow removal (local boundary-aware noise). With only 5 sampling steps, competitive results against specialized methods across medical and natural tasks demonstrate EDA's strong generalization capability for image restoration. Code is available at: https://github.com/PerceptionComputingLab/EDA.

扩散模型图像修复噪声建模通用性

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