用进化搜索自动优化扩散模型分阶段剪枝,提速同时保质。
Diff-ES: Stage-wise Structural Diffusion Pruning via Evolutionary Search
- 通过进化搜索自动发现各阶段最优稀疏度策略
- 在DiT和SDXL上实现显著加速且生成质量损失小
- 无需复制模型参数,内存效率高,适合部署
扩散模型在高质量图像生成上取得显著进展,但因多步去噪过程和大模型规模,计算开销巨大。现有结构化剪枝方法难以兼顾实际加速与图像质量保持。例如MosaicDiff依赖手动设定的阶段稀疏度调度,并在推理时拼接多个独立剪枝模型,增加内存开销。然而,扩散步骤的重要性高度非均匀且依赖模型。基于简单启发式或经验观察的调度往往泛化性差,导致性能不佳。为此,我们提出Diff-ES:一种基于进化搜索的分阶段结构化扩散剪枝框架,通过进化搜索自动优化阶段稀疏度调度,并利用内存高效的权重路由机制执行,无需模型参数复制。Diff-ES将扩散轨迹划分为多个阶段,动态激活阶段条件权重,天然兼容深度与宽度剪枝等现有方法。在DiT和SDXL上的大量实验表明,Diff-ES持续实现墙钟时间加速,生成质量下降极小,达到当前结构化扩散模型剪枝的最先进水平。
原文摘要 · Abstract (English)
Diffusion models have achieved remarkable success in high-fidelity image generation but remain computationally demanding due to their multi-step denoising process and large model sizes. Although prior work improves efficiency either by reducing sampling steps or by compressing model parameters, existing structured pruning approaches still struggle to balance real acceleration and image quality preservation. In particular, prior methods such as MosaicDiff rely on heuristic, manually tuned stage-wise sparsity schedules and stitch multiple independently pruned models during inference, which increases memory overhead. However, the importance of diffusion steps is highly non-uniform and model-dependent. As a result, schedules derived from simple heuristics or empirical observations often fail to generalize and may lead to suboptimal performance. To this end, we introduce \textbf{Diff-ES}, a stage-wise structural \textbf{Diff}usion pruning framework via \textbf{E}volutionary \textbf{S}earch, which optimizes the stage-wise sparsity schedule and executes it through memory-efficient weight routing without model duplication. Diff-ES divides the diffusion trajectory into multiple stages, automatically discovers an optimal stage-wise sparsity schedule via evolutionary search, and activates stage-conditioned weights dynamically without duplicating model parameters. Our framework naturally integrates with existing structured pruning methods for diffusion models including depth and width pruning. Extensive experiments on DiT and SDXL demonstrate that Diff-ES consistently achieves wall-clock speedups while incurring minimal degradation in generation quality, establishing state-of-the-art performance for structured diffusion model pruning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。