Fewer Denoising Steps or Cheaper Per-Step Inference: Towards Compute-Optimal Diffusion Model Deployment
Zhenbang Du, Yonggan Fu, Lifu Wang, Jiayi Qian, Xiao Luo, Yingyan (Celine) Lin
Abstract
Diffusion models have shown remarkable success across generative tasks, yet their high computational demands challenge deployment on resource-limited platforms. This paper investigates a critical question for compute-optimal diffusion model deployment: Under a post-training setting without fine-tuning, is it more effective to reduce the number of denoising steps or to use a cheaper per-step inference? Intuitively, reducing the number of denoising steps increases the variability of the distributions across steps, making the model more sensitive to compression. In contrast, keeping more denoising steps makes the differences smaller, preserving redundancy, and making post-training compression more feasible. To systematically examine this, we propose PostDiff, a training-free framework for accelerating pre-trained diffusion models by reducing redundancy at both the input level and module level in a post-training manner. At the input level, we propose a mixed-resolution denoising scheme based on the insight that reducing generation resolution in early denoising steps can enhance low-frequency components and improve final generation fidelity. At the module level, we employ a hybrid module caching strategy to reuse computations across denoising steps. Extensive experiments and ablation studies demonstrate that (1) PostDiff can significantly improve the fidelity-efficiency trade-off of state-of-the-art diffusion models, and (2) to boost efficiency while maintaining decent generation fidelity, reducing per-step inference cost is often more effective than reducing the number of denoising steps. Our code is available at https://github.com/GATECH-EIC/PostDiff.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Builds on32
- Denoising Diffusion Probabilistic ModelsJonathan Ho, Ajay Jain, Pieter AbbeelNeurIPS 2020 · 35,902 citations
- Diffusion Models Beat GANs on Image SynthesisPrafulla Dhariwal, Alexander Quinn NicholNeurIPS 2021 · 13,211 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Denoising Diffusion Implicit ModelsJiaming Song, Chenlin Meng, Stefano ErmonICLR 2021 · 11,743 citations
- Scalable Diffusion Models with TransformersWilliam Peebles, Saining XieICCV 2023 · 5,568 citations
Related papers
- AccuQuant: Simulating Multiple Denoising Steps for Quantizing Diffusion ModelsSeunghoon Lee, Jeongwoo Choi, Byunggwan Son, Jaehyeon Moon et al.NeurIPS 2025 · 3 citations
- Modulated Diffusion: Accelerating Generative Modeling with Modulated QuantizationWeizhi Gao, Zhichao Hou, Junqi Yin, Feiyi Wang et al.ICML 2025
- Leveraging Early-Stage Robustness in Diffusion Models for Efficient and High-Quality Image SynthesisYulhwa Kim, Dongwon Jo, Hyesung Jeon, Taesu Kim et al.NeurIPS 2023 · 15 citations
- PixelRush: Ultra-Fast, Training-Free High-Resolution Image Generation via One-step DiffusionHong-Phuc Lai, Phong Nguyen, Anh TranCVPR 2026 · 2 citations
- Beyond Uniformity: Sample and Frequency Meta Weighting for Post-Training Quantization of Diffusion ModelsVan Cuong Pham, Anh Hoang, Cuong Nguyen, Trung Le et al.ICLR 2026
