Diffusion Models: Associative Memory with Creative Spark

The opaque nature of data memorization in large language models presents a critical challenge for understanding their true generative capabilities. This research probes the inner workings of Uniform-based Discrete Diffusion Models (UDDMs), revealing a fundamental link to Associative Memory (AM) principles. According to the authors' findings on arXiv, UDDMs inherently store and retrieve training data, mirroring AMs that use basins of attraction to reliably recover memories.

Emergent Creativity from Associative Recall

Unlike traditional AMs like Hopfield networks that explicitly define an energy function, UDDMs achieve stable attractors through conditional likelihood maximization. This perspective broadens our understanding of how generative models can both recall specific data points and exhibit emergent creative capabilities. The research posits that energy functions are not strictly necessary for stable memory recall.

The Dataset Size Threshold for Generalization

A key breakthrough is the identification of a sharp transition from memorization to generalization in UDDMs, directly governed by training dataset size. As the dataset grows, the basins of attraction around training examples shrink, while those for unseen test examples expand. Eventually, these basins converge, indicating a shift towards generalization. This dynamic offers a quantitative lens on how language diffusion models memorization is managed and overcome.

Conditional Entropy: A Practical Memorization Probe

The researchers propose conditional entropy as a practical and effective metric for detecting this memorization-to-generalization transition. Vanishing conditional entropy signals strong memorization, whereas finite conditional entropy across most tokens indicates the model operates in a generalization regime. This offers a deployable method for assessing the true generative behavior of UDDMs and understanding language diffusion models memorization in real-world applications.