Your Discrete Graph Diffusion Model is Secretly Memorizing
cs.LG, cs.AI
Submitted: 2026-10-09
Updated: 2026-10-09
Comments: Accepted at NeurIPS 2026 Workshop BeNTo: Beyond Next Token Prediction
License: http://creativecommons.org/licenses/by/4.0/
The gist: Discrete graph diffusion models (GDMs) corrupt the graph structure by adding noise independently to nodes and edges under a Markovian noise model.
Terminology
Abstract
Discrete graph diffusion models (GDMs) corrupt the graph structure by adding noise independently to nodes and edges under a Markovian noise model. In the unconditional setting without node features, an equivariant denoiser trained on a single graph is able to generate novel graphs and sample-level metrics such as edge overlap and WL kernel similarity find no evidence of memorization. In this work, we challenge this view and show for the first time that graph diffusion models indeed exhibit memorization. Analyzing the reverse diffusion trajectories of GDMs trained on two-block stochastic block models (SBMs), we find a strong train-data bias: the model recovers a corrupted training graph almost perfectly compared to a held-out test graph. We term this conditional memorization, as it surfaces only when the reverse trajectory is conditioned on a corrupted training graph and is not apparent during inference. It coincides with a growing generalization gap, where the training loss improves as the loss on held-out graphs degrades, and it cannot be mitigated by early stopping. The onset of memorization always precedes the emergence of community structure in the generated graphs, thus every checkpoint earlier produces structureless graphs. This behavior holds across all node sizes we test.
Sources
- The statistical thermodynamics of generative diffusion models: Phase transitions, symmetry breaking and critical instability
- Extracting Training Data from Diffusion Models
- On the Power of Edge Independent Graph Models
- Implicit Regularisation in Diffusion Models: An Algorithm-Dependent Generalisation Analysis
- Biased Generalization in Diffusion Models
- On Memorization in Diffusion Models
- Denoising Diffusion Probabilistic Models
- Beyond Sample Copying: Structural Memorization in Diffusion Models
- Improving Equivariant Networks with Probabilistic Symmetry Breaking
- Video Diffusion Models: A Survey
- LGDC: Latent Graph Diffusion via Spectrum-Preserving Coarsening
- Categorical Flow Maps
- Diffusion Art or Digital Forgery? Investigating Data Replication in Diffusion Models
- Generative Modeling by Estimating Gradients of the Data Distribution
- DiGress: Discrete Denoising diffusion for graph generation
- A Survey of Diffusion Based Image Generation Models: Issues and Their Solutions
Related papers
- Polynomial-Augmented Neural Networks (PANNs) with Weak Orthogonality Constraints for Enhanced Function and PDE Approximation
- AIRL-S: Unifying Reinforcement Learning and Search-Based Test-Time Scaling via Adversarial Inverse Reinforcement Learning
- Transformers as Bayesian In-Context Experimenters: Smoothness-Adaptive Efficient ATE Estimation
- Convergence issues in Relational Concept Analysis based on AOC-posets
- Beliefs Beyond Posteriors: The Predictive Cost of ELBO Binding
- Understanding Diffusion Models via Ratio-Based Function Approximation with SignReLU Networks