arXiv:2501.13349cs.CV2025-01被引 3

通过多尺度分解降低扩散模型计算量,生成更高效。

MSF: Efficient Diffusion Model Via Multi-Scale Latent Factorize

  • 将图像特征分解为低频结构与高频细节,分两阶段生成
  • 在256×256和512×512下FID分别达2.08和2.47,优于DiT
  • 相同采样步数下速度提升4倍,适合高分辨率生成任务

尽管基于扩散的生成模型在视觉内容创作中取得显著进展,传统方法在处理高分辨率图像时仍面临计算挑战,因其需从噪声输入中逐步去噪整张图像。受傅里叶与小波分析等信号处理技术启发,本文提出一种基于多尺度潜在因子分解的扩散框架。该框架将预训练VAE的潜在特征分解为低频基信号(捕捉核心结构)和高频残差信号(贡献纹理等细粒度信息),并据此设计两阶段生成流程:先生成低分辨率基图,再生成高分辨率残差。由于残差信息易于建模,该阶段可显著减少采样步数。此信号分离思想类似小波分析的频段分离,使架构更简洁直观。在类别条件下的ImageNet基准上,本方法在256×256和512×512分辨率下分别取得2.08和2.47的FID分数,优于DiT基线(2.27和3.04),且在相同采样步数下实现4倍加速。

原文摘要 · Abstract (English)

While diffusion-based generative models have made significant strides in visual content creation, conventional approaches face computational challenges, especially for high-resolution images, as they denoise the entire image from noisy inputs. This contrasts with signal processing techniques, such as Fourier and wavelet analyses, which often employ hierarchical decompositions. Inspired by such principles, particularly the idea of signal separation, we introduce a diffusion framework leveraging multi-scale latent factorization. Our framework uniquely decomposes the denoising target, typically latent features from a pretrained Variational Autoencoder, into a low-frequency base signal capturing core structural information and a high-frequency residual signal that contributes finer, high-frequency details like textures. This decomposition into base and residual components directly informs our two-stage image generation process, which first produces the low-resolution base, followed by the generation of the high-resolution residual. Our proposed architecture facilitates reduced sampling steps during the residual learning stage, owing to the inherent ease of modeling residual information, which confers advantages over conventional full-resolution generation techniques. This specific approach of decomposing the signal into a base and a residual, conceptually akin to how wavelet analysis can separate different frequency bands, yields a more streamlined and intuitive design distinct from generic hierarchical models. Our method, \name\ (Multi-Scale Factorization), demonstrates its effectiveness by achieving FID scores of 2.08 ($256\times256$) and 2.47 ($512\times512$) on class-conditional ImageNet benchmarks, outperforming the DiT baseline (2.27 and 3.04 respectively) while also delivering a $4\times$ speed-up with the same number of sampling steps.

扩散模型高效生成多尺度分解

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。