Language Diffusion Models are Associative Memories Capable of Retrieving Unseen Data
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
When do language diffusion models memorize their training data, and how to quantitatively assess their true generative regime? We address these questions by showing that Uniform-based Discrete Diffusion Models (UDDMs) fundamentally behave as Associative Memories (AMs) textitwith emergent creative capabilities. The core idea of an AM is to reliably recover stored data points as textitmemories by establishing distinct basins of attraction around them. Historically, models like Hopfield networks use an explicit energy function to guarantee these stable attractors. We broaden this perspective by leveraging the observation that energy is not strictly necessary, as basins of attraction can also be formed via conditional likelihood maximization. By evaluating token recovery of textittraining and textittest examples, we identify in UDDMs a sharp memorization-to-generalization transition governed by the size of the training dataset: as it increases, basins around training examples shrink and basins around unseen test examples expand, until both later converge to the same level. Crucially, we can detect this transition using only the conditional entropy of predicted token sequences: memorization is characterized by vanishing conditional entropy, while in the generalization regime the conditional entropy of most tokens remains finite. Thus, conditional entropy offers a practical probe for the memorization-to-generalization transition in deployed models.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 40 of its 55 references that are in the map. First posted on arXiv 29 Apr 2026.
Builds on
40 of 55 references.
Closest ancestors · 12
Same direction, newest first
- 2025Memorization to Generalization: Emergence of Diffusion Models from Associative Memory27 May
- 2024Losing dimensions: Geometric memorization in generative diffusion11 Oct
- 2024Detecting, Explaining, and Mitigating Memorization in Diffusion Models31 Jul
- 2024Dynamical Regimes of Diffusion Models28 Feb
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representations4 Oct
- 2023In search of dispersed memories: Generative diffusion models are associative memory networks29 Sep
- 2023Understanding and Mitigating Copying in Diffusion Models31 May
- 2022Scalable Diffusion Models with Transformers19 Dec
- 2021High-Resolution Image Synthesis with Latent Diffusion Models20 Dec
- 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 · 28
- 2017Attention Is All You Needcited by 60,361
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,058
- 2022Large Language Models are Zero-Shot Reasonerscited by 4,374
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
- 2023Extracting Training Data from Diffusion Modelscited by 732
- 2017The Implicit Bias of Gradient Descent on Separable Datacited by 655
- 2021Argmax Flows and Multinomial Diffusion: Learning Categorical Distributionscited by 574
- 2022Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Modelscited by 424
- 2024Discrete Flow Matchingcited by 317
- 2013One Billion Word Benchmark for Measuring Progress in Statistical Language Modelingcited by 308
16 more
- 2024Generative Flows on Discrete State-Spaces: Enabling Multimodal Flows with Applications to Protein Co-Designcited by 305
- 2016Dense Associative Memory for Pattern Recognitioncited by 236
- 2025The Diffusion Dualitycited by 135
- 2020Large Associative Memory Problem in Neurobiology and Machine Learningcited by 112
- 2024An analytic theory of creativity in convolutional diffusion modelscited by 104
- 2023Energy Transformercited by 71
- 2023A Reproducible Extraction of Training Images from Diffusion Modelscited by 61
- 2021Hierarchical Associative Memorycited by 34
- 2026Deriving Neural Scaling Laws from the statistics of natural languagecited by 23
- 2025Modern Methods in Associative Memorycited by 15
- 2021Tractability from overparametrization: The example of the negative perceptroncited by 12
- 2021Supervised perceptron learning vs unsupervised Hebbian unlearning: Approaching optimal memory retrieval in Hopfield-like networkscited by 9
- 2023Neuron-Astrocyte Associative Memorycited by 9
- 2024Random Features Hopfield Networks generalize retrieval to previously unseen examplescited by 8
- 2025Pseudo-likelihood produces associative memories able to generalize, even for asymmetric couplingscited by 2
- 2026The Rules-and-Facts Model for Simultaneous Generalization and Memorization in Neural Networkscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Memory in Plain Sight: Surveying the Uncanny Resemblances of Associative Memories and Diffusion Models
- 2026DreamingGoose: Staged Distillation from Autoregressive Transformers to Bidirectional Recurrent Diffusion Language Models
- 2025Why Diffusion Models Don't Memorize: The Role of Implicit Dynamical Regularization in Training
- 2025Bigger Isn't Always Memorizing: Early Stopping Overparameterized Diffusion Models
- 2026Early Signatures of Memorization in Diffusion Models via Basin Geometry and Cyclic Denoising
- 2026Memorisation, convergence and generalisation in generative models
- 2026Characterizing Memorization in Diffusion Language Models: Generalized Extraction and Sampling Effects
- 2026On the Memorization of Consistency Distillation for Diffusion Models