万物皆可缩放:尺度不变扩散与连续超分辨率 Everything at Every Scale: Scale-Invariant Diffusion with Continuous Super-Resolution
单一框架统一无条件和连续超分辨率,无需条件分支或重新训练
前置知识
扩散模型(Diffusion Model)
生成模型,通过学习逆转一个固定的前向加噪过程来生成数据。前向过程逐渐将数据样本$x_0$转化为各向同性的高斯噪声,通过公式$x_t = \sqrt{\bar{\alpha}_t}x_0 + \sqrt{1-\bar{\alpha}_t}\epsilon$实现,其中$\epsilon \sim N(0, I)$。训练一个神经网络$\epsilon_\theta(x_t, t)$来预测噪声,通过最小化$E[\|\epsilon - \epsilon_\theta(x_t, t)\|^2]$。采样时从纯噪声开始,迭代去除噪声恢复数据分布
本文核心框架基于扩散模型,理解标准扩散是理解其尺度不变变体的基础
尺度不变性(Scale Invariance)
系统在坐标缩放$r \to br$下统计性质保持不变的性质,表现为统计可观测量服从幂律$O(k) \propto k^{-\alpha}$。临界物理系统(如Ising模型在临界温度)和自然图像都具有近似尺度不变性。自然图像的径向平均功率谱遵循$k^{-2}$衰减,这意味着不同频率成分具有相似的统计结构。重正化群理论形式化了粗粒化和重标度下的分布变换
本文的核心洞察:利用数据集的尺度不变性设计频域扩散过程,使生成和超分辨率成为同一轨迹的不同起点
频域(Frequency Space/DCT)
通过离散余弦变换(DCT)将图像从像素空间转换到频率空间表示。DCT将图像分解为不同频率的余弦基函数组合,系数代表各频率成分的强度。频域表示便于分析尺度结构,因为不同频率对应不同尺度的图像特征。本文中,$X_0(k)$表示图像的DCT系数,$S_0(k) = E[|X_0(k)|^2] - E[X_0(k)]^2$是经验方差谱,估计每模的方差
本文方法在频域操作,按频率顺序衰减信号,这是实现尺度不变扩散的关键
超分辨率(Super-Resolution)
从低分辨率输入重建高分辨率图像的逆问题。传统方法依赖条件扩散,需要低分辨率图像通过显式条件路径输入,通常还需要类标签或无分类器引导。条件模型需要针对不同缩放因子重新训练。本文提出无条件的超分辨率协议:低分辨率输入是模型自身前向过程的中间状态,同一反向过程恢复缺失的细尺度
本文将超分辨率统一到无条件和生成框架,是主要应用场景之一
研究动机
现有扩散模型和超分辨率方法存在任务分离的问题。图像生成从噪声创建图像,超分辨率从粗略输入重建细粒度细节,虽然实践中不同,但本质上都是逆转跨尺度的信息丢失。现有方法需要为每个任务设计特定架构、条件分支、无分类器引导,并为每个缩放因子重新训练。例如,条件超分辨率模型如SR3、ResShift需要显式的低分辨率条件输入,通常还需要类标签或引导技巧。这种任务特定方法增加了系统复杂度,限制了模型在不同缩放因子间的连续适应性。自然图像和临界物理系统都展现出尺度不变性,但现有扩散模型没有显式利用这一层级结构,而是平等地对待所有尺度
本文的目标是本文的目标是设计一个尺度不变的扩散框架,统一无条件和连续超分辨率任务,使其能在单一反向过程中完成。具体而言,目标是创建一个前向过程,按照从细到粗的顺序衰减图像内容,同时注入频谱匹配的高斯噪声,使尺度成为扩散动力学的显式坐标。训练后的单一反向过程仅通过改变起始时间步就能执行生成和连续超分辨率,无需任务特定架构、条件分支、无分类器引导或每个缩放因子的重新训练。评估目标包括在无条件CIFAR-10生成上达到与最先进方法竞争的性能,在ImageNet上实现2×到8×连续超分辨率,并在科学数据集上重现尺度不变统计量
与已有工作不同的是,本文的独特切入角度是:不是将扩散视为对所有尺度同时腐蚀,而是按顺序一个尺度一个尺度地擦除它们。这种尺度有序的粗粒化视角将扩散构建为尊重跨尺度自相似性的去噪过程,从构造上就是渐进的超分辨率。每个反向步骤添加更细尺度,从纯噪声运行完整反向过程会逐尺度生成图像。生成和超分辨率成为同一框架的不同表现形式:生成是超分辨率的极端情况(输入完全不包含信号),超分辨率是从中间状态初始化的同一反向过程,都只是反向粗粒化问题,区别仅在于反向过程从哪里开始。关键创新在于在频域设计前向过程,利用数据集的近似$k^{-2}$功率律谱作为噪声协方差,使每个中间状态都是原始图像的自相似粗粒化噪声版本
核心方法
SKILD方法整体思路是利用自然图像和临界物理系统的尺度不变性,在频域设计一个尺度不变的扩散前向过程。直觉上,图像在不同尺度上共享重复结构,放大后的自然图像仍然自然,有些自然物体本身就是自相似的,纹理、边缘和结构在不同尺度重复。统计上,这反映在自然图像功率谱上,它们在宽频率范围遵循近似幂律$k^{-2}$,这是近似尺度不变性的标志。物理中的重正化群理论为组织图像和纯噪声之间的变换提供了自然方式。技术路线上,SKILD在DCT频域定义前向过程:从高到低频率衰减信号,同时注入具有数据集频谱特性的高斯噪声而非白噪声,使模型学习移除统计上与其要生成的数据相似的噪声
核心创新点是在频域定义尺度不变的前向过程,与标准扩散有两个本质区别。第一,前向过程在衰减低频内容之前先衰减高频内容,这在频域通过频率依赖的衰减实现。第二,每步注入的噪声携带数据集本身的频谱$S_0(k)$而非白噪声,使模型学习移除统计上类似其学习生成的数据的噪声。这两点使每个中间状态都是原始图像的自相似粗粒化噪声版本。与条件超分辨率模型的本质区别在于:SKILD不需要任何额外条件,低分辨率输入是模型自身前向过程的中间状态,同一反向过程完成缺失的细尺度。与标准扩散的本质区别在于:频率依赖的衰减使尺度成为显式坐标,生成和超分辨率成为同一轨迹的不同起始点
方法步骤详情
方法步骤包括:首先,计算数据集的径向平均方差功率谱$S_0(k)$,用幂律$S_0(k) = C(k^2 + k_0^2)^{-a}$拟合,实验验证CIFAR-10、ImageNet-128和ImageNet-256都遵循$k^{-2}$衰减。其次,在频域定义连续前向边际$X(k, t) = e^{-k^2\lambda(t)/2} \odot X_0(k) + \sqrt{1-e^{-k^2\lambda(t)}} \odot \sqrt{S_0(k)} \epsilon_t$,其中$\odot$是Hadamard乘积,$\lambda(t)$是标量调度函数随$t$单调递增,$\epsilon_t \sim N(0, I)$。当$t$增长时,$e^{-k^2\lambda(t)/2}$在频域变窄,高频模在低频之前被衰减。噪声前因子使前向边际在每个$t$保持每模协方差$S_0(k)$,并在信号项消失时收敛到$N(0, S_0(k))$。在像素空间,相同过程对信号进行高斯核卷积并添加空间相关噪声,其相关长度随$t$增长。第三,使用DDPM离散化,设$\bar{\alpha}_n(k) = e^{-k^2\lambda_n}$,则$X_n(k) = \sqrt{\bar{\alpha}_n} \odot X_0(k) + \sqrt{1-\bar{\alpha}_n} \odot \sqrt{S_0(k)} \epsilon_n$。由于所有协方差在频域对角,反向后验是高斯的:$q(X_{n-1}|X_n, X_0) = N(\mu_q, S_0\tilde{\beta}_n)$,其中$\tilde{\beta}_n = \beta_n(1-\bar{\alpha}_{n-1})/(1-\bar{\alpha}_n)$,$\mu_q(n, X_n) = (X_n/\sqrt{\alpha}_n) - (\beta_n/\sqrt{1-\bar{\alpha}_n})\sqrt{S_0}\epsilon_n$。祖先采样通过$X_{n-1} = \mu_q(n, X_n) + \sqrt{S_0\tilde{\beta}_n} \odot \epsilon_n$进行。第四,训练使用$\epsilon$-预测网络,损失为$L = E_{n,X_0,\epsilon}[\|\epsilon - \epsilon_\theta(n, X_n)\|_2^2]$。实现细节:很小的$\alpha_n(k)$值可能导致高频模的反向更新过大,因此在祖先采样中将其下限设为$10^{-6}$;零模因为高斯信号滤波未被触及,因此引入低频截止$k_c$,对$\|k\| \leq k_c$的模使用$max(\|k\|, k_c)$作为调度。第五,评估两种调度:对数线性调度$\lambda(t) = t \cdot 10^{\lambda_i + (\lambda_f - \lambda_i)t}$在对数尺度上大致均匀移动截止频率;线性调度$\lambda(t) = \theta t/(\lambda_f(1-t) + \lambda_i)^2$在线性尺度上大致均匀移动截止频率。参数$\lambda_i$主要设置高频早期行为,$\lambda_f$、$k_c$和$\theta$主要设置低频晚期行为。最后,对于超分辨率,通过信噪比$SNR_n(k) = \bar{\alpha}_n(k)/(1-\bar{\alpha}_n(k))$选择反向起始时间步,阈值0.1对应在频域应用有效低通滤波,低于有效分辨率截止的模保留信号而高频模被衰减并主导噪声。从对应低分辨率输入的所选时间步的前向边际开始采样,然后逆转到$t_0$,将超分辨率转化为部分反向扩散问题
技术新颖性
技术新颖性体现在多个方面:首先,首次在频域设计频率依赖的衰减前向过程,使尺度成为扩散动力学的显式坐标。其次,噪声协方差匹配数据集频谱$S_0(k)$而非白噪声,这使模型学习移除统计上类似其学习生成的数据的噪声。第三,统一无条件和连续超分辨率到单一反向过程,仅通过改变起始时间步,无需条件分支、无分类器引导或每个缩放因子的重新训练。第四,提出有效分辨率协议,通过SNR阈值定义超分辨率的起始时间步,这与标准bicubic下-上采样在MSE和PSNR上相当。第五,在科学数据集上评估尺度不变性,使用连接四点关联函数这一统计物理可观测量,而非仅依赖视觉质量,为自相似性提供超越感知质量的评估标准
实验结果
核心发现在三个实验设置中验证。无条件CIFAR-10生成:SKILD达到FID 2.65和Inception Score 9.63,在频率/尺度感知模型中表现最佳,与最先进频率无关模型如EDM(FID 1.97)竞争。使用线性调度在约400K训练步达到最佳FID,但IS持续改善而FID在后期恶化,表明低频重构是以物体为中心数据集如CIFAR-10生成的瓶颈。ImageNet超分辨率:在ImageNet-256上4×超分辨率,SKILD在LPIPS(0.186)、CLIPIQA(0.612)和MUSIQ(59.226)上超越所有比较方法,在SSIM(0.683)上达到第二佳。PSNR(24.10)更偏重原始像素精度,而强调人类感知的指标偏向SKILD方法。关键发现是单一训练的ImageNet模型通过仅改变起始时间步支持2×到8×连续超分辨率因子。临界Ising模型科学基准:从32×32有效分辨率起始状态超分辨率到128×128临界Ising场,SKILD重建的连接四点关联函数在每个patch大小都紧密跟踪真实情况,而SR3强扩散基线显著偏离真实情况。这证明SKILD可以重建缺失的细尺度同时保留定义尺度不变律的可观测量,严格的统计测试而非仅视觉相似性。消融研究验证SKILD在图像生成上对广泛调度范围稳健,大多数调度在400K训练步内达到FID低于或接近5,所有调度达到IS接近或高于9。不同调度间收敛速度有差异,但方法不依赖单一脆弱调度。与常见像素空间扩散调度相比,FID和IS收敛轨迹显示尺度不变扩散的独特动态
查看结构化数据
| 任务 | 指标 | 本文 | 基线 | 提升 |
|---|---|---|---|---|
| CIFAR-10无条件生成 | FID | 2.65 | DDPM 3.17, DDPM++ 2.78, NCSN++ 2.20, EDM 1.97, WaveDiff 4.01, IHDM 5.02 | 在频率/尺度感知模型中最佳,接近最先进频率无关模型 |
| CIFAR-10无条件生成 | Inception Score | 9.63 | DDPM++ 9.46, NCSN++ 9.64, EDM 9.89, IHDM 7.70 | 在频率/尺度感知模型中最佳 |
| ImageNet-256 4×超分辨率 | LPIPS↓ | 0.186 | SR3 0.231, SinSR 0.221, I2SB 0.206, ResShift 0.231 | 最佳,超越强条件扩散基线 |
| ImageNet-256 4×超分辨率 | CLIPIQA↑ | 0.612 | ResShift 0.592, SR3 0.611, LDM-30 0.572 | 最佳 |
| ImageNet-256 4×超分辨率 | MUSIQ↑ | 59.226 | BSRGAN 54.697, SwinIR 53.790, ResShift 53.660, I2SB 53.936 | 最佳 |
| ImageNet-256 4×超分辨率 | SSIM↑ | 0.683 | I2SB 0.730, ResShift 0.677, SinSR 0.657 | 第二佳,仅次于I2SB |
| ImageNet-256 4×超分辨率 | PSNR↑ | 24.10 | I2SB 26.76, DDRM 25.56, ResShift 25.01 | 中等,PSNR偏重像素精度 |
| 临界Ising模型超分辨率 | 连接四点关联函数 | 紧密跟踪真实情况 | SR3显著偏离真实情况 | 精确重建多尺度物理结构 |
局限与改进
作者承认的局限性包括:首先,采样需要1000个祖先步骤,这比一些先进的采样器慢,快速采样器或针对模依赖调度的定制求解器是自然的下一步。其次,无条件生成对低频结构生成敏感,低频生成的错误在反向链早期传播并约束后期高频生成,这反映在FID在早期达到最佳而IS持续改善的trade-off上。第三,超分辨率协议使用精确前向边缘分布作为低分辨率输入,扩展到现实世界降质如未知相机和压缩管道是重要方向。第四,Ising实验在一个临界系统上建立尺度不变扩散,扩展到额外物理系统和高阶可观测量将拓宽科学基准。观察到的局限性包括:调度参数敏感,虽然方法对广泛调度范围稳健,但最佳性能需要仔细调优$\lambda_i$、$\lambda_f$、$k_c$和$\theta$四个参数。计算开销:频域操作增加前向和反向过程的计算成本,虽然具体成本未在论文中量化。模型使用现成神经网络(来自NCSN++家族),新的架构设计针对本文模型可能显著提高性能。虽然理论上支持连续缩放因子,实际实现受限于离散时间步调度,选择特定缩放因子需要找到对应的起始时间步。在ImageNet超分辨率上,PSNR指标不占优势,这可能限制在需要精确像素精度的应用中使用
独立分析的弱点
独立分析的弱点包括:第一,无条件生成的低频瓶颈。作者观察到FID在训练早期达到最佳而IS持续改善,这表明低频重构是以物体为中心数据集的瓶颈。改进方向可以是改进架构以更好建模低频结构,或者引入显式的低频约束或先验。第二,采样效率。1000步祖先采样比一些先进采样器慢,改进方向包括开发针对模依赖调度的快速采样器,如ODE求解器、渐进蒸馏或一致性模型。第三,现实世界降质。当前超分辨率协议限于精确前向边缘分布,不适合真实世界的未知相机模糊、压缩伪影或噪声。改进方向包括学习真实世界降质的模型,或将本文尺度不变框架与逆问题方法结合。第四,调度参数敏感性。虽然方法对广泛调度范围稳健,但最佳性能需要调优四个参数,改进方向可以是自适应调度学习或自动化超参数搜索。第五,架构限制。模型使用现成NCSN++架构,可能不是频域操作最优,改进方向包括设计专门针对频域扩散的架构,如更有效地处理频率依赖的结构。第六,评估局限性。论文主要在合成数据集和理想条件评估,缺乏在真实世界图像上的广泛评估,改进方向包括在更多数据集和真实场景上评估。第七,理论理解。方法基于尺度不变性的类比,但缺乏严格的理论保证,改进方向包括提供尺度不变扩散的理论分析,如收敛保证或误差界
未来方向
作者提出的未来工作包括:开发快速采样器或针对模依赖调度的定制求解器;改进无条件生成的低频结构生成,稳定全局结构而不牺牲细粒度细节;扩展到现实世界降质如未知相机和压缩管道;将Ising实验扩展到额外物理系统和高阶可观测量;设计针对本文模型的新架构以显著提高性能。基于成果可延伸的未来工作包括:将尺度不变扩散扩展到其他模态如视频、3D数据或音频,这些模态也展现尺度不变性;探索尺度不变扩散与其他生成框架的结合,如GAN或VAE;研究尺度不变扩散在科学计算中的应用,如多尺度物理模拟或反问题;开发自适应调度学习,自动从数据学习最优$\lambda(t)$;探索尺度不变扩散在分布外泛化或域适应中的应用,因为尺度不变性可能跨域共享;研究尺度不变扩散的连续极限和理论性质,如与重正化群流的形式化连接;开发评估尺度不变性的新指标,超越连接四点关联函数,可能包括高阶统计量或分形维度;探索尺度不变扩散在条件生成中的应用,如文本到图像或类别到图像,可能通过在特定尺度注入条件信息;研究尺度不变扩散的可解释性和可控性,如理解不同频率成分对生成的贡献或选择性编辑特定尺度
复现评估
复现评估方面,论文承诺代码和数据在GitHub公开:https://github.com/JazzyCH/SKILD。论文在附录中提供详细的超参数和计算细节,包括架构配置、调度参数、训练设置和评估协议。使用标准PyTorch框架和现成NCSN++架构,降低了复现门槛。数据集都是公开的:CIFAR-10、ImageNet、以及用Wolff集群算法自生成的临界Ising配置。论文提供具体实验设置细节,如使用50K生成样本评估CIFAR-10,使用ImageNet验证集随机3K图像子集评估超分辨率,使用1K样本评估Ising超分辨率。然而,论文未报告具体计算资源需求,如GPU小时数、硬件规格或训练时间,这增加了精确复现的难度。超分辨率协议使用精确前向边缘分布作为低分辨率输入,这可能需要额外的存储或计算来保存这些状态。论文在附录中提供额外实验细节和消融研究,增加了透明度。代码开源状态需要实际验证,但论文承诺的开放性总体支持复现。复现难度中等:方法概念清晰,实现细节充分,但调优调度参数可能需要经验,以及计算资源需求可能较高
论文图表
Figure 2展示自然图像数据集的方差功率谱,包含两个子图。(a)显示CIFAR-10(32)、ImageNet-128和ImageNet-256的光谱,它们在共享频率范围内展现相似的幂律衰减,表明近似尺度不变性;(b)显示ImageNet-256每个颜色通道(RGB)独立计算的方差,幂律拟合恢复$k^{-2}$频率衰减。这张图用对数-对数坐标展示功率谱与径向频率的关系,验证了自然图像遵循近似$k^{-2}$幂律的假设
这张图对理解论文很重要,因为它提供了SKILD方法设计的关键依据:自然图像的功率谱遵循近似幂律$k^{-2}$衰减,这表明自然图像具有近似尺度不变性。SKILD使用这一数据集频谱作为噪声协方差,这是本文方法与标准扩散使用白噪声的关键区别
Figure 4包含两个子图。(a)显示线性调度$\bar{\alpha}$随时间步$t$的变化,以及相应的FID和IS曲线,展示最佳FID和IS的调度。曲线显示低频模比高频模更晚被衰减,频率$\bar{\alpha}$表示频率分量的保留程度。(b)显示训练过程中的模式崩溃,FID在约200K训练步达到最佳3.2后恶化到3.8,而IS持续改善从9.4到10.2。这表明低频重构是以物体为中心数据集的瓶颈
这张图对理解论文很重要,因为它展示了SKILD的关键训练动态:线性调度的具体形式、不同频率的衰减顺序,以及训练过程中FID和IS的trade-off。这解释了为什么低频重构成为瓶颈以及为什么方法对以物体为中心的数据集敏感