Organizations: Deakin Applied Artificial Intelligence Initiative, Deakin University, Australia · VinSmart Future, Vietnam · School of Information Technology, Deakin University, Australia
One-step diffusion distillation achieves strong performance on small datasets but remains insufficient for high-resolution or complex text-to-image settings. Extending to multistep distillation often requires training data, suffers from early-step degradation, inconsistent outputs across sampling budgets, and incurs high computational cost. We identify that these limitations stem from simultaneous training, where a single adapted network approximates multiple score functions of different student distributions at once, causing biased distribution matching and unstable optimization. Therefore, we propose Progressive Multistep Data-free Distillation (PMDD), which assigns a dedicated adapted network per student distribution while avoiding the memory overhead of maintaining multiple adapted networks. Combined with a stage-wise unforget loss to preserve earlier-step generations, PMDD reduces optimization bias in the student, provides faster convergence, and a single unified model that supports multiple inference budgets. Extensive experiments show that PMDD matches or outperforms data-free few-step baselines at every multistep budget, reaches FID 1.99 on ImageNet 64x64 in two steps, below its 512-NFE teacher (2.32), and achieves FID 7.76 on zero-shot COCO 512x512, while requiring 4-5x less training compute than data-free DMDv2 to support three inference budgets.
Figures & tables
Figure 1: Generations of PMDD and data-free DMDv2 across different inference budgets (NFE: neural function evaluations). DMDv2 shows strong inconsistency across inference budgets and between previously distilled 1-step and 2-step models. In contrast, PMDD maintains consistent structure and high-quality outputs across all sampling budgets using a single model.
Figure 2: Overview of our progressive distillation framework at stage 3. For clarity, the unforgetting loss and adapted network xφ are omitted from the figure.
Method
NFE (↓)
FID (↓)
CLIP (↑)
Teacher Model
SD v1.5 (CFG=3) [ 35 ]
50
8.78
-
SD v1.5 (CFG=8) [ 35 ]
50
13.45
0.322
One-step Distill.
DMD (CFG=3) [ 51 ]
1
11.49
-
InstaFlow-1.7B [ 23 ]
1
11.8
0.309
Table 1: (a) : Text-to-image generation performance on 30K COCO 2014 samples for Stable Diffusion v1.5 distillation. Data-free (DF) methods are marked in brown with the best and second-best approaches highlighted in bold and underline , respectively. (b) : Image generation results on ImageNet64; the teacher uses 512 NFE, as reported in [ 51 ] .
Figure 3: HPSv2 win rate: PMDD vs. DF-DMDv2.
Methods
Step
LPIPS ( ↓ )
SSIM ( ↑ )
DMDv2
1 vs 2
0.510
0.303
1 vs 3
0.534
0.309
2 vs 3
0.476
0.411
PMDD
1 vs 2
0.352
0.651
1 vs 3
0.387
0.613
2 vs 3
0.360
0.637
Table 2: Quantitative evaluation of cross-step structural consistency.
Figure 6
Step
DMDv2 (Data / Time)
PMDD (Data / Time)
Efficiency (Data / Time)
1-step
82M / 110 hrs
56M / 65 hrs
1.5 × / 1.7 ×
2-step
82M / 120 hrs
0.32M / 12 hrs
256 × / 10 ×
3-step
82M / 145 hrs
1.3M / 8hrs
63× / 18 ×
Total
246M / 375 hrs
57.6M / 85hrs
4.3× / 4.4 ×
Table 4: Training cost on SDv1.5 (million synthetic samples / wall-clock hours). DMDv2 trains one model per budget; PMDD’s 2- and 3-step rows are incremental stage costs.
Methods
Steps
FID ( ↓ )
CLIP ( ↑ )
GenEval (↑)
PixArt α [ 4 ]
20
8.85
28.07
0.51
SD3-medium [ 7 ]
28
11.92
27.83
0.62
FLUX-dev [ 17 ]
50
10.15
27.47
0.67
DF-DMDv2 [ 50 ]
1
10.94
27.62
0.45
2
12.08
27.68
0.44
3
10.83
27.72
0.49
Table 5: Data-free distillation of PixArt- α on 1024×1024 MJHQ-30k. Baselines are reproduced with official code and the same teacher; SD3-medium and FLUX-dev are reference generators. Best/second-best distilled results in bold / underline .
Figure 5: Effect of α on 1-step sampling (NFE = 1) for a 3-step PMDD model. αm† retains all previous checkpoints, whereas α retains only the latest.
Figure 6: Effect of unforget weight α on 3-step inference for COCO 2014. αm† preserves all previous checkpoints, while α preserves only the latest.
Figure 7: Ablation Study of using the previously distilled model goldi−1 for sampling xti on ImageNet64.
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 8: FID on ImageNet 64×64 during the second (two-step) training stage, sampled with NFE =2 and NFE =1 . Iterations count from the start of stage 2, which is initialized from the one-step student. Horizontal lines mark the 2-step baselines: Progressive Distillation (FID 8.95), Consistency Distillation (FID 4.70) and Moment Matching (FID 3.86); dots mark the first iteration at which PMDD (NFE =2 ) falls below each.
Type
Method
NFE (↓)
FID (↓)
Teacher
VP-EDM [ 13 ]
35
1.79
One-step
GET-Base [ 51 ]
1
6.25
Meng et al. [ 29 ]
1
5.98
2-Rectified Flow † [ 22 ]
1
4.85
DMD (w/o KL) [ 51 ]
1
3.82
DMD [ 51 ]
1
2.66
Appendix
Table 6: Results on CIFAR-10. PMDD uses LSiDstu (Appendix D.3). Data-free (DF) distillation methods are shown in brown color and unconditional methods are indicated with † . The best and second best DF methods are highlighted in bold and underscored , respectively.
Method
Precision (↑)
Recall (↑)
2-step
DMDv2
0.59
0.46
PMDD
0.62
0.53
3-step
DMDv2
0.62
0.46
PMDD
0.64
0.52
Appendix
Table 7: Precision and recall of data-free DMDv2 and PMDD on SD v1.5.
NFE
LDMstu
LSiDstu
1
3.49
2.52
2
2.91
2.19
Appendix
Table 8: Effect of the student loss on CIFAR-10 (FID).
FID (NFE =2 )
FID (NFE =1 )
α=0.1
2.19
7.84
α=1.0
2.21
6.78
α=3.0
2.21
4.91
Appendix
Table 9: Effect of unforget weight α on 2-step PMDD (CIFAR-10).
Figure 9: Ablation study of using goldi−1 for sampling xti (2-step, LDMstu ). Left: FID on CIFAR-10 and ImageNet64. Middle/right: adapted-network and student losses on ImageNet64.
Method
Adapted network
CIFAR-10
ImageNet64
DF-DMDv2
shared
3.27
2.49
DF-DMDv2
stage-conditioned
3.22
2.44
PMDD
per stage
2.19
1.99
Appendix
Table 10: Stage conditioning of the adapted network (2-step, FID). DF-DMDv2 rows use LDMstu ; PMDD results are from Table 1 (b) and Table 6 .
Figure 10: Additional qualitative comparison of 3-step PMDD outputs (NFE = 3, CFG = 1.0) with previously generated 1-step and 2-step reference images.
Figure 11: Qualitative comparison of 3-step PMDD outputs (NFE = 3, CFG = 1.0) with previously generated 1-step and 2-step reference images.
Figure 12: Uncurated samples of our text-to-image model
Figure 13: Samples from our 1-step student generator on ImageNet (FID=2.58).
Figure 14: Samples from our 2-step student generator on ImageNet (FID=1.99).
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.
Haoyu Li, Tingyan Wen, Lin Qi +6
Tsinghua University · Central Media Technology Institute, Huawei
Flow and diffusion models suffer from slow inference due to computationally expensive numerical integration. Distillation provides a promising way for a student model to learn from a teacher's dynamics, enabling one-step or few-step generation. However, existing methods often depend on curated distillation datasets, costly teacher rollouts, or auxiliary proxy networks, which complicate model training and scaling. In this work, we propose Consistent Distribution Matching, a simulation-free and data-free distillation method for accelerating diffusion and flow models while preserving strong generative capacity. Our key insight is to unify sample generation and score estimation with one student network. Thus, our framework uses only two models, a frozen teacher and a trainable student, and optimizes one objective. We prove that minimizing our objective indicates Wasserstein convergence of the student flow-map pushforwards to the teacher marginals. On ImageNet 256×256, our method attains an FID of 2.04 with a single function evaluation (1-NFE) and a 4-NFE FID of 1.37 within 40 epochs of training, surpassing the state-of-the-art distillation baselines without data. Our code code and model are available at https://consistentdmd.github.io/.
Yuxiang Fu, Qi Yan, Zike Wu +4
University of British Columbia · Vector Institute for AI · Canada CIFAR AI Chair
Discrete diffusion models excel at visual synthesis but rely on slow, iterative decoding. Existing single-step distillation methods attempt to bypass this bottleneck, either by training auxiliary score networks that effectively double compute, or by introducing specialized parameterizations and multi-stage pipelines that fragment optimization. In this paper, we introduce Fixed-Point Distillation (FPD), an end-to-end framework that constructs local correction targets by partially corrupting the student's one-step draft and refining it with a single teacher step. To compute the training objective in a semantically meaningful space, we lift discrete tokens into continuous features and apply a multi-bandwidth drift loss that iteratively accumulates these corrections. To backpropagate through the discrete bottleneck, we employ a straight-through estimator that feeds exact hard-sampled tokens to the teacher and decoder during the forward pass, ensuring that training and inference operate on the same codebook manifold, while routing continuous gradients back to the student logits. This fully differentiable pathway additionally accommodates an optional unconditional adversarial objective to enhance perceptual realism. Evaluations on both class- and text-conditional generation validate the effectiveness of our framework. FPD achieves competitive visual fidelity and structural alignment within a single inference step, narrowing the gap to multi-step teachers while outperforming existing discrete distillation baselines.