1.x-Distill: Breaking the Diversity, Quality, and Efficiency Barrier in Distribution Matching Distillation
Organizations: Tsinghua University · Central Media Technology Institute, Huawei
Abstract
Diffusion models produce high-quality text-to-image results, but their iterative denoising is computationally expensive.Distribution Matching Distillation (DMD) emerges as a promising path to few-step distillation, but suffers from diversity collapse and fidelity degradation when reduced to two steps or fewer. We present 1.x-Distill, the first fractional-step distillation framework that breaks the integer-step constraint of prior few-step methods and establishes 1.x-step generation as a practical regime for distilled diffusion models.Specifically, we first analyze the overlooked role of teacher CFG in DMD and introduce a simple yet effective modification to suppress mode collapse. Then, to improve performance under extreme steps, we introduce Stagewise Focused Distillation, a two-stage strategy that learns coarse structure through diversity-preserving distribution matching and refines details with inference-consistent adversarial distillation. Furthermore, we design a lightweight compensation module for Distill--Cache co-Training, which naturally incorporates block-level caching into our distillation pipeline.Experiments on SD3-Medium and SD3.5-Large show that 1.x-Distill surpasses prior few-step methods, achieving better quality and diversity at 1.67 and 1.74 effective NFEs, respectively, with up to 33x speedup over original 28x2 NFE sampling.
Figures & tables
| Method | Step | #NFE | FID [ 10 ] | CLIP [ 9 ] | AS [ 39 ] | PS [ 15 ] | IR [ 48 ] | HPSv2 [ 47 ] | Img-free |
| Stable Diffusion 3 Medium | |||||||||
| Base Model | 28 | – | 0.3176 | 5.6348 | 22.5554 | 1.0429 | 30.7197 | ✗ | |
| Hyper-SD [ 33 ] | 4 | 15.5475 | 0.3127 | 4.9582 | 21.6407 | 0.7543 | 28.7578 | ✗ | |
| PCM [ 44 ] | 4 | 4 | 17.5605 | 0.3102 | 5.7743 | 22.0690 | 0.6715 | 29.0864 | ✗ |
| Flash [ 2 ] | 4 | 4 | 15.6443 | 0.3166 | 5.5485 | 22.3879 | 0.8938 | 29.4835 | ✗ |
| DMD2 ∗ [ 49 ] | 4 | 4 | 14.7125 | 0.3122 | 5.4632 | 22.4120 | 0.9981 | 31.0152 | ✗ |
| Stage I | Stage II | CLIP | AS | PS | IR | HPSv2 |
|---|---|---|---|---|---|---|
| ✓: Uniform sampling | ✗ | 0.3154 | 5.1432 | 22.4546 | 1.1081 | 31.4725 |
| ✓: a | ✗ | 0.3161 | 5.6044 | 22.4337 | 1.1170 | 31.8419 |
| ✓: b | ✗ | 0.3155 | 5.5442 | 22.3769 | 1.1020 | 31.8944 |
| ✓: c | ✗ | 0.3159 | 5.6210 | 22.4694 | 1.1226 | 32.0208 |
| ✓: d | ✗ | 0.3127 | 4.6196 | 22.4185 | 1.1419 | 30.2768 |
| ✗ | ✓ | 0.2144 | 3.3287 | 13.1621 | 0.1352 | 19.8334 |
Appendix figures & tables7 assets
Supplementary material from the paper’s appendix.
Appendix
| Config | SD3-Medium | SD3.5-Large |
|---|---|---|
| Resolution | ||
| Total Batch Size | 32 | |
| Precision | bfloat16 ( in float32 ) | |
| Stage I | ||
| Generator Learning Rate | 5e-6 | 2e-6 |
| Fake Learning Rate | 4e-5 | 2e-5 |
| Stage I | Stage II | #NFE | Latency(s) | CLIP | PS | IR | HPS | ||
|---|---|---|---|---|---|---|---|---|---|
| Cache | PixGAN | ||||||||
| ✓ | ✗ | ✓ | – | 2.00 | 0.7413 | 0.3184 | 22.6995 | 1.1601 | 33.0293 |
| ✓ | ✓ | ✗ | – | 1.75 | 0.6538 | 0.3170 | 22.2042 | 1.0152 | 30.9216 |
| ✓ | ✓ | ✓ | None | 1.75 | 0.6561 | 0.3205 | 22.3113 | 1.0500 | 31.0544 |
| ✓ | ✓ | ✓ | Simple MLP (seg.) | 1.75 | 0.6617 | 0.3208 | 22.5161 | 1.1312 | 32.2550 |
| ✓ | ✓ | ✓ | Simple MLP (per-block) | 1.75 | 0.6649 | 0.3196 | 22.4728 | 1.1184 | 32.0713 |
| Stage | Cached blocks | #NFE | Latency(s) | CLIP | PS | IR | HPS |
|---|---|---|---|---|---|---|---|
| SFD w/o cache | – | 2.00 | 0.7413 | 0.3184 | 22.6995 | 1.1601 | 33.0293 |
| DCT | 3–8 | 1.75 | 0.6617 | 0.3208 | 22.5161 | 1.1312 | 32.2550 |
| DCT | 10–15 | 1.75 | 0.6620 | 0.3196 | 22.1887 | 1.0218 | 30.9846 |
| DCT | 16–21 | 1.75 | 0.6621 | 0.3179 | 21.7315 | 0.8734 | 29.2148 |
| DCT | 3–6 + 10–13 | 1.75 | 0.6627 | 0.3201 | 22.3419 | 1.0726 | 31.5327 |
| DCT | 10–13 + 16–19 | 1.75 | 0.6628 | 0.3188 | 21.9642 | 0.9481 | 30.1029 |
Diffusion Model Distillation
Diffusion model distillation transfers a large model’s generative behavior to faster or smaller models that need fewer sampling steps.
See all 186 papers