无需训练，这个新方法实现了生成图像尺寸、分辨率自由
无需训练，这个新方法实现了生成图像尺寸、分辨率自由
近日，来自香港中文大学-商汤科技联合实验室等机构的研究者们提出了FouriScale，该方法在利用预训练扩散模型生成高分辨率图像方面取得了显著提升。
近期，扩散模型凭借其出色的性能已超越GAN和自回归模型，成为生成式模型的主流选择。基于扩散模型的文本到图像生成模型（如SD、SDXL、Midjourney和Imagen）展现了生成高质量图像的惊人能力。通常，这些模型在特定分辨率下进行训练，以确保在现有硬件上实现高效处理和稳定的模型训练。
图 1：采用不同方法在SDXL 1.0下生成2048×2048图像的对比。
然而，当这些预训练的扩散模型在超出训练分辨率时生成图像，通常会出现模式重复和严重的人工伪影（artifacts）问题，如图1最左侧所示。
为了解决这一问题，来自香港中文大学-商汤科技联合实验室等机构的研究者们在一篇论文中深入研究了扩散模型中常用的UNet结构的卷积层，并从频域分析的角度提出了FouriScale，如图2所示。
图 2：FouriScale的流程（橙色线）示意图，目的是保证跨分辨率的一致性。
FouriScale通过引入空洞卷积操作和低通滤波操作来替换预训练扩散模型中的原始卷积层，旨在实现不同分辨率下的结构和尺度一致性。配合“填充然后裁剪”策略，该方法能够灵活生成不同尺寸和长宽比的图像。此外，借助FouriScale作为指导，该方法在生成任意尺寸的高分辨率图像时，能够保证完整的图像结构和卓越的图像质量。FouriScale无需任何离线预计算，具有良好的兼容性和可扩展性。
定量和定性实验结果表明，FouriScale在利用预训练扩散模型生成高分辨率图像方面取得了显著提升。
方法介绍
1. 空洞卷积保证跨分辨率下的结构一致性
扩散模型的去噪网络通常是在特定分辨率的图像或潜在空间上训练的，这个网络通常采用U-Net结构。作者的目标是在推理阶段使用去噪网络的参数生成分辨率更高的图像，而无需重新训练。为了避免推理分辨率下的结构失真，作者尝试在默认分辨率和高分辨率之间建立结构一致性。对于U-Net中的卷积层，结构一致性可表述为：
其中k是原本的卷积核，k'是为更大分辨率定制的新卷积核。根据空间下采样的频域表示，如下：
可以将公式（3）写为：
这个公式表明了理想卷积核k'的傅里叶频谱应该是由s×s个卷积核k的傅里叶频谱拼接而成的。换句话说，k'的傅里叶频谱应该有周期性重复，这个重复模式是k的傅里叶频谱。
广泛使用的空洞卷积正好满足这个要求。空洞卷积的频域周期性可以通过下式表示：
当利用预训练扩散模型（训练分辨率为（h,w））生成(H,W)的高分辨率图像时，空洞卷积的参数使用原始卷积核，扩张因子为(H/h, W/w)，是理想的卷积核k'。
2. 低通滤波保证跨分辨率下的尺度一致性
然而，只利用空洞卷积无法完美地解决问题，如图3左上角所示，只使用空洞卷积仍然在细节上存在模式重复的现象。作者认为这是因为空间下采样的频率混叠现象改变了频域分量，导致了不同分辨率下频域分布的差异。为了保证跨分辨率下的尺度一致性，他们引入了低通滤波来过滤掉高频分量，以去除空间下采样后的频率混叠问题。从图3右侧对比曲线可以看到，在使用低通滤波后，高低分辨率下的频率分布更加接近，从而保证了尺度一致。从图3左下角图看到，在使用低通滤波后，细节的模式重复现象有明显地改善。
图 3：(a)是否采用低通滤波的视觉对比。（b）不采用低通滤波的傅立叶相对对数幅值曲线。(c)采用低通滤波的傅立叶相对对数幅值曲线。
3. 适应于任意尺寸的图像生成
以上的方式只能适应于生成分辨率与默认推理分辨率的长宽比一致时，为了使FouriScale适应于任意尺寸的图像生成，作者采用了一种“填充然后裁剪”的方式，方法1中展示了结合了该策略的FouriScale的伪代码。
4. FouriScale引导
由于FouriScale中的频域操作，不可避免地使生成的图像出现了细节缺失与不期望的伪影问题。为了解决这一问题，如图4，作者提出了将FouriScale作为引导的方式。具体来说，在原本的条件生成估计以及无条件生成估计的基础上，他们引入一个额外的条件生成估计。这个额外的条件生成估计的生成过程同样采用空洞卷积，但是使用更加温和的低通滤波，从而保证细节不丢失。同时他们将利用FouriScale输出的条件生成估计中的注意力分数替换掉这一额外的条件生成估计中的注意力分数，由于注意力分数包含着生成图像中的结构信息，这一操作将FouriScale中正确的图像结构信息引入，同时保证了图像质量。
图 4：(a) FouriScale引导示意图。（b）不采用FouriScale作为引导的生成图像，有明显的伪影和细节错误。(c)采用FouriScale作为引导的生成图像。
实验
1. 定量试验结果
作者遵循[1]的方法，测试了三个文生图模型（包括SD 1.5，SD 2.1和SDXL 1.0），生成四种更高分辨率的图像。测试的分辨率是它们各自训练分辨率的4倍、6.25倍、8倍和16倍像素数量。在Laion-5B上随机采样30000/10000个图文对测试的结果如表1所示：
表 1：不同的无需训练方法的定量结果对比
他们的方法在各个预训练模型，不同分辨率下都获得了最优的结果。
2. 定性试验结果
如图5所示，他们的方法在各个预训练模型，不同分辨率下都能够保证图像生成质量与一致的结构。
图 5：不同的无需训练方法的生成图像对比
结论
本文提出了FouriScale用于增强预训练扩散模型生成高分辨率图像的能力。FouriScale从频域分析出来，通过空洞卷积和低通滤波操作改善了不同分辨率下的结构和尺度一致性，解决了重复模式和结构失真等关键挑战。采用“填充然后裁剪”策略并利用FouriScale作为指导，增强了文本到图像生成的灵活性和生成质量，同时适应了不同的长宽比生成。定量和定性的实验对比表明，FouriScale能够在不同预训练模型，不同分辨率下都能够保证更高的图像生成质量。
论文地址：https://arxiv.org/abs/2403.12963
开源代码：https://github.com/LeonHLJ/FouriScale
论文标题：FouriScale: A Frequency Perspective on Training-Free High-Resolution Image Synthesis