通过动态路由与联合优化,让扩散模型在少步数下高效生成高质量图像。
ALTER: All-in-One Layer Pruning and Temporal Expert Routing for Efficient Diffusion Generation
- 统一优化层剪枝与时间路由,用可训练超网络动态决策
- 20步推理达原模型50步画质,仅用25.9%计算量
- 适合资源受限场景下的快速图像生成应用
扩散模型虽能生成高保真图像,但其迭代去噪过程导致推理计算开销大,限制了在资源受限环境中的部署。现有加速方法多采用固定策略,难以捕捉生成过程的时间变化;而常见的先剪枝后微调流程因预训练权重与最终参数不匹配,存在次优问题。为此,我们提出ALTER:一种将扩散模型转化为高效时序专家混合体的统一框架。ALTER通过可训练超网络,在单阶段优化中同步实现层剪枝、专家路由与模型微调,动态生成每一步的剪枝策略并路由至专用剪枝子网络。该联合优化机制显著提升效率且保持生成质量:在仅20步推理下,达到与原版50步Stable Diffusion v2.1相当的视觉效果,仅需25.9%总MACs,实现3.64倍加速和35%稀疏度。
原文摘要 · Abstract (English)
Diffusion models have demonstrated exceptional capabilities in generating high-fidelity images. However, their iterative denoising process results in significant computational overhead during inference, limiting their practical deployment in resource-constrained environments. Existing acceleration methods often adopt uniform strategies that fail to capture the temporal variations during diffusion generation, while the commonly adopted sequential pruning-then-fine-tuning strategy suffers from sub-optimality due to the misalignment between pruning decisions made on pretrained weights and the model's final parameters. To address these limitations, we introduce ALTER: All-in-One Layer Pruning and Temporal Expert Routing, a unified framework that transforms diffusion models into a mixture of efficient temporal experts. ALTER achieves a single-stage optimization that unifies layer pruning, expert routing, and model fine-tuning by employing a trainable hypernetwork, which dynamically generates layer pruning decisions and manages timestep routing to specialized, pruned expert sub-networks throughout the ongoing fine-tuning of the UNet. This unified co-optimization strategy enables significant efficiency gains while preserving high generative quality. Specifically, ALTER achieves same-level visual fidelity to the original 50-step Stable Diffusion v2.1 model while utilizing only 25.9% of its total MACs with just 20 inference steps and delivering a 3.64x speedup through 35% sparsity.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。