Sparse-dLLM: Accelerating Diffusion LLMs with Dynamic Cache Eviction

Yuerong Song, Xiaoran Liu, Ruixiao Li, Zhigeng Liu, Zengfeng Huang +3 more
8/4/2025
cs.CL

Official code: OpenMOSS/Sparse-dLLM (29 stars, last commit 2025-10-16).

Abstract

Diffusion Large Language Models (dLLMs) enable breakthroughs in reasoning and parallel decoding but suffer from prohibitive quadratic computational complexity and memory overhead during inference. Current caching techniques accelerate decoding by storing full-layer states, yet impose substantial memory usage that limit long-context applications. Our analysis of attention patterns in dLLMs reveals persistent cross-layer sparsity, with pivotal tokens remaining salient across decoding steps and low-relevance tokens staying unimportant, motivating selective cache eviction. We propose Sparse-dLLM, the first training-free framework integrating dynamic cache eviction with sparse attention via delayed bidirectional sparse caching. By leveraging the stability of token saliency over steps, it retains critical tokens and dynamically evicts unimportant prefix/suffix entries using an attention-guided strategy. Extensive experiments on LLaDA and Dream series demonstrate Sparse-dLLM achieves up to 10$\times$ higher throughput than vanilla dLLMs, with comparable performance and similar peak memory costs, outperforming previous methods in efficiency and effectiveness. The code is available at https://github.com/OpenMOSS/Sparse-dLLM.

View on arXivView PDF

Code Implementations(1)

293Shell, PythonOct 13, 20251 years ago

Similar Papers

Cite this paper

@article{song2025sparsedllm,
  title  = {Sparse-dLLM: Accelerating Diffusion LLMs with Dynamic Cache Eviction},
  author = {Yuerong Song and Xiaoran Liu and Ruixiao Li and Zhigeng Liu and Zengfeng Huang and Qipeng Guo and Ziwei He and Xipeng Qiu},
  year   = {2025},
  eprint = {2508.02558},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2508.02558v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2508.02558.svg)](https://codeofpaper.com/paper/2508.02558)

Discussion