arXiv · 2603.20155
Beyond Single Tokens: Distilling Discrete Diffusion Models via Discrete MMD
Abstract
It is currently difficult to distill discrete diffusion models. In contrast, continuous diffusion literature has many distillation approaches methods that can reduce sampling steps to a handful. Our method, Discrete Moment Matching Distillation (D-MMD), leverages ideas that have been highly successful in the continuous domain. Whereas previous discrete distillation methods collapse, D-MMD maintains high quality and diversity (given sufficient sampling steps). This is demonstrated on both text and image datasets. Moreover, the newly distilled generators can outperform their teachers.
Explore related subjects
Keep this discovery
Emiel Hoogeboom, David Ruhe, Jonathan Heek, Thomas Mensink, Tim Salimans. 2026-03-20. Beyond Single Tokens: Distilling Discrete Diffusion Models via Discrete MMD. https://arxiv.org/abs/2603.20155
Cite the original work for its findings. Save a collection to share your selection of sources.