DecoCal: Decoding with Calibration in Diffusion Large Language Models
Fan Xu, Huixuan Zhang, Xiaojun Wan
Abstract
Diffusion Large Language Models (DLLMs) generate text via iterative masked-token denoising, supporting parallel prediction and bidirectional context modeling. Despite these advantages, decoding remains challenging: many tokens appear predictable early, yet single-step predictions are often unstable, exhibiting temporal oscillations or overconfidence, making it difficult to determine which tokens can be safely committed. To address these challenges, we propose DecoCal 1 , a Decoding framework that explicitly performs Calibration of tokenlevel confidence across diffusion steps and leverages the calibrated results to guide decoding decisions. Specifically, DecoCal aggregates historical predictions to maintain calibrated confidence, triggering unmasking only when a token is sufficiently stable, while a remasking mechanism allows revision of premature commitments. This calibration-based design enables early decoding of reliably converged tokens while deferring or correcting unstable ones, balancing reliability and speed. Experiments on multiple DLLMs and benchmarks show that DecoCal improves generation accuracy compared to existing strategies. Our results highlight the importance of temporal calibration in unlocking the full potential of diffusion-based language generation. 𝐩 ! (#) 𝐪 !%& (#) 𝐪 ! (#) 𝑥 (') [MASK] 𝑥 (() [MASK] 𝐾𝐿 a. Confidence Check b. Consistency Check Calibration Sequence [MASK] 𝑥 (') 𝑥 (&) 𝑥 (() 𝑥 ()) [MASK] Unmask Remask 𝑥 (') [MASK] 𝑥 (() [MASK] [MASK] 𝑥 (') 𝑥 (&) [MASK] 𝑥 ()) [MASK] Cool-down Delayed Consistency Check
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c28c3361-37b3-4ff9-9719-e2c97b962055Builds on11
- Diffusion-LM Improves Controllable Text GenerationXiang Lisa Li, John Thickstun, Ishaan Gulrajani, Percy Liang et al.NeurIPS 2022 · 1,546 citations
- Large Language Diffusion ModelsShen Nie, Fengqi Zhu, Zebin You, Xiaolu Zhang et al.NeurIPS 2025 · 949 citations
- Self-Consistency Improves Chain of Thought Reasoning in Language ModelsXuezhi Wang, Jason Wei, Dale Schuurmans, Quoc V. Le et al.ICLR 2023 · 681 citations
- MaskGIT: Masked Generative Image TransformerHuiwen Chang, Han Zhang, Lu Jiang, Ce Liu et al.CVPR 2022 · 346 citations
- Latent Diffusion for Language GenerationJustin Lovelace, Varsha Kishore, Chao Wan, Eliot Shekhtman et al.NeurIPS 2023 · 177 citations
Related papers
- CreditDecoding: Accelerating Parallel Decoding in Diffusion Large Language Models with Trace CreditKangyu Wang, Zhiyun Jiang, Haibo Feng, Weijia Zhao et al.ACL 2026 · 11 citations
- Don't Settle Too Early: Self-Reflective Remasking for Diffusion Language ModelsZemin Huang, Yuhang Wang, Zhiyang Chen, Guo-Jun QiICLR 2026 · 40 citations
- Efficient Diffusion LLMs via Temporal-Spatial Parallel Decoding and Confidence ExtrapolationZekai Li, Ji Liu, Yiqing Huang, Ziqiong Liu et al.ICML 2026 · 1 citation
- Hierarchy Decoding: A Training-free Parallel Decoding Strategy for Diffusion Large Language ModelsXiaojing Qi, Lun Du, Xinyuan Zhang, Lanning Wei et al.ICLR 2026
- FAIR-Calib: Frontier-Aware Instability-Reweighted Calibration for Post-Training Quantization of Diffusion Large Language ModelsHaoyu Huang, Linlin Yang, Sheng Xu, Boyu Liu et al.ICML 2026
