DiffRatio: Training One-Step Diffusion Models Without Teacher Supervision
Authors: Wenlin Chen, Mingtian Zhang, Jiajun He, Zijing Ou, José Miguel Hernández-Lobato, Bernhard Schölkopf, David Barber
Organizations: 1Bosch (China) Investment Ltd. · 2UCL · University of Cambridge · 4Imperial College London · 5Max Planck Institute for Intelligent Systems, Tübingen.
Score-based distillation methods train one-step diffusion models in two stages: they first train a teacher score model, then distill it into a one-step student model. However, this distillation process can introduce bias from two sources: errors in the teacher score model and student score estimate. We propose DiffRatio, a new approach for training one-step diffusion models without teacher supervision based on a density-ratio formulation that connects score-based distribution matching and GAN-style adversarial training. Instead of using a teacher score model to provide training targets, DiffRatio directly learns a log density ratio between the student and data distributions across diffusion time steps. As a pre-trained score model is only used to initialize the one-step generator and does not supervise training, this design simplifies the training pipeline, mitigates gradient estimation bias, and reduces the size of auxiliary networks. The learned density ratio can be used as a verifier, enabling a principled inference-time parallel scaling method that further improves sample quality without external rewards or extra sequential computation. DiffRatio achieves strong one-step generation results on CIFAR-10 and ImageNet (64x64 and 512x512), outperforming most teacher-supervised distillation methods.
Figures & tables
Figure 1Table 2Figure 3
Figure 3: Histograms of the learned density ratio w .
Figure 5
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
Figure 5: Visualization of samples from DiffRatio-DiJS trained on CIFAR10 (FID=2.39, IS=9.93).
Figure 6: Visualization of samples from DiffRatio-DiJS trained on ImageNet 64 × 64 (FID=1.54).
Figure 7: Visualization of samples from DiffRatio-DiJS-M trained on ImageNet 512 × 512 (FID=1.41).