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

Wide-In, Narrow-Out: Revokable Decoding for Efficient and Effective DLLMs

About

Diffusion Large Language Models (DLLMs) have emerged as a compelling alternative to Autoregressive models, designed for fast parallel generation. However, existing DLLMs are plagued by a severe quality-speed trade-off, where faster parallel decoding leads to significant performance degradation. We attribute this to the irreversibility of standard decoding in DLLMs, which is easily polarized into the wrong decoding direction along with early error context accumulation. To resolve this, we introduce Wide-In, Narrow-Out (WINO), a training-free decoding algorithm that enables revokable decoding in DLLMs. WINO employs a parallel draft-and-verify mechanism, aggressively drafting multiple tokens while simultaneously using the model's bidirectional context to verify and re-mask suspicious ones for refinement. Verified in open-source DLLMs like LLaDA and MMaDA, WINO is shown to decisively improve the quality-speed trade-off. For instance, on the GSM8K math benchmark, it accelerates inference by 6$\times$ while improving accuracy by 2.58%; on Flickr30K captioning, it achieves a 10$\times$ speedup with higher performance. More comprehensive experiments are conducted to demonstrate the superiority and provide an in-depth understanding of WINO.

Feng Hong, Geng Yu, Yushi Ye, Haicheng Huang, Huangjie Zheng, Ya Zhang, Yanfeng Wang, Jiangchao Yao• 2025

Related benchmarks

TaskDatasetResultRank
Mathematical ReasoningMATH 500
Accuracy34.2
589
ReasoningARC
Accuracy85.31
269
Mathematical ReasoningCountdown
Accuracy33.2
252
Mathematical ReasoningGSM8K--
246
Logical reasoningSudoku
Accuracy15.2
152
Code GenerationHumanEval
Accuracy (%)54.88
77
Puzzle SolvingSudoku
Test Accuracy15.2
42
Mathematical ReasoningMATH500
Accuracy44.4
32
Mathematical ReasoningGSM8K
Accuracy82.03
32
Code GenerationMBPP
Accuracy57
32
Showing 10 of 18 rows

Other info

Follow for update