Discrete Diffusion Models: A Unified Framework from Tokenization to Generation

📅 2026-07-15
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
Discrete diffusion models currently lack a unified theoretical framework, hindering systematic comparison across existing approaches. This work proposes a cohesive perspective grounded in discrete state-space formulations, integrating diverse methodologies—such as transition matrices, masking/absorbing states, and score/ratio-based formulations—into a common design space. By elucidating the intrinsic connections and trade-offs among training objectives, inference algorithms, and evaluation protocols, the study clarifies the design spectrum of discrete diffusion models. Furthermore, through the unification of multiple diffusion paradigms alongside systematic optimization and scalability analyses, this research establishes a foundation for principled method comparison, guides future investigations, and fosters synergistic development across the field.
📝 Abstract
Discrete denoising diffusion models (DDMs) have recently emerged as a compelling alternative to autoregressive (AR) modeling for discrete data, offering parallel generation and iterative global refinement capabilities. Unlike continuous diffusion, where the state space is fixed, DDMs are fundamentally shaped by how the discrete state space is constructed: the tokenization scheme, the vocabulary topology, and domain-specific structural alphabets. This work introduces a unified conceptual framework that views discrete diffusion models through the construction of the underlying discrete state space. Within this framework, existing formulations, including transition-matrix, masking/absorbing-state, and score/ratio-based approaches, emerge as different instantiations of a common design space. The framework further exposes common design trade-offs across training objectives, inference algorithms, scaling behavior, systems optimization, and evaluation protocols, suggesting several promising directions for future research.
Problem

Research questions and friction points this paper is trying to address.

Discrete Diffusion Models
Tokenization
State Space
Vocabulary Topology
Structural Alphabets
Innovation

Methods, ideas, or system contributions that make the work stand out.

Discrete Diffusion Models
Unified Framework
State Space Construction
Tokenization
Denoising Diffusion
Ye Yuan
Ye Yuan
McGill University, Mila - Quebec AI Institute
Generative ModelingBlack Box OptimizationKnowledge-Centric NLPLLMs
W
Weien Li
McGill University, Mila - Quebec AI Institute
R
Rui Song
McGill University, Mila - Quebec AI Institute
Z
Zeyu Li
McGill University, Mila - Quebec AI Institute
H
Haochen Liu
University of Cambridge
X
Xiangyu Kong
McGill University, Mila - Quebec AI Institute
Zixuan Dong
Zixuan Dong
New York University
Reinforcement LearningDeep LearningNeural Collapse
L
Linfeng Du
McGill University, Mila - Quebec AI Institute
Z
Zipeng Sun
McGill University, Mila - Quebec AI Institute
W
Weixu Zhang
McGill University, Mila - Quebec AI Institute
Jiaxin Huang
Jiaxin Huang
MBUZAI
Machine LearningMedical Image Analysis3D Vision
C
Changjiang Han
MBZUAI - Mohamed bin Zayed University of Artificial Intelligence
Y
Yonghan Yang
MBZUAI - Mohamed bin Zayed University of Artificial Intelligence
Z
Zichen Zhao
MBZUAI - Mohamed bin Zayed University of Artificial Intelligence
Xiuyuan Hu
Xiuyuan Hu
PhD candidate at Tsinghua University
AI for ScienceMachine Learning
Haolun Wu
Haolun Wu
Researcher at Mila, McGill, Stanford | Prev. intern at Google, DeepMind, MSR
Knowledge RepresentationInformation RetrievalHuman-centric AI
Yankai Chen
Yankai Chen
Postdoctoral Associate, Cornell University
Information RetrievalKnowledge MiningLarge Language ModelsAgentic AI
Fengran Mo
Fengran Mo
Ph.D. Student, Université de Montréal
Conversational AIInformation RetrievalNatural Language ProcessingMultilingualism
Jikun Kang
Jikun Kang
LMTS at Salesforce
Machine LeanringReinforcement Learning
Bowei He
Bowei He
City University of Hong Kong, MBZUAI
Data MiningLanguage ModelGenAI4ScienceAgentic AI
Philip S. Yu
Philip S. Yu
Professor of Computer Science, University of Illinons at Chicago
Data miningDatabasePrivacy
X
Xue Liu
MBZUAI - Mohamed bin Zayed University of Artificial Intelligence, McGill University, Mila - Quebec AI Institute