arXiv:2611.11365 · v1
MG-Cache: Multi-Granularity Caching for Accelerating Diffusion Transformer Inference
摘要
Diffusion Transformer (DiT) models, while having become a powerful backbone for generative synthesis, suffer from high inference cost on generating high-resolution images. Diffusion caching has emerged as a promising acceleration technique to reduce inference cost. However, existing caching strategies fail to accurately identify the varying importance of different computational modules and rarely coordinate cache reusing across the timestep and layer granularities, making it difficult to control error accumulation. To address these limitations, we propose multi-granularity caching (MG-Cache) for DiTs, a novel and fully training-free approach to exploit computational redundancy both on timesteps and Transformer layers. Our core innovation contains a two-stage caching strategy: an offline timestep profiling analysis that identifies redundant timesteps for cache reuse, followed by an adaptive layer-wise caching strategy that combines input stability with importance of each Transformer layer to selectively cache stable Attention and MLP submodules. Extensive experiments on mainstream open-source DiT models validate the effectiveness of our approach, achieving significant acceleration while maintaining high fidelity to the uncached output, with up to a 4.18x acceleration on HunyuanVideo.
作者备注：Published in Pacific Graphics 2026 (PG'26), https://doi.org/10.2312/pg.20261030
期刊信息：Zesen Chen, Youxuan Xu, Shigang Li. MG-Cache: Multi-Granularity Caching for Accelerating Diffusion Transformer Inference. Pacific Graphics 2026