RACC: Regret-Aware Confidence Calibration for Consistent Masked Discrete Diffusion Decoding

Qinglin Zeng, Jusheng Zhang, Jing Yang, Ningyuan Liu, Keze Wang


Abstract
Masked Discrete Diffusion Models (MDMs) enable parallel generation via iterative refinement. However, we identify a critical decisional mismatch. The MDM architecture is inherently dynamic and capable of sensing context shifts. In contrast, prevailing decoding paradigms remain static and myopic. They treat each denoising step as an isolated snapshot, effectively discarding valuable temporal feedback that signals logical conflicts. To bridge this gap, we propose Regret-Aware Confidence Calibration (RACC). This training-free framework aligns decoding decisions with the model’s latent self-correction capabilities. RACC introduces a momentum anchor to track confidence trajectories. When a token’s probability drops abruptly below its historical trend, the system triggers a "regret" signal. Unlike expensive re-masking or lookahead search, RACC utilizes this signal to proactively demote unstable candidates. Extensive experiments on reasoning benchmarks, such as HumanEval and GSM8K, demonstrate that RACC significantly improves generation consistency. Crucially, RACC achieves these gains with zero additional inference overhead, effectively balancing decoding quality and efficiency.
Anthology ID:
2026.findings-acl.1138
Volume:
Findings of the Association for Computational Linguistics: ACL 2026
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
Findings
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
22656–22672
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1138/
DOI:
Bibkey:
Cite (ACL):
Qinglin Zeng, Jusheng Zhang, Jing Yang, Ningyuan Liu, and Keze Wang. 2026. RACC: Regret-Aware Confidence Calibration for Consistent Masked Discrete Diffusion Decoding. In Findings of the Association for Computational Linguistics: ACL 2026, pages 22656–22672, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
RACC: Regret-Aware Confidence Calibration for Consistent Masked Discrete Diffusion Decoding (Zeng et al., Findings 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.findings-acl.1138.pdf
Checklist:
 2026.findings-acl.1138.checklist.pdf