arXiv:2503.06923cs.CVcs.AI2025-03ICCV被引 123

用泰勒展开预测扩散模型未来特征,实现高效无损加速

From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers

论文配图:From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers
图 1 · 摘自论文原文
  • 基于前序特征用泰勒级数预测未来特征,避免重复计算
  • 在FLUX和HunyuanVideo上实现4.99×和5.00×加速且几乎无质量损失
  • 无需额外训练,适合实时图像视频生成场景

扩散变换器(DiT)在高保真图像与视频生成中取得突破,但其计算开销限制了实时应用。现有特征缓存方法在时间步间隔较大时因特征相似性下降导致误差显著上升,影响生成质量。本文提出TaylorSeer,首次证明可通过前序特征预测未来时间步的特征。基于特征在时间上变化平缓连续的特性,采用微分方法近似高阶导数,利用泰勒展开预测未来特征。大量实验表明,该方法在图像与视频生成中均具显著有效性,尤其在高加速比下表现突出。例如,在FLUX和HunyuanVideo上分别实现4.99×和5.00×加速且几乎无质量损失;在DiT上以4.53×加速获得3.41更低的FID,优于当前最先进方法。

原文摘要 · Abstract (English)

Diffusion Transformers (DiT) have revolutionized high-fidelity image and video synthesis, yet their computational demands remain prohibitive for real-time applications. To solve this problem, feature caching has been proposed to accelerate diffusion models by caching the features in the previous timesteps and then reusing them in the following timesteps. However, at timesteps with significant intervals, the feature similarity in diffusion models decreases substantially, leading to a pronounced increase in errors introduced by feature caching, significantly harming the generation quality. To solve this problem, we propose TaylorSeer, which firstly shows that features of diffusion models at future timesteps can be predicted based on their values at previous timesteps. Based on the fact that features change slowly and continuously across timesteps, TaylorSeer employs a differential method to approximate the higher-order derivatives of features and predict features in future timesteps with Taylor series expansion. Extensive experiments demonstrate its significant effectiveness in both image and video synthesis, especially in high acceleration ratios. For instance, it achieves an almost lossless acceleration of 4.99$\times$ on FLUX and 5.00$\times$ on HunyuanVideo without additional training. On DiT, it achieves $3.41$ lower FID compared with previous SOTA at $4.53$$\times$ acceleration. %Our code is provided in the supplementary materials and will be made publicly available on GitHub. Our codes have been released in Github:https://github.com/Shenyi-Z/TaylorSeer

扩散模型加速生成泰勒展开视频生成

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。