AI RESEARCH

Fine-Tuning Masked Diffusion for Provable Self-Correction

arXiv CS.LG

ArXi:2510.01384v4 Announce Type: replace A natural desideratum for generative models is self-correction--detecting and revising low-quality tokens at inference. While Masked Diffusion Models (MDMs) have emerged as a promising approach for generative modeling in discrete spaces, their capacity for self-correction remains poorly understood. Prior attempts to incorporate self-correction into MDMs either require overhauling MDM architectures/