2025
KLASS: KL-Guided Fast Inference in Masked Diffusion Models
NeurIPS 2025spotlight
Masked diffusion models have demonstrated competitive results on various tasks including language generation. However, due to its iterative refinement process, the inference is often bottlenecked by slow and static sampling speed. To overcome this problem, we introduce `KL-Adaptive Stability Samplin…