像素空间扩散模型提速4.75倍
训练图像生成模型时,大家默认先在压缩的“潜空间”里学,再生成像素图。这篇发现直接在大像素空间里训练,收敛慢得多,于是提出先潜空间、后像素空间的两段式训练,最终在保持或超过原有质量的同时,推理速度快了3到4倍。它不是你明天就能用上的,但指明了未来模型提速的一个可行方向。
📄 原文摘要(英文)
This paper investigates an increasingly important topic in generative modeling: pixel-space diffusion models. Although numerous studies have explored this topic, most focus on small-scale or class-conditional settings. Consequently, a practical recipe for training pixel-space models that rival or exceed well-established latent-space counterparts remains elusive. Through a comprehensive empirical study, we first observe that direct large-scale pre-training in pixel space converges substantially more slowly than in latent space. This observation motivates a latent-to-pixel strategy that acquires generative priors efficiently in latent space and transitions to pixel space during post-training. We then systematically investigate the key design choices governing this transition, including weight initialization, data composition, prediction target, decoder architecture, and noise schedule, and identify a practical recipe that makes the resulting pixel-space models match or outperform their latent-space counterparts while delivering 3.18 to 4.75 times end-to-end inference speedups. We hope that our findings provide useful empirical insights and practical guidelines for future research on pixel-space generation.