Information-Theoretic Discrete Diffusion
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
We present an information-theoretic framework for discrete diffusion models that yields principled estimators of log-likelihood using score-matching losses. Inspired by the I-MMSE identity for the Gaussian setup, we derive analogous results for the discrete setting. Specifically, we introduce the Information-Minimum Denoising Score Entropy (I-MDSE) relation, which links mutual information between data and its diffused version to the minimum denoising score entropy (DSE) loss. We extend this theory to masked diffusion and establish the Information-Minimum Denoising Cross-Entropy (I-MDCE) relation, connecting cross-entropy losses to mutual information in discrete masked processes. These results provide a time-integral decomposition of the log-likelihood of the data in terms of optimal score-based losses, showing that commonly used losses such as DSE and DCE are not merely variational bounds but tight and principled estimators of log-likelihood. The I-MDCE decomposition further enables practical extensions, including time-free formula, conditional likelihood estimation in prompt-response tasks, and coupled Monte Carlo estimation of likelihood ratios. Experiments on synthetic and real-world data confirm the accuracy, variance stability, and utility of our estimators. The code is publicly available at https://github.com/Dongjae0324/infodis.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 33 of its 46 references that are in the map. First posted on arXiv 28 Oct 2025.
Builds on
33 of 46 references.
Closest ancestors · 9
Same direction, newest first
- 2023Interpretable Diffusion via Information Decomposition12 Oct
- 2023Information-Theoretic Diffusion7 Feb
- 2022Score-based Continuous-time Discrete Diffusion Models30 Nov
- 2022A Continuous Time Framework for Discrete Denoising Models30 May
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
- 2021Variational Diffusion Models1 Jul
- 2021Maximum Likelihood Training of Score-Based Diffusion Models22 Jan
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 24
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2024The Llama 3 Herd of Modelscited by 14,192
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2022Photorealistic Text-to-Image Diffusion Models with Deep Language Understandingcited by 6,322
- 2018Think you have Solved Question Answering? Try ARC, the AI2 Reasoning Challengecited by 4,710
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,067
- 2019HellaSwag: Can a Machine Really Finish Your Sentence?cited by 4,018
- 2019PIQA: Reasoning about Physical Commonsense in Natural Languagecited by 2,818
- 2020DiffWave: A Versatile Diffusion Model for Audio Synthesiscited by 1,339
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2025Large Language Diffusion Modelscited by 872
12 more
- 2023BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Datasetcited by 828
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2020WaveGrad: Estimating Gradients for Waveform Generationcited by 620
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 566
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
- 2025LLaDA 1.5: Variance-Reduced Preference Optimization for Large Language Diffusion Modelscited by 273
- 2021Autoregressive Diffusion Modelscited by 208
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 144
- 2013A Deep and Tractable Density Estimatorcited by 67
- 2022Training and Inference on Any-Order Autoregressive Models the Right Waycited by 64
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2026Minimax Optimality of Score-Entropy Discrete Diffusion
- 2025MMG: Mutual Information Estimation via the MMSE Gap in Diffusion
- 2024O(d/T) Convergence Theory for Diffusion Probabilistic Models under Minimal Assumptions
- 2026Mean-to-Score Discrete Diffusion: Posterior-Mean Denoisers for Score Entropy
- 2022From Denoising Diffusions to Denoising Markov Models
- 2026Rethinking the Diffusion Model from a Langevin Perspective
- 2025Information Theoretic Learning for Diffusion Models with Warm Start
- 2025Information-Theoretic Proofs for Diffusion Sampling