像素空间扩散模型提速3倍,训练策略大揭秘
训练图像生成模型,通常先在“潜空间”里学,再转到“像素空间”细化,但直接在大规模像素空间训练会慢很多。这篇研究提出先高效地在潜空间获取生成能力,再在后期训练中切换到像素空间,并系统探索了权重初始化、数据组成、预测目标、解码器架构和噪声调度等关键设计,最终让像素空间模型在匹配或超越潜空间模型的同时,推理速度提升3.18到4.75倍。
📄 原文摘要(英文)
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.