PUMBA Paper Aligns Masked Diffusion LM Training With Inference Trajectories

kastnerkyle · x · 2026-10-07

PUMBA is a new paper addressing a core mismatch in masked diffusion language models: they are trained one way but sampled another. The authors study making MDM training more aware of the trajectories followed at inference time, which leads to more efficient text generation. Paper link in thread.

Original post →

More from Research

Research channel →