Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching

About

Autoregressive Models (ARMs) have long dominated the landscape of Large Language Models. Recently, a new paradigm has emerged in the form of diffusion-based Large Language Models (dLLMs), which generate text by iteratively denoising masked segments. This approach has shown significant advantages and potential. However, dLLMs suffer from high inference latency. Traditional ARM acceleration techniques, such as Key-Value caching, are incompatible with dLLMs due to their bidirectional attention mechanism. To address this specific challenge, our work begins with a key observation that dLLM inference involves a static prompt and a partially dynamic response, where most tokens remain stable across adjacent denoising steps. Based on this, we propose dLLM-Cache, a training-free adaptive caching framework that combines long-interval prompt caching with partial response updates guided by feature similarity. This design enables efficient reuse of intermediate computations without compromising model performance. Extensive experiments on representative dLLMs, including LLaDA 8B and Dream 7B, show that dLLM-Cache achieves up to 9.1x FLOPs reduction on LongBench-HotpotQA while maintaining competitive output quality. Notably, our method brings dLLM inference latency close to that of ARMs under many settings. The code for this work is publicly available at: https://github.com/maomaocun/dLLM-cache.

Zhiyuan Liu, Yicun Yang, Yaojie Zhang, Junjie Chen, Chang Zou, Qingyan Wei, Shaobo Wang, Yichen Zhu, Linfeng Zhang• 2025

Related benchmarks

TaskDatasetResultRank
Multimodal UnderstandingMMBench--
887
Visual Question AnsweringChartQA
Accuracy78.1
620
Multimodal UnderstandingSEED-Bench
Accuracy72.1
571
Multitask Language UnderstandingMMLU
Accuracy73.2
568
Document Visual Question AnsweringDocVQA
ANLS82.1
301
Mathematical ReasoningGSM8K
Speed Up (x)3.67
246
Language UnderstandingMMLU 5-shot
Accuracy73.08
160
Code GenerationHumanEval 0-shot
Accuracy54.31
100
Code GenerationHumanEval
Score39.02
78
Multimodal UnderstandingMME Perception--
74
Showing 10 of 62 rows

Other info

Follow for update