让扩散模型更快生成,同时恢复多样性的新方法
Distilling Diversity and Control in Diffusion Models
- 用初始步保留基础模型结构,后续切换到轻量模型
- 仅1个关键时间步的干预即恢复生成多样性
- 适合追求高效且多样图像生成的开发者
蒸馏后的扩散模型虽在更少步骤内生成图像,但同一提示下的样本多样性下降。我们发现蒸馏模型保留了原始模型的表征结构:概念滑块和LoRA等控制机制可直接迁移,且滑块空间分析显示其具备多样性所需的变化方向,却无法激活。问题根源在于生成动态差异:基础模型将结构决策分散在多个步骤,而蒸馏模型几乎在第一步就确定最终图像结构。为此,我们提出多样性蒸馏:仅在第一关键步骤使用基础模型,之后切换至蒸馏模型。该方法有效恢复多样性,同时保持计算效率。我们提供了因果验证与理论支持,证明首步决策是多样性瓶颈所在。代码与数据见 https://distillation.baulab.info/
原文摘要 · Abstract (English)
Distilled diffusion models generate images in far fewer timesteps but suffer from reduced sample diversity when generating multiple outputs from the same prompt. To understand this phenomenon, we first investigate whether distillation damages concept representations by examining if the required diversity is properly learned. Surprisingly, distilled models retain the base model's representational structure: control mechanisms like Concept Sliders and LoRAs transfer seamlessly without retraining, and SliderSpace analysis reveals distilled models possess variational directions needed for diversity yet fail to activate them. This redirects our investigation to understanding how the generation dynamics differ between base and distilled models. Using $\hat{\mathbf{x}}_{0}$ trajectory visualization, we discover distilled models commit to their final image structure almost immediately at the first timestep, while base models distribute structural decisions across many steps. To test whether this first-step commitment causes the diversity loss, we introduce diversity distillation, a hybrid approach using the base model for only the first critical timestep before switching to the distilled model. This single intervention restores sample diversity while maintaining computational efficiency. We provide both causal validation and theoretical support showing why the very first timestep concentrates the diversity bottleneck in distilled models. Our code and data are available at https://distillation.baulab.info/
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。