提出新方法,让图像生成模型一步完成且更高效。
WaDi: Weight Direction-aware Distillation for One-step Image Synthesis
- 通过分析权重方向变化,发现其比大小变化更重要。
- 用低秩旋转矩阵仅用10%参数实现顶尖生成质量。
- 适合需要快速生成的部署场景,尤其支持可控生成。
尽管扩散模型如Stable Diffusion在图像生成上表现优异,但其推理速度慢限制了实际应用。近期工作通过将多步扩散过程压缩为单步生成器来加速推理。为深入理解这一过程,我们分析了单步学生模型与多步教师模型在U-Net/DiT权重上的变化。结果表明,权重方向的变化远超其模长变化,凸显方向调整的关键作用。基于此,我们提出低秩旋转权重方向(LoRaD)——一种专为单步扩散蒸馏设计的参数高效适配器,利用可学习的低秩旋转矩阵建模结构化方向变化。我们将LoRaD集成至变分分数蒸馏(VSD),形成权重方向感知蒸馏(WaDi)——一种新型单步蒸馏框架。WaDi在COCO 2014和COCO 2017上达到当前最优FID得分,且仅需原U-Net/DiT约10%的可训练参数。此外,蒸馏后的单步模型展现出强大泛化能力,在可控生成、关系逆推和高分辨率合成等下游任务中均有良好表现。
原文摘要 · Abstract (English)
Despite the impressive performance of diffusion models such as Stable Diffusion (SD) in image generation, their slow inference limits practical deployment. Recent works accelerate inference by distilling multi-step diffusion into one-step generators. To better understand the distillation mechanism, we analyze U-Net/DiT weight changes between one-step students and their multi-step teacher counterparts. Our analysis reveals that changes in weight direction significantly exceed those in weight norm, highlighting it as the key factor during distillation. Motivated by this insight, we propose the Low-rank Rotation of weight Direction (LoRaD), a parameter-efficient adapter tailored to one-step diffusion distillation. LoRaD is designed to model these structured directional changes using learnable low-rank rotation matrices. We further integrate LoRaD into Variational Score Distillation (VSD), resulting in Weight Direction-aware Distillation (WaDi)-a novel one-step distillation framework. WaDi achieves state-of-the-art FID scores on COCO 2014 and COCO 2017 while using only approximately 10% of the trainable parameters of the U-Net/DiT. Furthermore, the distilled one-step model demonstrates strong versatility and scalability, generalizing well to various downstream tasks such as controllable generation, relation inversion, and high-resolution synthesis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。