Sunwoo Hong

KLASS: KL-Guided Fast Inference in Masked Diffusion Models

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…