Accelerating Diffusion Sampling with Classifier-based Feature Distillation
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Although diffusion model has shown great potential for generating higher quality images than GANs, slow sampling speed hinders its wide application in practice. Progressive distillation is thus proposed for fast sampling by progressively aligning output images of N-step teacher sampler with N/2-step student sampler. In this paper, we argue that this distillation-based accelerating method can be further improved, especially for few-step samplers, with our proposed Classifier-based Feature Distillation (CFD). Instead of aligning output images, we distill teacher's sharpened feature distribution into the student with a dataset-independent classifier, making the student focus on those important features to improve performance. We also introduce a dataset-oriented loss to further optimize the model. Experiments on CIFAR-10 show the superiority of our method in achieving high quality and fast sampling. Code is provided at urlhttps://github.com/zju-SWJ/RCFD.
Key facts. Cited by 15 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 23 of its 29 references that are in the map. First posted on arXiv 22 Nov 2022.
Builds on
23 of 29 references.
Closest ancestors · 8
Same direction, newest first
- 2022DPM-Solver: A Fast ODE Solver for Diffusion Probabilistic Model Sampling in Around 10 Steps2 Jun
- 2022Pseudo Numerical Methods for Diffusion Models on Manifolds20 Feb
- 2021On Fast Sampling of Diffusion Probabilistic Models31 May
- 2021Diffusion Models Beat GANs on Image Synthesis11 May
- 2021Improved Denoising Diffusion Probabilistic Models18 Feb
- 2021Knowledge Distillation in Iterative Generative Models for Improved Sampling Speed7 Jan
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 15
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2015Distilling the Knowledge in a Neural Networkcited by 10,076
- 2016Densely Connected Convolutional Networkscited by 6,691
- 2016Improved Techniques for Training GANscited by 2,938
- 2014FitNets: Hints for Thin Deep Netscited by 1,619
- 2020Knowledge Distillation: A Surveycited by 1,578
- 2018Born Again Neural Networkscited by 539
- 2020Channel-wise Knowledge Distillation for Dense Predictioncited by 161
- 2020Cross-Layer Distillation with Semantic Calibrationcited by 141
- 2022Knowledge Distillation with the Reused Teacher Classifiercited by 140
- 2019Online Knowledge Distillation with Diverse Peerscited by 127
Built on by
Citing papers per year: 2023: 3; 2024: 8; 2025: 3; 2026: 1.
15 papers.
Sort:
By area:
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 59
- 2023A Comprehensive Survey on Knowledge Distillation of Diffusion Modelscited by 45
- 2024Simple and Fast Distillation of Diffusion Modelscited by 31
- 2024Towards a mathematical theory for consistency training in diffusion modelscited by 25
- 2023Catch-Up Distillation: You Only Need to Train Once for Accelerating Samplingcited by 13
- 2024Knowledge Distillation with Refined Logitscited by 9
- 2024Accelerating Diffusion Models with One-to-Many Knowledge Distillationcited by 9
- 2025FastFLUX: Pruning FLUX with Block-wise Replacement and Sandwich Trainingcited by 5
- 2024Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Modelscited by 4
- 2024Theoretical research on generative diffusion models: an overviewcited by 2
- 2025Image Recognition with Online Lightweight Vision Transformer: A Surveycited by 2
- 2024Distribution Backtracking Builds A Faster Convergence Trajectory for Diffusion Distillationcited by 1
- 2026Teacher-Guided Student Self-Knowledge Distillation Using Diffusion Model2 Feb 2026
- 2025FastFLUX: Pruning FLUX with Block-wise Replacement and Sandwich Training10 Jun 2025
- 2025Image Recognition with Online Lightweight Vision Transformer: A Survey6 May 2025
- 2025Asymmetric Decision-Making in Online Knowledge Distillation:Unifying Consensus and Divergence9 Mar 2025
- 2024Accelerating Diffusion Models with One-to-Many Knowledge Distillation5 Oct 2024
- 2024Simple and Fast Distillation of Diffusion Models29 Sep 2024
- 2024Distribution Backtracking Builds A Faster Convergence Trajectory for Diffusion Distillation28 Aug 2024
- 2024Knowledge Distillation with Refined Logits14 Aug 2024
- 2024Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Models30 Jun 2024
- 2024Theoretical research on generative diffusion models: an overview13 Apr 2024
- 2024Towards a mathematical theory for consistency training in diffusion models12 Feb 2024
- 2024Knowledge Translation: A New Pathway for Model Compression11 Jan 2024
- 2023On the Design Fundamentals of Diffusion Models: A Surveycited by 59
- 2023A Comprehensive Survey on Knowledge Distillation of Diffusion Modelscited by 45
- 2024Simple and Fast Distillation of Diffusion Modelscited by 31
- 2024Towards a mathematical theory for consistency training in diffusion modelscited by 25
- 2024Accelerating Diffusion Models with One-to-Many Knowledge Distillationcited by 9
- 2025FastFLUX: Pruning FLUX with Block-wise Replacement and Sandwich Trainingcited by 5
- 2024Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Modelscited by 4
- 2024Theoretical research on generative diffusion models: an overviewcited by 2
- 2024Distribution Backtracking Builds A Faster Convergence Trajectory for Diffusion Distillationcited by 1
- 2026Teacher-Guided Student Self-Knowledge Distillation Using Diffusion Modelcited by 0
- 2026Teacher-Guided Student Self-Knowledge Distillation Using Diffusion Model2 Feb 2026
- 2025FastFLUX: Pruning FLUX with Block-wise Replacement and Sandwich Training10 Jun 2025
- 2024Accelerating Diffusion Models with One-to-Many Knowledge Distillation5 Oct 2024
- 2024Simple and Fast Distillation of Diffusion Models29 Sep 2024
- 2024Distribution Backtracking Builds A Faster Convergence Trajectory for Diffusion Distillation28 Aug 2024
- 2024Maximum Entropy Inverse Reinforcement Learning of Diffusion Models with Energy-Based Models30 Jun 2024
- 2024Theoretical research on generative diffusion models: an overview13 Apr 2024
- 2024Towards a mathematical theory for consistency training in diffusion models12 Feb 2024
- 2023On the Design Fundamentals of Diffusion Models: A Survey7 Jun 2023
- 2023A Comprehensive Survey on Knowledge Distillation of Diffusion Models9 Apr 2023
- 2023Catch-Up Distillation: You Only Need to Train Once for Accelerating Samplingcited by 13
- 2024Knowledge Distillation with Refined Logitscited by 9
- 2025Image Recognition with Online Lightweight Vision Transformer: A Surveycited by 2
- 2024Knowledge Translation: A New Pathway for Model Compressioncited by 0
- 2025Asymmetric Decision-Making in Online Knowledge Distillation:Unifying Consensus and Divergencecited by 0
- 2025Image Recognition with Online Lightweight Vision Transformer: A Survey6 May 2025
- 2025Asymmetric Decision-Making in Online Knowledge Distillation:Unifying Consensus and Divergence9 Mar 2025
- 2024Knowledge Distillation with Refined Logits14 Aug 2024
- 2024Knowledge Translation: A New Pathway for Model Compression11 Jan 2024
- 2023Catch-Up Distillation: You Only Need to Train Once for Accelerating Sampling18 May 2023
Similar, not cited
Close in wording, no citation either way
- 2022Progressive Distillation for Fast Sampling of Diffusion Models
- 2023Adversarial Diffusion Distillation
- 2025Revisiting Diffusion Models: From Generative Pre-training to One-Step Generation
- 2024Unlocking Dataset Distillation with Diffusion Models
- 2026Teacher-Feature Drifting: One-Step Diffusion Distillation with Pretrained Diffusion Representations
- 2024Towards Consistent and Efficient Dataset Distillation via Diffusion-Driven Selection
- 2022On Distillation of Guided Diffusion Models
- 2026DisCa: Accelerating Video Diffusion Transformers with Distillation-Compatible Learnable Feature Caching