When Confidence Misleads: Suffix Anchoring and Anchor-Proximity Confidence Modulation for Diffusion Language Models
Abstract
Researchers investigate how confidence-based decoding in fully non-autoregressive models can be improved by addressing issues with EOT tokens and premature decoding through suffix-anchored confidence modulation.
Diffusion language models decode text by iteratively denoising masked token sequences, making the choice of which positions to decode a central inference-time decision. Most training-free decoding strategies use model confidence for position selection, assuming that high-confidence positions are ready to be decoded. In this work, we revisit this assumption by studying when confidence misleads fully non-autoregressive (fully non-AR) decoding. EOT tokens can receive high confidence and cause incomplete generation; inserting a suffix anchor can mitigate this issue but introduces local overconfidence near the anchor, causing anchor-adjacent tokens to be decoded too early. To address these issues, we propose Suffix-Anchored Confidence Modulation, a simple training-free method that inserts a short suffix anchor to encourage response completion and modulates confidence near the anchor according to decoding progress. This preserves the response-completion benefit of suffix anchoring while reducing premature decoding of anchor-adjacent tokens. Across text-only reasoning, vision-language reasoning, and code-generation benchmarks, our method consistently improves confidence-based fully non-AR decoding, outperforms explicit EOT suppression, and preserves the parallel decoding advantage of fully non-AR generation.
Community
This paper studies when confidence-based decoding can mislead fully non-autoregressive diffusion language models. One known failure case is EOT overconfidence, where the model assigns high confidence to end-of-text tokens too early, leading to extremely short or incomplete generations. The paper shows that inserting a short suffix anchor near the end of the response region can encourage the model to generate complete, meaningful content. However, this anchor can also make nearby positions overconfident, causing them to be decoded prematurely. To address this issue, the paper proposes Suffix-Anchored Confidence Modulation, a training-free decoding method that preserves the response-completion benefit of suffix anchoring while mitigating premature decoding near the anchor.
This is an automated message from the Librarian Bot. I found the following papers similar to this paper.
The following papers were recommended by the Semantic Scholar API
- Cluster-Level Attention-Guided Parallel Decoding for Masked Diffusion Language Models (2026)
- Stability-Weighted Decoding for Diffusion Language Models (2026)
- Breaking Block Boundaries: Anchor-based History-stable Decoding for Diffusion Large Language Models (2026)
- PSD: Pushing the Pareto Frontier of Diffusion LLMs via Parallel Speculative Decoding (2026)
- Early Decisions Matter: Proximity Bias and Initial Trajectory Shaping in Non-Autoregressive Diffusion Language Models (2026)
- Visual-Redundancy-Controlled Parallel Decoding for Diffusion-Based Multimodal Large Language Models (2026)
- $R^2$-dLLM: Accelerating Diffusion Large Language Models via Spatio-Temporal Redundancy Reduction (2026)
Please give a thumbs up to this comment if you found it helpful!
If you want recommendations for any Paper on Hugging Face checkout this Space
You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend
Get this paper in your agent:
hf papers read 2605.28181 Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash Models citing this paper 0
No model linking this paper
Datasets citing this paper 0
No dataset linking this paper
Spaces citing this paper 0
No Space linking this paper
Collections including this paper 0
No Collection including this paper