AI RESEARCH

Uniform Diffusion Models Revisited: Leave-One-Out Denoiser and Absorbing State Reformulation

arXiv CS.LG

ArXi:2605.22765v1 Announce Type: new Discrete diffusion models are often trained through clean-data prediction, but the prediction can be used in different ways to define the reverse dynamics. In Masked Diffusion Models (MDM) these choices largely coincide, whereas in Uniform Diffusion Models (UDM) they do not. We show that the standard plug-in bridge parameterization for UDM is not optimized by the denoising posterior, but by a leave-one-out posterior that predicts each clean token without using its own noisy observation.