Large Language Diffusion Models

Nie et al., 2025, Large Language Diffusion Models (Advances in Neural Information Processing Systems 38 (NeurIPS 2025))

Subscriber deck — free during early access

3 cards in “2.1 Probabilistic Formulation” · whole deck

2.1 Probabilistic Formulation

  1. Describe LLaDA’s forward and reverse processes.
    Forward: each token is independently replaced by a mask with probability \(t\), from \(t = 0\) (clean text) to \(t = 1\) (all masked). Reverse: a Transformer mask predictor, with no causal mask, sees the partly masked sequence and predicts all masked tokens at once, step by step from \(t = 1\) back to \(t = 0\).
  2. What is LLaDA’s training loss, and why does it matter that it is a bound?
    Cross-entropy on the masked positions only, weighted by \(1/t\): \(-\mathbb{E}\big[\frac{1}{t}\sum_i \mathbb{1}[x_t^i = \mathrm{M}] \log p_\theta(x_0^i \mid x_t)\big]\). This is an upper bound on the negative log-likelihood, so LLaDA is a principled generative model trained by (approximate) maximum likelihood.
  3. How does LLaDA’s training differ from BERT’s masked language modeling and from MaskGIT?
    BERT masks a fixed fraction (15%) of tokens; LLaDA samples the masking ratio \(t\) uniformly from 0 to 1 for each sequence, so it learns to generate from fully masked text. MaskGIT’s heuristic objective lacks the \(1/t\) weight and therefore the link to likelihood.

Source

  • Shen Nie, Fengqi Zhu, Zebin You, Xiaolu Zhang, Jingyang Ou, Jun Hu, Jun Zhou, Yankai Lin, Ji-Rong Wen and Chongxuan Li. 2025. Large Language Diffusion Models. In Advances in Neural Information Processing Systems 38 (NeurIPS 2025). https://arxiv.org/abs/2502.09992