Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

SpeCa: Accelerating Diffusion Transformers with Speculative Feature Caching

About

Diffusion models have revolutionized high-fidelity image and video synthesis, yet their computational demands remain prohibitive for real-time applications. These models face two fundamental challenges: strict temporal dependencies preventing parallelization, and computationally intensive forward passes required at each denoising step. Drawing inspiration from speculative decoding in large language models, we present SpeCa, a novel 'Forecast-then-verify' acceleration framework that effectively addresses both limitations. SpeCa's core innovation lies in introducing Speculative Sampling to diffusion models, predicting intermediate features for subsequent timesteps based on fully computed reference timesteps. Our approach implements a parameter-free verification mechanism that efficiently evaluates prediction reliability, enabling real-time decisions to accept or reject each prediction while incurring negligible computational overhead. Furthermore, SpeCa introduces sample-adaptive computation allocation that dynamically modulates resources based on generation complexity, allocating reduced computation for simpler samples while preserving intensive processing for complex instances. Experiments demonstrate 6.34x acceleration on FLUX with minimal quality degradation (5.5% drop), 7.3x speedup on DiT while preserving generation fidelity, and 79.84% VBench score at 6.1x acceleration for HunyuanVideo. The verification mechanism incurs minimal overhead (1.67%-3.5% of full inference costs), establishing a new paradigm for efficient diffusion model inference while maintaining generation quality even at aggressive acceleration ratios. Our codes have been released in Github: \textbf{https://github.com/Shenyi-Z/Cache4Diffusion}

Jiacheng Liu, Chang Zou, Yuanhuiyi Lyu, Fei Ren, Shaobo Wang, Kaixin Li, Linfeng Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Text-to-Video GenerationHunyuanVideo
LPIPS0.39
22
Robotic Manipulation (Square)Square PH
Success Rate88
16
Robotic Manipulation (Lift)Lift PH--
11
Robot ManipulationMixed Human (MH)
Success Rate (Lift)100
10
Multi-stage taskKitchen & BP
Success Rate (BP p1)0.97
10
Robot Manipulation (Push-T)Proficient Human (PH)
Success Rate (Criteria 1)67
5
Robot Manipulation (Can)Proficient Human (PH)
Success Rate (1)92
5
Robot Manipulation (Overall Suite Performance)Proficient Human (PH)
Avg Success Rate (Trial 1)76
5
Robot Manipulation (Tool)Proficient Human (PH)
Success Rate (Crit 1)38
5
Robot Manipulation (Transport)Proficient Human (PH)
Success Rate 168
5
Showing 10 of 10 rows

Other info

Follow for update