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

Simple diffusion:面向高分辨率图像的端到端扩散

Simple diffusion: End-to-end diffusion for high resolution images

Google Research(谷歌研究院)

Emiel Hoogeboom, Jonathan Heek, Tim Salimans

AI总结:

针对高分辨率图像像素空间扩散模型训练难、现有替代方案复杂度高的问题,本文通过调整噪声调度、局部缩放架构等四项简单技术,在ImageNet上实现了无采样修正扩散模型的SOTA图像生成效果。

摘要:

当前,在高分辨率图像的像素空间中应用扩散模型难度较大。现有方法转而聚焦于低维空间中的扩散(即潜扩散),或采用被称为级联模型的多层超分辨率生成方式。这类方法的缺点在于为扩散框架增添了额外的复杂度。\n本文旨在改进面向高分辨率图像的去噪扩散技术,同时尽可能保持模型简洁。论文围绕以下研究问题展开:如何在高分辨率图像上训练标准去噪扩散模型,且仍能获得与这些替代方法相当的性能?\n研究得出四项主要发现:1)需针对高分辨率图像调整噪声调度;2)仅缩放架构的特定部分即可满足需求;3)应在架构的特定位置添加dropout;4)下采样是避免生成高分辨率特征图的有效策略。结合这些简单却有效的技术,我们在ImageNet数据集上的图像生成任务中,取得了无采样修正器的扩散模型中的最优性能。

英文摘要:

Currently, applying diffusion models in pixel space of high resolution images is difficult. Instead, existing approaches focus on diffusion in lower dimensional spaces (latent diffusion), or have multiple super-resolution levels of generation referred to as cascades. The downside is that these approaches add additional complexity to the diffusion framework. This paper aims to improve denoising diffusion for high resolution images while keeping the model as simple as possible. The paper is centered around the research question: How can one train a standard denoising diffusion models on high resolution images, and still obtain performance comparable to these alternate approaches? The four main findings are: 1) the noise schedule should be adjusted for high resolution images, 2) It is sufficient to scale only a particular part of the architecture, 3) dropout should be added at specific locations in the architecture, and 4) downsampling is an effective strategy to avoid high resolution feature maps. Combining these simple yet effective techniques, we achieve state-of-the-art on image generation among diffusion models without sampling modifiers on ImageNet.

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

↑