Evaluating the Representation Space of Diffusion Models via Self-Supervised Principles
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Diffusion models have demonstrated remarkable generative capabilities and have also emerged as powerful self-supervised representation learners, yet the connection between these two abilities remains less explored. Drawing inspiration from self-supervised learning (SSL), we introduce a framework for jointly evaluating the representation and generation capabilities of diffusion models. Specifically, we decompose features into invariant and residual components and derive the Invariant Contamination Ratio (ICR), a Fisher-based metric that quantifies how residual variation contaminates invariant signal in feature space. We use this framework to analyze both discriminative and generative behavior of diffusion models. On the representation side, we find that invariance peaks at intermediate noise levels, which also yield the best downstream classification performance. On the generative side, we study how training transitions from genuine generalization to memorization in data-limited regimes, and show that ICR serves as a sensitive training-time indicator of early learning: increasing residual energy along Fisher directions marks the onset of memorization, detectable from training features alone without external evaluators or held-out test sets. Overall, our results show that diffusion models can be monitored from a self-supervised perspective through the geometry of their learned representations.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 74 of its 93 references that are in the map. First posted on arXiv 8 Jun 2026.
Builds on
74 of 93 references.
Closest ancestors · 37
Same direction, newest first
- 2026MCLR: Improving Conditional Modeling via Inter-Class Likelihood-Ratio Maximization and Unifying Classifier-Free Guidance with Alignment Objectives23 Mar
- 2025Generalization of Diffusion Models Arises with a Balanced Representation Space24 Dec
- 2025A Closer Look at Model Collapse: From a Generalization-to-Memorization Perspective20 Sep
- 2025Locality in Image Diffusion Models Emerges from Data Statistics11 Sep
- 2025On the Edge of Memorization in Diffusion Models25 Aug
- 2025Memorization to Generalization: Emergence of Diffusion Models from Associative Memory27 May
- 2025Why Diffusion Models Don't Memorize: The Role of Implicit Dynamical Regularization in Training23 May
- 2025Bigger Isn't Always Memorizing: Early Stopping Overparameterized Diffusion Models22 May
- 2025An Analytical Theory of Spectral Bias in the Learning Dynamics of Diffusion Models5 Mar
- 2025Understanding Representation Dynamics of Diffusion Models via Low-Dimensional Modeling9 Feb
- 2025Can Diffusion Models Learn Hidden Inter-Feature Rules Behind Images?7 Feb
- 2025Memorization and Regularization in Generative Diffusion Models27 Jan
25 more
- 2024CleanDIFT: Diffusion Features without Noise4 Dec
- 2024Towards a Mechanistic Explanation of Diffusion Model Generalization28 Nov
- 2024Understanding Generalizability of Diffusion Models Requires Rethinking the Hidden Gaussian Structure31 Oct
- 2024Losing dimensions: Geometric memorization in generative diffusion11 Oct
- 2024Representation Alignment for Generation: Training Diffusion Transformers Is Easier Than You Think9 Oct
- 2024Breaking the Curse of Dimensionality: Diffusion Models Efficiently Learn Low-Dimensional Distributions4 Sep
- 2024Diffusion Models and Representation Learning: A Survey30 Jun
- 2024Dynamical Regimes of Diffusion Models28 Feb
- 2024A Phase Transition in Diffusion Models Reveals the Hierarchical Nature of Data26 Feb
- 2024Deconstructing Denoising Diffusion Models for Self-Supervised Learning25 Jan
- 2024SiT: Exploring Flow and Diffusion-based Generative Models with Scalable Interpolant Transformers16 Jan
- 2023SODA: Bottleneck Diffusion Models for Representation Learning29 Nov
- 2023On the Generalization Properties of Diffusion Models3 Nov
- 2023Generalization in diffusion models arises from geometry-adaptive harmonic representations4 Oct
- 2023In search of dispersed memories: Generative diffusion models are associative memory networks29 Sep
- 2023Diffusion Model as Representation Learner21 Aug
- 2023Diffusion Models Beat GANs on Image Classification17 Jul
- 2023InfoDiffusion: Representation Learning Using Information Maximizing Diffusion Models14 Jun
- 2023Denoising Diffusion Autoencoders are Unified Self-supervised Learners17 Mar
- 2023Diffusion Models Generate Images Like Painters: an Analytical Theory of Outline First, Details Later4 Mar
- 2023Learning Data Representations with Joint Diffusion Models31 Jan
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2021Diffusion Autoencoders: Toward a Meaningful and Decodable Representation30 Nov
- 2020Denoising Diffusion Implicit Models6 Oct
- 2020Denoising Diffusion Probabilistic Models19 Jun
Other references · 37
- 2015U-Net: Convolutional Networks for Biomedical Image Segmentationcited by 19,604
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2023DINOv2: Learning Robust Visual Features without Supervisioncited by 7,764
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,339
- 2015Deep Unsupervised Learning using Nonequilibrium Thermodynamicscited by 7,005
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2014Going Deeper with Convolutionscited by 4,231
- 2021Barlow Twins: Self-Supervised Learning via Redundancy Reductioncited by 1,818
- 2020Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hyperspherecited by 1,393
- 2021VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learningcited by 1,013
25 more
- 2025FLUX.1 Kontext: Flow Matching for In-Context Image Generation and Editing in Latent Spacecited by 960
- 2020Prevalence of Neural Collapse during the terminal phase of deep learning trainingcited by 713
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 712
- 2021Label-Efficient Semantic Segmentation with Diffusion Modelscited by 495
- 2023Emergent Correspondence from Image Diffusioncited by 437
- 2023A Tale of Two Features: Stable Diffusion Complements DINO for Zero-Shot Semantic Correspondencecited by 297
- 2023DragDiffusion: Harnessing Diffusion Models for Interactive Point-based Image Editingcited by 293
- 2021A Geometric Analysis of Neural Collapse with Unconstrained Featurescited by 226
- 2023Diffusion Hyperfeatures: Searching Through Time and Space for Semantic Correspondencecited by 211
- 2023Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion modelscited by 199
- 2023Solving Inverse Problems with Latent Diffusion Models via Hard Data Consistencycited by 180
- 2022A Self-Supervised Descriptor for Image Copy Detectioncited by 175
- 2022RankMe: Assessing the downstream performance of pretrained self-supervised representations by their rankcited by 145
- 2024An analytic theory of creativity in convolutional diffusion modelscited by 103
- 2021On the Role of Neural Collapse in Transfer Learningcited by 100
- 2025What matters for Representation Alignment: Global Information or Spatial Structure?cited by 76
- 2020Why Do Better Loss Functions Lead to Less Transferable Features?cited by 73
- 2021Diffusion-Based Representation Learningcited by 57
- 2023DreamTeacher: Pretraining Image Backbones with Deep Generative Modelscited by 37
- 2023Understanding Deep Representation Learning via Layerwise Feature Compression and Discriminationcited by 31
- 2023LiDAR: Sensing Linear Probing Performance in Joint Embedding SSL Architecturescited by 27
- 2022Understanding and Improving Transfer Learning of Deep Models via Neural Collapsecited by 24
- 2023DiffAug: A Diffuse-and-Denoise Augmentation for Training Robust Classifierscited by 5
- 2026ForcingDAS: Unified and Robust Data Assimilation via Diffusion Forcingcited by 2
- 2026The two clocks and the innovation window: When and how generative models learn rulescited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2024On the Feature Learning in Diffusion Models
- 2021High Fidelity Visualization of What Your Self-Supervised Representation Knows About
- 2025Learning Diffusion Models with Flexible Representation Guidance
- 2026Revisiting Spectral Representations in Generative Diffusion Models
- 2026First Learn, Then Memorize: The Spectral Bias of Diffusion Models
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Models
- 2026Understanding diffusion models requires rethinking (again) generalization
- 2023The Emergence of Reproducibility and Generalizability in Diffusion Models