cs.LGSep 28, 2026

Simplex Diffusion Models

Authors: Justin Deschenaux, Alexandre Galashov, Andrew Campbell, Li Kevin Wenliang, James Thornton, Arnaud Doucet, Valentin De Bortoli

Organizations: Google DeepMind · EPFL · UCL Gatsby

Abstract

Diffusion models have revolutionized generative modeling for continuous data through the gradual refinement of a belief state. This iterative refinement has not yet carried over to discrete diffusion models, which discard uncertainty at intermediate steps through categorical sampling (information collapse). We propose Simplex Diffusion Models (SDMs), a framework that lifts the diffusion process to the probability simplex to represent beliefs over categories. SDMs admit probability paths with closed-form reverse transitions and can be trained with a simple cross-entropy loss. Contrary to earlier proposals such as Dirichlet Flow Matching which requires integrating an ordinary differential equation, we introduce a DDIM-like sampler with a tunable level of stochasticity. Because SDMs operate on samples on the simplex, they can carry uncertainty across denoising steps, which mitigates information collapse. On OpenWebText, SDMs are competitive with strong Discrete Diffusion baselines, achieving 17.017.0 GenPPL at 5.465.46 unigram entropy in 64 sampling steps, close to real validation data. Even without Self-Conditioning (SC), SDMs outperform masked and uniform diffusion (with SC or predictor-corrector sampling) on code generation (TinyGSM, T=0.1T=0.1; 49.0%49.0\% vs. 45.8%45.8\%). Distilled down to 8 steps, SDMs solve 32.1%32.1\% of GSM8K problems, more than distilled Discrete Diffusion models with 128 steps (21.4%21.4\%).

Figures & tables

Appendix figures & tables51 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. Simplex Relaxation for Discrete Diffusion

    Aug 11, 2026Jinya Sakurai, Patrick Pynadath, Satoshi Hayakawa +4Diffusion ModelsData Generation

  2. Distribution Matching Distillation for Continuous Diffusion Language Models

    Sep 30, 2026Paul Le Van Kiem, Dario Shariatian, Umut Simsekli +1Diffusion Language ModelsDiffusion Models

  3. BlockGen: Flexible Blockwise Sequence Modeling with Hybrid Samplers

    Jun 1, 2026Justin Deschenaux, Caglar GulcehreDiffusion ModelsDiffusion Language Models