arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2412.05488cs.CVcs.LGeess.IV

使用噪声水平校正增强扩散模型的样本生成

Enhancing Sample Generation of Diffusion Models using Noise Level Correction

Carnegie Mellon University(卡内基梅隆大学);Toyota Research Institute(丰田研究院)

Abulikemu Abuduweili, Chenyang Yuan, Changliu Liu, Frank Permenter

AI总结:

本文提出一种噪声水平校正网络,通过将估计噪声水平与含噪样本到数据流形的真实距离对齐来增强扩散模型的样本生成,该方法兼容现有去噪调度器并可扩展至多种图像恢复任务,显著提升了生成质量。

摘要:

扩散模型的去噪过程可解释为将含噪样本近似投影到数据流形上。此外,这些样本中的噪声水平近似于其到潜在流形的距离。基于此见解,我们提出了一种新方法,通过将估计的噪声水平与含噪样本到流形的真实距离对齐来增强样本生成。具体而言,我们引入了一个噪声水平校正网络,利用预训练的去噪网络,在去噪过程中改进噪声水平估计。此外,我们通过整合特定任务的约束条件,包括图像修复、去模糊、超分辨率、上色和压缩感知,将该方法扩展到各种图像恢复任务中。实验结果表明,我们的方法在无约束和有约束的生成场景中均显著提高了样本质量。值得注意的是,所提出的噪声水平校正框架与现有的去噪调度器(例如,DDIM)兼容,提供了额外的性能提升。

英文摘要:

The denoising process of diffusion models can be interpreted as an approximate projection of noisy samples onto the data manifold. Moreover, the noise level in these samples approximates their distance to the underlying manifold. Building on this insight, we propose a novel method to enhance sample generation by aligning the estimated noise level with the true distance of noisy samples to the manifold. Specifically, we introduce a noise level correction network, leveraging a pre-trained denoising network, to refine noise level estimates during the denoising process. Additionally, we extend this approach to various image restoration tasks by integrating task-specific constraints, including inpainting, deblurring, super-resolution, colorization, and compressed sensing. Experimental results demonstrate that our method significantly improves sample quality in both unconstrained and constrained generation scenarios. Notably, the proposed noise level correction framework is compatible with existing denoising schedulers (e.g., DDIM), offering additional performance improvements.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑