Sparse-dLLM: Accelerating Diffusion LLMs with Dynamic Cache Eviction
2508.02558v1
cs.CL
2025-08-09
Авторы:
Yuerong Song, Xiaoran Liu, Ruixiao Li, Zhigeng Liu, Zengfeng Huang, Qipeng Guo, Ziwei He, Xipeng Qiu
Резюме на русском
Diffusion Large Language Models (dLLMs) обеспечивают значительные успехи в области рассуждений и параллельного декодирования, однако их вычислительная сложность и памятная нагрузка в разумении достигают заметных проблем. Основной причиной является квадратичный рост сложности при операции сохранения всех полных слоев состояний. Для устранения этой проблемы, авторы предложили **Sparse-dLLM** — новую методологию, основанную на динамической вытеснении кэша и спарсинге в ходе расчетов. Основа нового подхода — исследование паттернов внимания в dLLMs, открывающее существующую спарсингную структуру, при этом ключевые токены сохраняются в процессе уменьшения памятных нагрузок. Исследования показали, что Sparse-dLLM обеспечивает до 10-кратного увеличения пропускной способности по сравнению с обычным dLLM, сохраняя показатели качества и оптимальные затраты памяти. Это демонстрирует эффективность нового подхода в решении проблемы в параллельных декодированиях.
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.
Ссылки и действия
Дополнительные ресурсы: