From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers
By: Jiacheng Liu , Chang Zou , Yuanhuiyi Lyu and more
Potential Business Impact:
Makes AI create pictures and videos much faster.
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
Similar Papers
Forecasting When to Forecast: Accelerating Diffusion Models with Confidence-Gated Taylor
CV and Pattern Recognition
Makes AI art tools much faster without losing quality.
Forecasting When to Forecast: Accelerating Diffusion Models with Confidence-Gated Taylor
CV and Pattern Recognition
Makes AI image makers much faster.
Foresight: Adaptive Layer Reuse for Accelerated and High-Quality Text-to-Video Generation
Machine Learning (CS)
Makes AI videos faster without losing quality.