Reducing Spatial Fitting Error in Distillation of Denoising Diffusion Models

被引:0
|
作者
Zhou, Shengzhe [1 ]
Li, Zejian [1 ]
Zhang, Shengyuan [2 ]
Hou, Lefan [2 ]
Yang, Changyuan [3 ]
Yang, Guang [3 ]
Yang, Zhiyuan [3 ]
Sun, Lingyun [2 ]
机构
[1] Zhejiang Univ, Sch Software Technol, Hangzhou, Peoples R China
[2] Zhejiang Univ, Coll Comp Sci & Technol, Hangzhou, Peoples R China
[3] Alibaba Grp, Hangzhou, Peoples R China
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
D O I
暂无
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Denoising Diffusion models have exhibited remarkable capabilities in image generation. However, generating high-quality samples requires a large number of iterations. Knowledge distillation for diffusion models is an effective method to address this limitation with a shortened sampling process but causes degraded generative quality. Based on our analysis with biasvariance decomposition and experimental observations, we attribute the degradation to the spatial fitting error occurring in the training of both the teacher and student model. Accordingly, we propose Spatial Fitting-Error Reduction Distillation model (SFERD). SFERD utilizes attention guidance from the teacher model and a designed semantic gradient predictor to reduce the student's fitting error. Empirically, our proposed model facilitates high-quality sample generation in a few function evaluations. We achieve an FID of 5.31 on CIFAR-10 and 9.39 on ImageNet 64x64 with only one step, outperforming existing diffusion methods. Our study provides a new perspective on diffusion distillation by highlighting the intrinsic denoising ability of models.
引用
收藏
页码:7686 / 7694
页数:9
相关论文
共 50 条
  • [1] Privacy Distillation: Reducing Re-identification Risk of Diffusion Models
    Fernandez, Virginia
    Sanchez, Pedro
    Pinaya, Walter Hugo Lopez
    Jacenkow, Grzegorz
    Tsaftaris, Sotirios A.
    Cardoso, M. Jorge
    DEEP GENERATIVE MODELS, DGM4MICCAI 2023, 2024, 14533 : 3 - 13
  • [2] DENOISING DIFFUSION MEDICAL MODELS
    Huy, Pham Ngoc
    Quan, Tran Minh
    2023 IEEE 20TH INTERNATIONAL SYMPOSIUM ON BIOMEDICAL IMAGING, ISBI, 2023,
  • [3] Quantum Denoising Diffusion Models
    Koelle, Michael
    Stenzel, Gerhard
    Stein, Jonas
    Zielinski, Sebastian
    Ommer, Bjoen
    Linnhoff-Popien, Claudia
    2024 IEEE INTERNATIONAL CONFERENCE ON QUANTUM SOFTWARE, IEEE QSW 2024, 2024, : 88 - 98
  • [4] Residual Denoising Diffusion Models
    Liu, Jiawei
    Wang, Qiang
    Fan, Huijie
    Wang, Yinong
    Tang, Yandong
    Qu, Liangqiong
    2024 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2024, 2024, : 2773 - 2783
  • [5] Denoising Diffusion Restoration Models
    Kawar, Bahjat
    Elad, Michael
    Ermon, Stefano
    Song, Jiaming
    ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS 35 (NEURIPS 2022), 2022,
  • [6] Fitting diffusion models in finance
    McLeish, DL
    Kolkiewicz, AW
    SELECTED PROCEEDINGS OF THE SYMPOSIUM ON ESTIMATING FUNCTIONS, 1997, 32 : 327 - 350
  • [7] On Distillation of Guided Diffusion Models
    Meng, Chenlin
    Rombach, Robin
    Gao, Ruiqi
    Kingma, Diederik
    Ermon, Stefano
    Ho, Jonathan
    Salimans, Tim
    2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, : 14297 - 14306
  • [8] Efficient Denoising of Ultrasonic Logging While Drilling Images: Multinoise Diffusion Denoising and Distillation
    Zhang, Wei
    Qu, Qiaofeng
    Qiu, Ao
    Li, Zhipeng
    Liu, Xien
    Li, Yanjun
    IEEE Transactions on Geoscience and Remote Sensing, 2025, 63
  • [9] Improved outcome models with denoising diffusion
    Dudas, D.
    Dilling, T. J.
    El Naqa, I.
    PHYSICA MEDICA-EUROPEAN JOURNAL OF MEDICAL PHYSICS, 2024, 119
  • [10] DENOISING TASK ROUTING FOR DIFFUSION MODELS
    Park, Byeongjun
    Woo, Sangmin
    Go, Hyojun
    Kim, Jin-Young
    Kim, Changick
    12th International Conference on Learning Representations, ICLR 2024, 2024,