Characterizing Memorization in Diffusion Language Models: Generalized Extraction and Sampling Effects
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Autoregressive language models (ARMs) have been shown to memorize and occasionally reproduce training data verbatim, raising concerns about privacy and copyright liability. Diffusion language models (DLMs) have recently emerged as a competitive alternative, yet their memorization behavior remains largely unexplored due to fundamental differences in generation dynamics. To address this gap, we present a systematic theoretical and empirical characterization of memorization in DLMs. We propose a generalized probabilistic extraction framework that unifies prefix-conditioned decoding and diffusion-based generation under arbitrary masking patterns and stochastic sampling trajectories. Theorem 4.3 establishes a monotonic relationship between sampling resolution and memorization: increasing resolution strictly increases the probability of exact training data extraction, implying that autoregressive decoding corresponds to a limiting case of diffusion-based generation by setting the sampling resolution maximal. Extensive experiments across model scales and sampling strategies validate our theoretical predictions. Under aligned prefix-conditioned evaluations, we further demonstrate that DLMs exhibit substantially lower memorization-based leakage of personally identifiable information (PII) compared to ARMs.
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 32 of its 41 references that are in the map. First posted on arXiv 2 Mar 2026.
Builds on
32 of 41 references.
Closest ancestors · 3
Same direction, newest first
- 2021Structured Denoising Diffusion Models in Discrete State-Spaces7 Jul
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 29
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2020Scaling Laws for Neural Language Modelscited by 6,411
- 2022Training Compute-Optimal Large Language Modelscited by 2,927
- 2022Emergent Abilities of Large Language Modelscited by 2,483
- 2020Extracting Training Data from Large Language Modelscited by 2,144
- 2018Hierarchical Neural Story Generationcited by 1,182
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,006
- 2025Large Language Diffusion Modelscited by 872
- 2024Simple and Effective Masked Diffusion Language Modelscited by 800
- 2022Quantifying Memorization Across Neural Language Modelscited by 755
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 606
17 more
- 2022DiffuSeq: Sequence to Sequence Text Generation with Diffusion Modelscited by 439
- 2023The Reversal Curse: LLMs trained on "A is B" fail to learn "B is A"cited by 399
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 371
- 2023Analyzing Leakage of Personally Identifiable Information in Language Modelscited by 278
- 2022Are Large Pre-Trained Language Models Leaking Your Personal Information?cited by 195
- 2024Scaling up Masked Diffusion Models on Textcited by 195
- 2025LLaDA2.0: Scaling Up Diffusion Language Models to 100Bcited by 154
- 2023ProPILE: Probing Privacy Leakage in Large Language Modelscited by 127
- 2023PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Modelcited by 56
- 2024Demystifying Verbatim Memorization in Large Language Modelscited by 55
- 2024Measuring memorization in language models via probabilistic extractioncited by 40
- 2025Extracting memorized pieces of (copyrighted) books from open-weight language modelscited by 27
- 2026Extracting books from production language modelscited by 16
- 2024ROME: Memorization Insights from Text, Logits and Representationcited by 8
- 2025Trustworthy Machine Learning via Memorization and the Granular Long-Tail: A Survey on Interactions, Tradeoffs, and Beyondcited by 6
- 2026Do LLMs Really Memorize Personally Identifiable Information? Revisiting PII Leakage with a Cue-Controlled Memorization Frameworkcited by 3
- 2025Shared Path: Unraveling Memorization in Multilingual LLMs through Language Similaritiescited by 2
Built on by
Citing papers per year: 2026: 3.
3 papers.
Similar, not cited
Close in wording, no citation either way
- 2024Detecting, Explaining, and Mitigating Memorization in Diffusion Models
- 2026Differences in Text Generated by Diffusion and Autoregressive Language Models
- 2025On the Edge of Memorization in Diffusion Models
- 2024Undesirable Memorization in Large Language Models: A Survey
- 2026Diffusion Models Preferentially Memorize Prototypical Examples or: Why Does My Diffusion Model Love Slop?
- 2023On Memorization in Diffusion Models
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models
- 2025A Survey on Diffusion Language Models