基于潜在扩散模型的高分辨率图像合成

High-Resolution Image Synthesis with Latent Diffusion Models

罗宾·罗姆巴赫 Robin Rombach · LMU Munich · 2021-12-20 · arXiv:2112.10752 ↗ · 被引 25840

打开互动全文版(逐段中英对照 + 图/公式 + 论文问答)→

摘要 · Abstract

通过将图像形成过程分解为去噪自编码器的顺序应用,扩散模型在图像数据及其他领域取得了最先进的合成结果。此外,其公式允许一种引导机制来控制图像生成过程而无需重新训练。然而,由于这些模型通常直接在像素空间操作,强大扩散模型的优化往往消耗数百个 GPU 天,且由于顺序评估导致推理成本高昂。为了在有限计算资源下训练扩散模型,同时保持其质量和灵活性,我们将其应用于强大预训练自编码器的潜在空间。与以往工作相比,在这种表示上训练扩散模型首次能够在复杂度降低和细节保留之间达到接近最优的点,极大地提升了视觉保真度。通过将交叉注意力层引入模型架构,我们将扩散模型转变为强大且灵活的生成器,适用于文本或边界框等通用条件输入,并以卷积方式实现高分辨率合成。我们的潜在扩散模型在图像修复任务上达到了新的最先进水平,并在包括无条件图像生成、语义场景合成和超分辨率在内的各种任务上表现出极具竞争力的性能,同时相比基于像素的扩散模型显著降低了计算需求。代码可在 https://github.com/CompVis/latent-diffusion 获取。

By decomposing the image formation process into a sequential application of denoising autoencoders, diffusion models (DMs) achieve state-of-the-art synthesis results on image data and beyond. Additionally, their formulation allows for a guiding mechanism to control the image generation process without retraining. However, since these models typically operate directly in pixel space, optimization of powerful DMs often consumes hundreds of GPU days and inference is expensive due to sequential evaluations. To enable DM training on limited computational resources while retaining their quality and flexibility, we apply them in the latent space of powerful pretrained autoencoders. In contrast to previous work, training diffusion models on such a representation allows for the first time to reach a near-optimal point between complexity reduction and detail preservation, greatly boosting visual fidelity. By introducing cross-attention layers into the model architecture, we turn diffusion models into powerful and flexible generators for general conditioning inputs such as text or bounding boxes and high-resolution synthesis becomes possible in a convolutional manner. Our latent diffusion models (LDMs) achieve a new state of the art for image inpainting and highly competitive performance on various tasks, including unconditional image generation, semantic scene synthesis, and super-resolution, while significantly reducing computational requirements compared to pixel-based DMs. Code is available at https://github.com/CompVis/latent-diffusion .

核心贡献 · Key contributions

局限 · Limitations

论文章节 · Sections(共 15)

阅读逐段中英对照全文 →