PulseExploreJournal ClubDebatesTrendingResearchersJournals
Instagram
HomeExploreJournal ClubTrending
Synapse
⌘+K
Synapse
October 10, 20250 citationsOpen Access

Self Speculative Decoding for Diffusion Large Language Models

View Full Paper
YGYifeng GaoZJZhonghui JiYWYuxuan Wang

Key Points

  • SSD achieves a speedup of up to 3.46× while maintaining output quality similar to stepwise decoding.
  • By leveraging the model's bidirectional attention, SSD eliminates redundancy and memory overhead in predictions.
  • The hierarchical verification trees allow the model to progressively accept multiple tokens in one forward pass.
  • Experiments demonstrate SSD's efficiency across open-source models like LLaDA and Dream with practical implications.

Abstract

Diffusion-based Large Language Models (dLLMs) have emerged as a competitive alternative to autoregressive models, offering unique advantages through bidirectional attention and parallel generation paradigms. However, the generation results of current parallel decoding methods deviate from stepwise decoding, introducing potential performance degradation, which limits their practical deployment. To address this problem, we propose Self Speculative Decoding (SSD), a lossless inference acceleration method that leverages the dLLM itself as both speculative decoding drafter and verifier without auxiliary modules. SSD introduces a self-drafting mechanism where the model generates predictions for multiple positions, then verifies them through hierarchical verification trees in a single forward pass. Unlike traditional speculative decoding that requires separate draft models, SSD eliminates model redundancy and memory overhead by exploiting the dLLM's inherent parallel prediction capability for multiple positions. This self-speculative approach allows the model to progressively verify and accept multiple tokens in a single forward pass. Our experiments demonstrate that SSD achieves up to 3. 46 speedup while keeping the output identical to stepwise decoding on open source models such as LLaDA and Dream. Code will be made publicly available on GitHub.

Ask AI
Helpful
Bookmark
Share
View Full Paper

Cite This Study

Gao et al. (2025) studied this question.

synapsesocial.com/papers/68e90f526476c097794aa44fhttps://doi.org/10.48550/arxiv.2510.04147
Ask AI
Helpful
Bookmark
Share
View Full Paper