An Empirical Study of Training Pixel-Space Text-to-Image Diffusion Models
Organizations: The Hong Kong University of Science and Technology · Alibaba Token Hub, Alibaba Group · Nanjing University · University of California, San Diego
Abstract
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.
Figures & tables
| Model | NFE | Latency (s) | GenEval | DPG | OneIG | LongText |
|---|---|---|---|---|---|---|
| Z-Image (latent-space) | 100 | 20.12 | 0.7510 | 86.91 | 0.566 | 0.9332 |
| L2P (pixel-space) | 100 | 18.26 | 0.7612 | 86.00 | 0.499 | 0.7798 |
| Ours (pixel-space) | 100 | 4.56 (4.41 ) | 0.7644 | 87.60 | 0.564 | 0.9252 |
| Z-Image-Turbo (latent-space) | 4 | 0.95 | 0.7625 | 85.31 | 0.520 | 0.8639 |
| Ours-Turbo (pixel-space) | 4 | 0.20 (4.75 ) | 0.7698 | 86.85 | 0.512 | 0.8732 |
| FLUX2-klein-Base (latent-space) | 100 | 20.25 | 0.8027 | 84.89 | 0.541 | 0.5456 |
Appendix figures & tables2 assets
Supplementary material from the paper’s appendix.
Appendix
| Section 4.1 | Section 4.2 | Section 4.3 | |
|---|---|---|---|
| Weight initialization | From scratch / Latent-pretrained | Latent-pretrained | Latent-pretrained |
| Training data | Self-generated + Real | Real images / Self-generated samples / Other-model samples / Self-generated + Real | Self-generated + Real |
| Prediction target | -prediction | -prediction | -prediction / -prediction |
| Decoder architecture | DiP | DiP | DiP |
| Noise scale | |||
| Patch size | ps16 | ps16 | ps16 |
| Section 4.4 | Section 4.5 | Section 5.1 | |
|---|---|---|---|
| Weight initialization | Latent-pretrained | Latent-pretrained | Latent-pretrained |
| Training data | Self-generated + Real | Self-generated + Real | Self-generated + Real |
| Prediction target | -prediction | -prediction | -prediction |
| Decoder architecture | JiT / DiP / PiT / Deco / VAE | DiP | DiP |
| Noise scale | |||
| Patch size | ps16 | ps16 | ps16 / ps32 / ps32-adapt16 / ps64-adapt32 |
Pixel-Space Diffusion Models
Pixel-space diffusion models generate or restore images by learning to denoise raw pixels rather than compressed latent codes.
See all 56 papers