arXiv cs.CL
7/23/2026

Multi-Mask Diffusion Language Models for Few-Step Generation
Short summary
MultiMDM extends masked diffusion models by pushing each clean token toward a designated mask before mixing over the mask set, enabling drafting capability in the backward process. The authors derive a closed-form ELBO supporting continual training from pretrained MDMs and introduce a discrete-state consistency distillation with shared-Gumbel coupling. Experiments show MultiMDM provides an effective foundation for principled few-step generation.
- •MultiMDM preserves masking structure for few-step generation in diffusion LMs
- •Closed-form ELBO supports continual training from pretrained MDMs
- •Discrete-state consistency distillation with shared-Gumbel coupling reduces pathwise entropy
Generated with AI, which can make mistakes.
Is this a good recommendation for you?