发现扩散模型训练中可提前激活的稀疏子网络,显著提速且不损失生成质量。
Early-Bird Diffusion: Investigating and Leveraging Timestep-Aware Early-Bird Tickets in Diffusion Models for Efficient Training
- 利用早期出现的稀疏子网络,按时间步重要性动态调整稀疏度。
- 训练速度提升2.9至5.8倍,比传统剪枝微调快10.3倍。
- 适合追求高效训练的图像生成研究者和工业部署团队。
扩散模型(DMs)训练需大量计算资源,因需在多个时间步进行多次前向与反向传播。本文提出EB-Diff-Train方法,通过研究并利用早期鸟票(Early-Bird, EB)票——训练初期即出现且保持高生成质量的稀疏子网络,实现高效训练。我们首次验证了传统EB票在扩散模型中的存在,可在不完全训练稠密模型的情况下实现竞争性生成质量。进一步提出扩散专用的EB票,基于不同时间步区域的重要性差异,动态调节稀疏程度:非关键区域大幅稀疏,关键区域保留计算资源。基于此,我们开发了一种时间步感知的高效训练技术,可并行训练这些票,并在推理时融合生成图像。大量实验验证了传统与时间步感知EB票的存在性及有效性。该方法可显著降低时空开销——训练速度比未剪枝稠密模型快2.9至5.8倍,比标准训练-剪枝-微调流程快达10.3倍,且不牺牲生成质量。代码已开源:https://github.com/GATECH-EIC/Early-Bird-Diffusion。
原文摘要 · Abstract (English)
Training diffusion models (DMs) requires substantial computational resources due to multiple forward and backward passes across numerous timesteps, motivating research into efficient training techniques. In this paper, we propose EB-Diff-Train, a new efficient DM training approach that is orthogonal to other methods of accelerating DM training, by investigating and leveraging Early-Bird (EB) tickets -- sparse subnetworks that manifest early in the training process and maintain high generation quality. We first investigate the existence of traditional EB tickets in DMs, enabling competitive generation quality without fully training a dense model. Then, we delve into the concept of diffusion-dedicated EB tickets, drawing on insights from varying importance of different timestep regions. These tickets adapt their sparsity levels according to the importance of corresponding timestep regions, allowing for aggressive sparsity during non-critical regions while conserving computational resources for crucial timestep regions. Building on this, we develop an efficient DM training technique that derives timestep-aware EB tickets, trains them in parallel, and combines them during inference for image generation. Extensive experiments validate the existence of both traditional and timestep-aware EB tickets, as well as the effectiveness of our proposed EB-Diff-Train method. This approach can significantly reduce training time both spatially and temporally -- achieving 2.9$\times$ to 5.8$\times$ speedups over training unpruned dense models, and up to 10.3$\times$ faster training compared to standard train-prune-finetune pipelines -- without compromising generative quality. Our code is available at https://github.com/GATECH-EIC/Early-Bird-Diffusion.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。