ConceptioArchivearXiv CS
arXiv CSopen access

Evaluating the Representation Space of Diffusion Models via Self-Supervised Principles

Unknown · 2026 · arxiv_cs
arXiv CS · Papers · License: Open Access · 2026
Open Source ↗Direct PDF ↓
machine learning, deep learning, neural networks

Evaluating the Representation Space of Diffusion Models via SelfSupervised Principles Xiao Li† , Yixuan Jia† , Zekai Zhang, Xiang Li, Lianghe Shi, Jinxin Zhou1 , Zhihui Zhu1 , Liyue Shen, Qing Qu‡ ·

University of Michigan † Joint first author

1 Ohio State University

‡ Corresponding author

arXiv:2606.09718v1 [cs.LG] 8 Jun 2026

Abstract 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. Keywords: Diffusion model, Representation learning, Self-supervised learning Date: June 9, 2026 Correspondence: [email protected] Resources: Project Website Training data sample augment+noise 𝑎(𝒙! ) 𝒙! ~𝑝"#$# Invariant

𝒔 = 𝔼! [𝒉|𝒙" ]

https://deepthink-umich.github.io

𝜮% ≔ Cov𝒙! (𝒔) 𝜮' ≔ Cov𝒙!,) 𝝃 𝜮% 𝒗* = 𝜆* 𝜮' 𝒗* .

Residual 𝝃=𝒉−𝒔

Low ICR Views cluster tightly, means spread far

High ICR Views scatter widely, means bunch together

Invariant mean features 𝒔

Augmented view features 𝝃

Fisher SNR eigenvalues 𝜆" ≥ 𝜆# ≥ ... ≥ 𝜆$ ICR Score

𝐈𝐂𝐑 = (a)

: Diffusion feature extraction: 𝒉 𝑎 𝒙! ICR tracks representation quality across noise levels

= 𝒔 𝒙! + 𝝃(𝑎, 𝒙! )

1 1 1 + ∑* 𝜆* 𝑑

(b)

ICR mirrors generation quality (FID) in data-rich regime

ICR predicts memorization in data-limited regime

(c)

Figure 1: Overview of the ICR Framework. Each training image is augmented and passed through the diffusion

feature extractor, decomposing representations into an invariant component 𝒔 and a residual 𝝃; their covariances define ICR (a–b). ICR serves as a unified diagnostic: it identifies the optimal noise level for classification tasks, tracks generative quality without sampling, and anticipates memorization onset during training (c).

1

Contents 1

Introduction

3

2

Problem Setup

5

3

Representation-Level Evaluation Based on Self-Supervised Principles 3.1 What Makes a Good Representation in SSL? . . . . . . . . . . . . . . . . . . . . . . . . 3.2 A Geometric Framework for Representation Evaluation . . . . . . . . . . . . . . . . .

5 6 6

4

ICR across Noise Levels: a Semantic Window for Representation Learning

8

5

Invariance and Expansion: From Generalization to Memorization 9 5.1 ICR Tracks FID in the Data-Rich Regime . . . . . . . . . . . . . . . . . . . . . . . . . . 10 5.2 Early Learning in the Representation Space Under Limited Data . . . . . . . . . . . . 11 5.3 How Feature Expansion Differs in Data-Rich and Data-Limited Regimes . . . . . . . 12

6

Conclusion

13

A Related Work

19

B Additional Discussions & Experiments B.1 Component Dynamics Across the Noise Schedule . . . . . . . . . . . . . . . . . . . . B.2 ICR and Classification Accuracy in the Data-limited Regime . . . . . . . . . . . . . . B.3 Discussion on the Alignment and Uniformity Metrics [45] . . . . . . . . . . . . . . . B.4 Discussion on the Class Separation and Silhouette Score Metrics . . . . . . . . . . . . B.5 Connection to Neural Collapse . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . . B.6 Technical Details on Calculating ICR . . . . . . . . . . . . . . . . . . . . . . . . . . . .

20 20 21 21 22 24 25

C Experimental Details

27

D Alignment between Optimal Test Loss and ICR

29

2

1 Introduction In recent years, diffusion models [1, 2, 3] have achieved remarkable success in generative modeling, serving as the backbone for inverse problems [4, 5, 6, 7] and many prominent large-scale generative systems, such as Stable Diffusion, Flux, and Veo [2, 8, 9, 10, 11]. Beyond their generative capabilities, recent studies [12, 13, 14, 15, 16] have demonstrated the superior unsupervised representation learning abilities of diffusion models, where the diffusion representation is extracted from the bottleneck layer at certain timesteps of the learned denoiser. These works leverage diffusion representations for various downstream tasks, including classification, segmentation, and image correspondence, often achieving performance comparable to or even exceeding established self-supervised learning (SSL) methods. In parallel, regularizing diffusion representations using powerful self-supervised learners, such as DINOv2 [17] and MAE [18], can significantly improve the training efficiency and generation quality of diffusion models [19, 20]. This highlights the strong interplay between representation learning and generative modeling within these paradigms. Despite these advances, the representation learning paradigms of diffusion models and traditional SSL remain quite different. Diffusion models are trained with a denoising objective, recovering clean signals from Gaussian corrupted inputs, whereas most SSL methods [21, 22, 23, 24, 17] are explicitly designed to enforce invariance to data augmentations while preserving a rich, diverse embedding space. The distinction between training objectives raises natural questions about diffusion representation spaces: to what extent do they implicitly capture the beneficial characteristics directly optimized in SSL, and how do these properties evolve across varying noise levels and throughout the learning dynamics? Bridging diffusion representations with those studied in standard SSL can help clarify how diffusion training shapes features and how these features might be further improved to benefit diffusion models. Moreover, adopting a representation-centered viewpoint of diffusion models offers an intrinsic way to understand these systems: if diffusion models are indeed powerful representation learners, the properties of their learned representations should encode clear signatures of whether the model is capturing low-dimensional structure in image manifolds [25, 26] rather than merely overfitting to the idiosyncratic details of the training data. Moreover, this perspective is particularly helpful for understanding and monitoring the generalizability of diffusion models during training. Evaluating how well a model goes beyond memorizing the training set is practically difficult, especially in data-limited regimes where recent work has documented a distinct “early learning” phase [27, 28, 29, 30, 31]: the model first learns to generalize and then gradually starts to memorize individual samples. In this regard, prior work [32] showed that standard generation metrics such as Fréchet distance (FID) are not reliable memorization detectors, while exhaustive nearest neighbor tests [33, 34] rely on large numbers of generated samples and are often expensive to run. By importing insights from SSL and evaluating diffusion models through the geometry of their learned representations, we obtain intrinsic, training time signals that track the quality of the representation space. These signals offer a practical way to gauge generalized generation quality as training progresses and to identify good stopping points before overfitting dominates. Summary of contributions. In this work we revisit diffusion models from a self-supervised representation learning perspective. Guided by classic SSL principles, we focus on two properties of internal features: representation invariance, reflecting the stability of shared content across random perturbations, and representation expansion, reflecting how well the features spread out in the available embedding space. We capture these two properties with a new intrinsic metric, the Invariant Contamination Ratio (ICR), which measures how much augmentation and noise sensitive variation 3

Figure 2: Nearest neighbors of invariant and residual components on ImageNet 64 × 64. We use a pretrained EDM diffusion model and, for each ImageNet training image, sample 9 augmented views and extract bottleneck representations. Left: t-SNE visualization of invariant representations 𝒔; views of the same base image form tight clusters, and the marked nearest and farthest examples reflect their relative positions in this space. Top right: For a reference image (left), nearest and farthest neighbors among the base images are selected using cosine similarity on 𝒔. Bottom right: For a reference augmented view (left), nearest and farthest neighbors across all augmented views are selected using cosine similarity on 𝝃. Neighbors based on 𝒔 tend to be semantically similar to the query, whereas neighbors based on 𝝃 often appear semantically unrelated. contaminates the stable part of the representation space. To construct ICR, we introduce an invariance–residual decomposition of diffusion features that separates perturbation invariant structure from residual variation. Because ICR is label-free and can be computed entirely from training features, it can be monitored throughout training and across noise levels. Empirically, we find that ICR provides a reliable proxy for noise level dependent downstream representation performance and cleanly separates generalization from memorization during training in data-limited regimes. In summary, our main contributions are as follows: • A representation-based evaluation metric for diffusion models. We introduce an invariance–residual decomposition of diffusion representations and from it define the Invariant Contamination Ratio (ICR), a single label-free scalar that measures how much of the representation space is occupied by augmentation and noise-sensitive variation rather than stable structure. ICR can be computed solely from training features without labels or external networks. • Finding the optimal representation across noise levels. On standard image benchmarks, we show that the diffusion noise schedule admits an intermediate semantic window where ICR is minimized and linear classification accuracy is maximized. This gives a simple SSL-based rule to select noise scales that yield the strongest diffusion features for downstream tasks. • Tracking generalization and memorization in training dynamics. By following ICR over training, we observe distinct learning phases: in data-rich regimes it decreases steadily with improving generative quality, while in data-limited regimes it exhibits a U-shaped early learning pattern that precedes the rise of memorization. Thus ICR provides a practical, label-free early stopping signal, and its decomposition reveals that feature expansion during training is driven by invariant structure when data are abundant and by residual variation when data are scarce.

4

2 Problem Setup Preliminaries on diffusion models. Diffusion models define a forward process that gradually perturbs data 𝒙0 ∼ 𝑝data toward a Gaussian distribution via the stochastic differential equation d𝒙 𝑡 = 𝑓 (𝑡)𝒙 𝑡 d𝑡 + 𝑔(𝑡) d𝒘 𝑡 ,

𝑡 ∈ [0, 1],

where 𝑓 and 𝑔 are scalar functions and {𝒘 𝑡 } is a standard Wiener process. Let 𝑝 𝑡 denote the density of 𝒙 𝑡 and note that 𝑝 0 = 𝑝data . For simplicity, we consider the variance preserving setting 𝒙 𝑡 = 𝒙0 + 𝜎𝑡 𝝐 with 𝝐 ∼ 𝒩 (0, 𝑰). The reverse time process that maps noise back to data uses the score ∇ log 𝑝 𝑡 (𝒙 𝑡 ) and is given by the reverse SDE [35] d𝒙 𝑡 = 𝑓 (𝑡)𝒙 𝑡 − 𝑔 2 (𝑡)∇ log 𝑝 𝑡 (𝒙 𝑡 ) d𝑡 + 𝑔(𝑡) d𝒘 ¯ 𝑡,



where {𝒘 ¯ 𝑡 } is an independent Wiener process. This enables diffusion models to generate new samples from the underlying data distribution 𝑝 data by initializing from pure Gaussian noise and iteratively denoising via the score function. Training loss of diffusion models. Modern diffusion models are typically trained to approximate the score function ∇ log 𝑝 𝑡 (𝒙 𝑡 ). By Tweedie’s formula [36],

E [𝒙0 | 𝒙 𝑡 ] = 𝒙 𝑡 + 𝜎𝑡2 ∇ log 𝑝 𝑡 (𝒙 𝑡 ),

(1)

this is equivalent to learning the posterior mean E[𝒙0 | 𝒙 𝑡 ] via a denoising autoencoder 𝒙 𝜽 (𝒙 𝑡 , 𝑡) [15, 13, 37]. Concretely, we minimize the weighted denoising loss min 𝜽

𝑁 ∫ 1 Õ 𝑖=1

0

𝜆𝑡 E𝝐



(𝑖) (𝑖) 𝒙 𝜽 (𝒙 𝑡 , 𝑡) − 𝒙 0

2

 d𝑡,

(2)

(𝑖) 𝑖.𝑖.𝑑.

where 𝒙0

∼ 𝑝 data for 𝑖 = 1, . . . , 𝑁 and 𝜆𝑡 weights different noise levels.

Layer selection for extracting representations. We freeze the diffusion backbone and extract representations from the layer that gives the strongest downstream performance according to [13]. In practice, this corresponds to a layer near the bottleneck of the U-Net architecture [38, 39] and the middle transformer block of SiT [40].

3 Representation-Level Evaluation Based on Self-Supervised Principles As discussed in the introduction, recent work shows that diffusion models can act as strong self supervised representation learners, supporting competitive performance across various downstream tasks [12, 13, 15, 41]. However, their training paradigm differs markedly from standard self-supervised learning (SSL): while most SSL methods rely on explicit contrastive or predictive objectives to shape the embedding space [22, 42, 43, 17], diffusion models are trained with a denoising objective that reconstructs clean signals from Gaussian corrupted inputs. This difference raises a fundamental question: To what extent does the denoising objective naturally satisfy the geometric properties of "good" representations sought in the SSL literature?

5

In diffusion models, the internal representations are high-dimensional and evolve across different noise scales, making it non-trivial to separate stable information from idiosyncratic variation. To evaluate this, we propose an evaluation metric rooted in the principles of modern image-based SSL.

3.1 What Makes a Good Representation in SSL? Modern SSL methods [44, 45, 22, 21, 24, 17] are often built around two complementary principles: • Representation invariance: Representations extracted from different stochastic perturbations of a sample are encouraged to remain stable in the embedding space. • Representation expansion: Representations should maintain a rich, spread-out structure across different images. The embedding distribution should avoid dimensional collapse and utilize many directions in the representation space to preserve unique image identities. In the rest of this work, we use these principles to track how diffusion representations evolve across noise levels and training. We introduce a simple decomposition that splits each representation into a perturbation invariant component, stable across noisy and augmented views, and a residual component that captures variation induced by these perturbations.

3.2 A Geometric Framework for Representation Evaluation Guided by the SSL principles in Section 3.1, we seek a representation-level metric that can be monitored across noise levels and training, and that reflects how much of the active representation space is devoted to perturbation invariant structure. Informally, we want this diagnostic to (i) measure relative invariance rather than an absolute distance scale, (ii) be robust to overall representation expansion during training, and (iii) remain label-free and efficiently computable from representations. A natural starting point is the Alignment and Uniformity criteria [45], which have been highly successful in characterizing contrastive SSL encoders. However, Alignment is an absolute squared distance between two augmented views of the same image and grows when representations take more directions, so it can increase even when the representation becomes more semantically stable, while Uniformity only measures how spread out representations are and does not distinguish invariant structure from augmentation-sensitive noise. (see Appendix B.3 for details). These limitations motivate a different construction that explicitly separates invariant information from view-specific variation. Instead of working directly with distances between raw representations, we first decompose the representation space into stable and varying components, then leverage the spectral properties of these components to derive a summary metric for representations. Invariant and residual decomposition in representation space. For each training image 𝒙 0 ∼ 𝑝 data , let 𝑎 ∼ 𝒜 denote a random perturbation encompassing both standard semantics-preserving transformations [22, 42] and the additive Gaussian noise 𝜖 ∼ 𝒩 (0, 𝜎𝑡2 𝑰) injected by the diffusion objective1 . Let 𝒉(·) ∈ R𝑑 be the representation extracted from a fixed layer of the diffusion model, and consider the random representation 𝒉(𝑎(𝒙 0 )) induced by the stochasticity of the perturbation 𝑎. We decompose this representation into its conditional mean and a residual: 𝒔(𝒙0 ) B E𝑎 𝒉(𝑎(𝒙 0 )) | 𝒙0 ,





𝝃(𝑎, 𝒙 0 ) B 𝒉(𝑎(𝒙 0 )) − 𝒔(𝒙0 ), 1 Specifically, we first apply augmentations to 𝒙

(3)

0 and then add Gaussian noise to the augmented view to obtain 𝑎(𝒙 0 ).

6

ICR

0.7

70

0.6

60

0.5

50

ICR Classification Acc.

0.4 0.0

0.02

0.14

0.59

Time step ( t)

(a) CIFAR10

0.8

60

0.7

50

0.6

40

0.5

ICR Classification Acc.

0.4

40 1.92

30

0.0

0.02

0.14

0.59

Time step ( t)

20 1.92

ICR Classification Acc.

0.9 0.8

Feature Test Acc. (%)

80

Feature Test Acc. (%) ICR

Feature Test Acc. (%) ICR

90

0.8

12 10

0.7

8

0.6

6 4

0.5

2

0.4 0.06

(b) CIFAR100

0.15

0.25

0.35

0.5

Time step ( t)

0.7

0.9

0

(c) ImageNet

Figure 3: Correspondence between ICR and classification accuracy across noise levels. For each pretrained backbone (EDM [39] on CIFAR10 and CIFAR100, SiT-XL/2 [40] on ImageNet), we extract bottleneck representations at multiple noise levels 𝜎𝑡 . At each 𝜎𝑡 , we estimate ICR (blue) using a subset of training representations and train a classifier on the full training representations, reporting accuracy on the test set (slate). Across datasets, the noise levels that minimize ICR coincide with those that maximize classification accuracy. yielding the additive form 𝒉(𝑎(𝒙0 )) = 𝒔(𝒙0 ) + 𝝃(𝑎, 𝒙0 ). In this decomposition, 𝒔(𝒙0 ) is the invariant component, which filters out transient variations to capture attributes resilient to corruption. Conversely, 𝝃(𝑎, 𝒙 0 ) is the residual component capturing the specific idiosyncratic variations of a single noisy view. Figure 2 provides empirical support for this interpretation: multiple perturbed views of a single image form a cluster in the representation space centered at 𝒔(𝒙0 ). Notably, nearest-neighbor searches based on 𝒔(𝒙0 ) retrieve semantically related images, whereas neighbors based solely on the residual 𝝃(𝑎, 𝒙 0 ) appear visually unrelated and lack shared category structure. Since E[𝝃 | 𝒙0 ] = 0, the law of total covariance implies that the total representation covariance 𝚺 ℎ decomposes into two components: 𝚺 ℎ = 𝚺 𝑠 + 𝚺𝜉 , where 𝚺𝑠 B Cov𝒙0 (𝒔),

𝚺𝜉 B Cov𝒙0 ,𝑎 (𝝃).

This decomposition allows us to translate core SSL principles into geometric properties of the representation space: • Representation expansion refers to the growth of the total covariance 𝚺 ℎ = 𝚺𝑠 + 𝚺𝜉 , which characterizes how features spread in the representation space across data samples. More specifically, it describes the extent to which the feature covariance occupies multiple directions, i.e., whether the representation is low-rank or broadly distributed in the embedding space. In practice, we track its trace, Tr(𝚺 ℎ ), which measures the total variance (energy) of the representation and reflects how much of the high-dimensional feature space the model utilizes for distinguishing between different images and their perturbed views. • Representation invariance is measured by the relative dominance of 𝚺𝑠 over 𝚺𝜉 . Unlike traditional SSL, diffusion models require a noise-dependent balance: at low noise levels, a larger residual 𝚺𝜉 is necessary for reconstructing fine-grained pixel details, whereas at intermediate noise levels, a higher degree of invariance is desired to capture stable semantic structures.

7

Fisher directions and invariant signal-to-noise ratios. Given the invariant and residual covariances (𝚺𝑠 , 𝚺𝜉 ), we consider the generalized eigenproblem [46]: 𝚺 𝑠 𝒗 𝑖 = 𝜆 𝑖 𝚺𝜉 𝒗 𝑖 ,

𝜆1 ≥ 𝜆2 ≥ · · · ≥ 𝜆 𝑑 ≥ 0,

with eigenvectors 𝒗 𝑖 orthogonal in the 𝚺𝜉 inner product. In practice, these matrices are estimated empirically across the training dataset, where 𝚺𝜉 aggregates residual variations within each sample and 𝚺𝑠 captures the spread of invariant identities across all samples. Each 𝒗 𝑖 represents a direction in representation space that is optimized to maximize the ratio of invariant signal energy to residual variation. Specifically, the corresponding eigenvalue 𝜆 𝑖 admits the Rayleigh quotient [47] representation2 : 𝜆𝑖 =

𝒗 ⊤ 𝚺𝑠 𝒗 . 𝒗≠0, 𝒗⊥𝚺𝜉 {𝒗1 ,...,𝒗 𝑖−1 } 𝒗 ⊤ 𝚺𝜉 𝒗 max

(4)

In this framework, a generalized eigenvalue 𝜆 𝑖 measures the invariant signal-to-noise ratio along the corresponding Fisher direction [48] 𝒗 𝑖 : it compares the variance of the stable invariant component 𝒗 ⊤ 𝚺𝑠 𝒗 to the residual variance 𝒗 ⊤ 𝚺𝜉 𝒗 in that specific direction. This follows the same generalized eigenstructure as classical Fisher Linear Discriminant Analysis [48], where 𝚺𝑠 and 𝚺𝜉 play roles analogous to between-class and within-class covariances, respectively, with each individual image effectively acting as its own class. Invariant Contamination Ratio (ICR). The collection of generalized eigenvalues {𝜆1 , . . . , 𝜆 𝑑 } provides a directional profile of how strongly invariant structures dominate residual variations. To summarize this into a single, trackable scalar that quantifies the health of the representation, we define the Invariant Contamination Ratio (ICR): ICR B

1

Í 1 + 1𝑑 𝑑𝑖=1 𝜆 𝑖

.

(5)

The term 1𝑑 𝜆 𝑖 represents the average invariant signal-to-noise ratio across all Fisher directions. When the invariant component 𝒔 dominates the residual 𝝃 across the majority of directions, this average is large, resulting in a low ICR. Conversely, as residual variation (or “contamination”) increases and begins to occupy a substantial portion of the representation space, the ICR approaches 1.3 We note that in practice, we estimate 𝚺𝑠 and 𝚺𝜉 from as few as two augmentations per image and a subset of training representations; implementation details are given in Section B.6.

Í

4 ICR across Noise Levels: a Semantic Window for Representation Learning The denoising objective of diffusion models is inherently multiscale: for each clean image 𝒙 0 the model sees a family of corrupted inputs {𝒙 𝑡 } indexed by the noise level 𝜎𝑡 , and thus induces a family of representations 𝒉 𝑡 (𝑎(𝒙 0 )). In this section we use ICR to answer two questions: • How does relative invariance vary across the diffusion noise schedule? • Can this internal measure predict which noise levels yield the best downstream representations? 2We assume 𝚺

𝜉 ≻ 0; in practice we can add a small 𝜏 > 0 and replace 𝚺𝜉 by 𝚺𝜉 + 𝜏𝑰 to ensure invertibility. 3 This choice matches the convention of generative metrics like FID, where lower values indicate a “cleaner” and more

robust representation.

8

30

ICR FID Score

0.38

0.46

20

0.34

ICR

FID Score

ICR

0.36

15

0.32 0.30

2.5m

7.5m

15m

25m

50m

Training Iter. (imgs)

0.42

100m

(a) CIFAR10, EDM, 𝑁 = 50K

160 120 100 80

0.43

5

0.26

180 140

0.45 0.44

10

0.28

ICR FID Score

0.47

25

FID Score

0.40

60 40 25k

75k

125k

175k

225k

Training Iter. (batches)

275k

(b) ImageNet-256, SiT-B/2, 𝑁 = 1.28M

Figure 4: ICR and FID dynamics in data rich diffusion training. We monitor generative performance (via FID) and ICR for EDM and SiT-B/2 based diffusion models trained on the full CIFAR10 and ImageNet datasets as training progresses. Both ICR (blue) and FID (brown) exhibit a monotonically decreasing trend, indicating improving internal representation invariance and sample quality over the course of training. ICR predicts classification performance and reveals a semantic window. We start from pretrained diffusion backbones on CIFAR datasets [49] and ImageNet [50]. For each noise level 𝜎𝑡 , we estimate ICR(𝜎𝑡 ) using a subset of training representations extracted from inputs corrupted at noise level 𝜎𝑡 , and train a linear classifier on the full training representations at the same 𝜎𝑡 . Figure 3 plots ICR and test accuracy as functions of 𝜎𝑡 . Across all datasets we observe a striking alignment: the ICR curve is U-shaped and attains a clear minimum at an intermediate noise level, while classification accuracy peaks in exactly the same range. We refer to this range as a semantic window: at very low noise, representations are too tied to fine-grained, augmentation-specific details; at very high noise, representations collapse toward noise; in between, relative invariance is strongest and the model uses its representation space in a way that is most useful for downstream tasks. Importantly, ICR is computed in a label-free way from training representations alone, yet it accurately predicts which noise scales will deliver the best linear probe performance. We further verify the robustness of this observation in the data-limited setting in Section B.2 and observe a consistent trend. These observations are consistent with prior work [51, 26] showing that diffusion sampling exhibits a coarse-to-fine transition, with different noise levels capturing structure at different granularities. They also suggest that our augmentation-based evaluation developed from self-supervised principles extend naturally to diffusion models, providing a simple bridge between these two frameworks. Motivated by this semantic window, in the next section we fix a representative intermediate noise level 𝜎★ near the ICR minimum and study how the representation space evolves over the course of training at this scale, relating the dynamics of invariance to generative quality and memorization.

5 Invariance and Expansion: From Generalization to Memorization In the previous section we examined how ICR varies across the diffusion noise schedule and identified an intermediate semantic window where downstream representations are strongest. We now fix a representative noise scale 𝜎★ in this window and track how the representation space evolves over the course of training. Our analysis proceeds in three steps. First, in the data-rich regime, we show that ICR closely tracks FID during training, linking improvements in generative quality to changes in the internal 9

1st row: Generated Images 2nd row: Nearest Neighbor in Training set

CIFAR10 (training set size: 4,096 images)

Training Iter = 2.5M

Training Iter = 8.5M

Training Iter = 200M

Figure 5: ICR as an early signal of memorization in data limited diffusion training (CIFAR10). We evaluate an EDM-based diffusion model trained on a subset of CIFAR10 (4096 images). Left: ICR (blue) follows a clear U shaped trajectory as training progresses, while the memorization ratio (red) remains near zero early on and begins to rise only after the ICR minimum. Right: Qualitative inspection at 2.5M, 8.5M, and 200M training images seen. Generated samples (top) and their nearest training neighbors (bottom) show that visual quality initially improves, but eventually the model begins to memorize individual training images, in line with the ICR curve. feature geometry. Second, in the data-limited regime, we observe an early learning phenomenon in the representation space: ICR exhibits a clear U-shaped trajectory and its minimum aligns with the onset of memorization. Finally, we examine trace-level statistics of the invariant and residual covariances to understand how feature expansion is allocated between invariant and residual components in the two regimes.

5.1 ICR Tracks FID in the Data-Rich Regime We first consider the data-rich regime, where the diffusion model is trained on the full training set. At each checkpoint we compute ICR from training features and evaluate the Fréchet Inception Distance (FID) between generated samples and the real data distribution. Figure 4 illustrates the resulting trajectories as functions of training progress. In this setting, ICR and FID exhibit a strong positive correlation: both demonstrate a monotonically decreasing trend. This alignment reflects an intuitive connection between generative quality and representation geometry. FID [52] (and related Fréchet-based metrics such as FDDINOv2 ) measures the distance between generated and real distributions in an external semantic feature space [53, 17], quantifying how well the model captures the data manifold. In contrast, ICR quantifies the “purity" of the internal diffusion representation by measuring the ratio of augmentation-sensitive residual energy to stable invariant signal. The simultaneous improvement in FID and ICR suggests that the improvement in generative ability is directly reflected in the refinement of the representation space. As the model better approximates 𝑝 data , its internal features shift from capturing transient, view-specific noise toward a stable, low-dimensional image structure. In our framework, this manifests as a higher signal-to-noise ratio in the Fisher directions, where the invariant component 𝚺𝑠 increasingly dominates the residual variation 𝚺𝜉 .

10

0.6 0.5

0.50

0.4

0.48

0.3

0.46

0.2

0.44

0.1

0.42

0.0 0.6m

6m

9m

15m 24m 30m 36m 42m

Training Iter. (imgs)

(a) ImageNet-64, EDM, 𝑁 = 10K

0.475 0.470 0.465 0.460 0.455 0.450 0.445 0.440 0.435 0.430

Memorization Ratio (0-1)

0.7

ICR

ICR

0.52

Memorization Ratio (0-1)

ICR Memorization Ratio

0.54

0.8

ICR Memorization Ratio

0.6 0.4 0.2 0.0

25k

75k

125k

175k

225k

Training Iter. (batches)

275k

(b) ImageNet-256, SiT-B/2, 𝑁 = 20K

Figure 6: ICR dynamics consistently anticipate memorization across large-scale datasets. We repeat the analysis of ICR (blue) and memorization ratio (red) on ImageNet in data limited settings. (a) EDM trained on a 10K image subset of ImageNet 64 × 64. (b) SiT-B/2 diffusion model trained on a 20K image subset of ImageNet 256 × 256. In both cases, ICR dips and then rises before the memorization ratio increases, mirroring the behavior on CIFAR10 experiments (Figure 5). Nearest neighbors in (Example 2)

Nearest neighbors in (Example 1) Training Iter = 0.6M imgs ICR = 0.546 Training Iter = 7M imgs ICR = 0.438 Training Iter = 50M imgs ICR = 0.504 Ref. Image

Farthest

Nearest

Ref. Image

Nearest

Farthest

Figure 7: Nearest neighbors of invariant components throughout limited data training. We visualize nearest neighbors in 𝒔 as in Figure 2. The neighbors qualitatively track ICR and the model’s generalization: in the first row (early training, near initialization), ICR is large and neighbors are not semantically meaningful; in the second row (intermediate training), ICR is smallest and neighbors are reasonable; in the third row (severe overfitting), ICR increases again and neighbor quality degrades.

5.2 Early Learning in the Representation Space Under Limited Data In this subsection, we investigate representation dynamics in the limited data regime. In this setting, recent studies [27, 28, 30, 31] have demonstrated an early learning phenomenon: image generation quality first improves and the model generalizes well in an initial phase of training, before eventually deteriorating as the model begins to memorize. To examine whether the representation space undergoes a similar early learning trajectory in data-limited settings, we track ICR and a memorization ratio [33, 34] across training. Concretely, we train an EDM [39] model on a subset of CIFAR10 with 𝑁 = 4096 images, an EDM on a 10K subset of ImageNet 64 × 64, and a SiT-B/2 [40] model on a 20K subset of ImageNet 256 × 256, and for each setting report ICR computed on training features together with the memorization ratio.4 In these experiments, as reported in Figures 5 and 6, ICR follows a clear U-shaped curve, in sharp contrast to the data-rich case in Figure 4. This indicates that feature invariance in the limited 4 The memorization ratio is computed by generating 10K images, extracting features for generated and training images

using an external encoder [33], and, for each generated image, checking whether its maximum cosine similarity to any training image exceeds 0.6. Such samples are counted as memorized.

11

Tr( S) Tr( )

0.6

0.4

0.6

0.4

0.2

0.2

0.0

Tr( S) Tr( )

0.8

Trace

Trace

0.8

2.5m

7.5m

15m

25m

50m

Training Iter (imgs)

0.0

100m

(a) CIFAR10, EDM, 𝑁 = 4096

2.5m

7.5m

15m

25m

50m

Training Iter (imgs)

100m

(b) CIFAR10, EDM, 𝑁 = 50K

Figure 8: How feature expansion differs in data-limited and data-rich diffusion training. We train two EDM-based diffusion models on CIFAR10 with different training set sizes and track the traces of the invariant and residual covariances over training (as labeled). data regime also exhibits an early learning pattern: it improves during the initial phase of training and then degrades as training continues. We moreover observe that the memorization ratio remains essentially zero around the ICR minimum and begins to increase only afterward. This suggests that the onset of memorization is reflected directly in the representation space: beyond the point where ICR is minimized, the model increasingly fits sample-specific idiosyncrasies [54] rather than shared, stable semantic structure. This transition in the representation is also visible in Figure 7, where we use 𝒔 to find nearest neighbors across the training trajectory in the data-limited case. During the middle phase of training, when ICR is small, the nearest neighbors are semantically close to the query, whereas in the early under-trained phase and the late heavily memorizing phase, where ICR is larger, the retrieved neighbors are noticeably less meaningful. Finally, we note that the ICR values reported here are computed purely from training features and do not require generation or an external evaluation network. Taken together with the previous subsection, this suggests that ICR can serve as a reliable “generalized” generation metric that is monitorable during training without sampling. In the data-rich regime, it closely tracks improvements in generative quality in a way that parallels FID. In the data limited regime, where prior work has pointed out that FID is not entirely trustworthy for detecting memorization [32], ICR can act as a coarse, easy-to-monitor early stopping signal, complementary to standard generation-based metrics such as the memorization ratio [33].

5.3 How Feature Expansion Differs in Data-Rich and Data-Limited Regimes In the previous subsections, we highlighted that the relative feature invariance, as measured by ICR, follows very different trajectories in the data-rich and data-limited regimes. However, ICR is a relative quantity: it summarizes how much residual variance contaminates invariant directions, but does not reveal how the total representation energy evolves. In particular, when ICR increases in the late stage of training under limited data, it is not clear whether this reflects an overall shrinkage of the representation space, a reallocation of variance from invariant to residual components, or some combination of both. To disentangle these possibilities, we examine the traces of the invariant and residual covariances over training. In Figure 8, we train two EDM-based diffusion models with different training set sizes to compare a data-limited setting (4096 images) with a data-abundant setting (50K images). Across both settings, the total feature energy, measured by Tr(𝚺𝑠 ) + Tr(𝚺𝜉 ), increases steadily over training. The regimes differ, however, in how this growth is allocated between invariant and residual components. 12

In the data-abundant case, Tr(𝚺𝑠 ) continues to increase throughout training, while Tr(𝚺𝜉 ) grows only mildly, indicating that additional feature capacity is predominantly devoted to invariant structure. In the data limited case, Tr(𝚺𝑠 ) rises initially but then saturates, whereas Tr(𝚺𝜉 ) continues to grow. This suggests that, once the limited semantic structure in the dataset has been largely extracted, further feature expansion is dominated by augmentation-sensitive residual variability. Consistent with this picture, ICR decreases with training in the data-abundant regime but exhibits a U-shaped trajectory under limited data, reflecting the late stage shift from invariant to residual energy.

6 Conclusion In this work we revisit diffusion models from a self-supervised representation learning perspective. We introduce an invariance–residual decomposition of diffusion representations and the Invariant Contamination Ratio (ICR), a label-free metric that measures how much augmentation and noisesensitive variation contaminates stable structure in the feature space. Using this framework, we demonstrate that diffusion noise levels admit a semantic window where ICR is minimized and downstream linear classification performance is maximized, providing a simple SSL-based way to identify the most informative denoising scales. Tracking ICR over training further reveals distinct learning phases: in data-rich regimes it decreases in tandem with improvements in generative quality, while in data-limited regimes, it exhibits an early learning pattern that anticipates the onset of memorization. These findings suggest that diffusion models can be monitored and evaluated through their own representation space, providing intrinsic training-time signals that complement conventional generation-based metrics.

Acknowledgment XL, YJ, XL, LS, ZZ (UM) and QQ acknowledge support from NSF CAREER CCF-2143904, NSF CCF2212066, NSF CCF-2212326, NSF IIS 2402950, ONR N000142512339, and Google Research Scholar and Google TPU Award. LS and QQ acknowledge support from DARPA HR00112520042. JZ and ZZ (OSU) acknowledge funding support from NSF IIS 2312840 and IIS 2402952. LS acknowledges funding support from NSF IIS 2435746. We also thank all the anonymous reviewers for their valuable suggestions and fruitful discussions.

References [1] J. Sohl-Dickstein, E. Weiss, N. Maheswaranathan, and S. Ganguli, “Deep unsupervised learning using nonequilibrium thermodynamics,” in International conference on machine learning, pp. 2256–2265, pmlr, 2015. [2] J. Ho, A. Jain, and P. Abbeel, “Denoising diffusion probabilistic models,” Advances in neural information processing systems, vol. 33, pp. 6840–6851, 2020. [3] J. Song, C. Meng, and S. Ermon, “Denoising diffusion implicit models,” in International Conference on Learning Representations, 2021. [4] I. Alkhouri, S. Liang, R. Wang, Q. Qu, and S. Ravishankar, “Diffusion-based adversarial purification for robust deep mri reconstruction,” in ICASSP 2024-2024 IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), pp. 12841–12845, IEEE, 2024. [5] Y. Jia, S. Chen, Y. Pan, X. Li, L. Shi, C. Jung, H. Yuan, I. Alkhouri, Y. C. Wu, S. Ravishankar, et al., “Forcingdas: Unified and robust data assimilation via diffusion forcing,” arXiv preprint arXiv:2605.14285, 2026.

13

[6] B. Song, S. M. Kwon, Z. Zhang, X. Hu, Q. Qu, and L. Shen, “Solving inverse problems with latent diffusion models via hard data consistency,” in International Conference on Learning Representations, vol. 2024, pp. 7624–7654, 2024. [7] X. Li, S. M. Kwon, S. Liang, I. R. Alkhouri, S. Ravishankar, and Q. Qu, “Decoupled data consistency with diffusion purification for image restoration,” arXiv preprint arXiv:2403.06054, 2024. [8] J. L. Watson, D. Juergens, N. R. Bennett, B. L. Trippe, J. Yim, H. E. Eisenach, W. Ahern, A. J. Borst, R. J. Ragotte, L. F. Milles, et al., “De novo design of protein structure and function with rfdiffusion,” Nature, 2023. [9] A. Lou, C. Meng, and S. Ermon, “Discrete diffusion modeling by estimating the ratios of the data distribution,” in International Conference on Machine Learning, pp. 32819–32848, PMLR, 2024. [10] B. F. Labs, S. Batifol, A. Blattmann, F. Boesel, S. Consul, C. Diagne, T. Dockhorn, J. English, Z. English, P. Esser, S. Kulal, K. Lacey, Y. Levi, C. Li, D. Lorenz, J. Müller, D. Podell, R. Rombach, H. Saini, A. Sauer, and L. Smith, “Flux.1 kontext: Flow matching for in-context image generation and editing in latent space,” arXiv preprint, 2025. [11] Google, “Veo 3: Google’s most capable video generation model,” tech. rep., Google, 2025. [12] D. Baranchuk, A. Voynov, I. Rubachev, V. Khrulkov, and A. Babenko, “Label-efficient semantic segmentation with diffusion models,” in International Conference on Learning Representations, 2022. [13] W. Xiang, H. Yang, D. Huang, and Y. Wang, “Denoising diffusion autoencoders are unified selfsupervised learners,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 15802– 15812, 2023. [14] S. Mukhopadhyay, M. Gwilliam, V. Agarwal, N. Padmanabhan, A. Swaminathan, S. Hegde, T. Zhou, and A. Shrivastava, “Diffusion models beat gans on image classification,” arXiv preprint arXiv:2307.08702, 2023. [15] X. Chen, Z. Liu, S. Xie, and K. He, “Deconstructing denoising diffusion models for self-supervised learning,” in International Conference on Learning Representations, vol. 2025, pp. 55458–55472, 2025. [16] L. Tang, M. Jia, Q. Wang, C. P. Phoo, and B. Hariharan, “Emergent correspondence from image diffusion,” Advances in Neural Information Processing Systems, vol. 36, pp. 1363–1389, 2023. [17] M. Oquab, T. Darcet, T. Moutakanni, H. Vo, M. Szafraniec, V. Khalidov, P. Fernandez, D. Haziza, F. Massa, A. El-Nouby, et al., “Dinov2: Learning robust visual features without supervision,” Transactions on Machine Learning Research, 2024. [18] K. He, X. Chen, S. Xie, Y. Li, P. Dollár, and R. Girshick, “Masked autoencoders are scalable vision learners,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 16000–16009, 2022. [19] S. Yu, S. Kwak, H. Jang, J. Jeong, J. Huang, J. Shin, and S. Xie, “Representation alignment for generation: Training diffusion transformers is easier than you think,” in International Conference on Learning Representations, 2025. [20] J. Singh, X. Leng, Z. Wu, L. Zheng, R. Zhang, E. Shechtman, and S. Xie, “What matters for representation alignment: Global information or spatial structure?,” in International Conference on Learning Representations, 2026. [21] A. Bardes, J. Ponce, and Y. LeCun, “Vicreg: Variance-invariance-covariance regularization for selfsupervised learning,” in International Conference on Learning Representations, 2022. [22] T. Chen, S. Kornblith, M. Norouzi, and G. Hinton, “A simple framework for contrastive learning of visual representations,” in International conference on machine learning, pp. 1597–1607, PMLR, 2020.

14

[23] S. Chen, H. Zhang, M. Guo, Y. Lu, P. Wang, and Q. Qu, “Exploring low-dimensional subspace in diffusion models for controllable image editing,” Advances in neural information processing systems, vol. 37, pp. 27340–27371, 2024. [24] J. Zbontar, L. Jing, I. Misra, Y. LeCun, and S. Deny, “Barlow twins: Self-supervised learning via redundancy reduction,” in International conference on machine learning, pp. 12310–12320, PMLR, 2021. [25] P. Wang, H. Zhang, Z. Zhang, S. Chen, Y. Ma, and Q. Qu, “Diffusion models learn low-dimensional distributions via subspace clustering,” arXiv preprint, 2024. [26] X. Li, Z. Zhang, X. Li, S. Chen, Z. Zhu, P. Wang, and Q. Qu, “Understanding representation dynamics of diffusion models via low-dimensional modeling,” Advances in Neural Information Processing Systems, vol. 38, pp. 107365–107404, 2026. [27] P. Li, Z. Li, H. Zhang, and J. Bian, “On the generalization properties of diffusion models,” Advances in Neural Information Processing Systems, vol. 36, pp. 2097–2127, 2023. [28] X. Li, Y. Dai, and Q. Qu, “Understanding generalizability of diffusion models requires rethinking the hidden gaussian structure,” Advances in neural information processing systems, vol. 37, pp. 57499–57538, 2024. [29] H. Zhang, Z. Huang, S. Chen, J. Zhou, Z. Zhang, P. Wang, and Q. Qu, “Understanding generalization in diffusion models via probability flow distance,” arXiv preprint arXiv:2505.20123, 2025. [30] R. Baptista, A. Dasgupta, N. B. Kovachki, A. Oberai, and A. M. Stuart, “Memorization and regularization in generative diffusion models,” arXiv preprint arXiv:2501.15785, 2025. [31] T. Bonnaire, R. Urfin, G. Biroli, and M. Mézard, “Why diffusion models don’t memorize: The role of implicit dynamical regularization in training,” Advances in Neural Information Processing Systems, vol. 38, pp. 141266–141286, 2026. [32] G. Stein, J. Cresswell, R. Hosseinzadeh, Y. Sui, B. Ross, V. Villecroze, Z. Liu, A. L. Caterini, E. Taylor, and G. Loaiza-Ganem, “Exposing flaws of generative model evaluation metrics and their unfair treatment of diffusion models,” Advances in Neural Information Processing Systems, vol. 36, pp. 3732–3784, 2023. [33] E. Pizzi, S. D. Roy, S. N. Ravindra, P. Goyal, and M. Douze, “A self-supervised descriptor for image copy detection,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 14532–14542, 2022. [34] H. Zhang, J. Zhou, Y. Lu, M. Guo, P. Wang, L. Shen, and Q. Qu, “The emergence of reproducibility and consistency in diffusion models,” in International Conference on Machine Learning, pp. 60558–60590, PMLR, 2024. [35] B. D. Anderson, “Reverse-time diffusion equation models,” Stochastic Processes and their Applications, vol. 12, no. 3, pp. 313–326, 1982. [36] B. Efron, “Tweedie’s formula and selection bias,” Journal of the American Statistical Association, 2011. [37] Z. Kadkhodaie, F. Guth, E. Simoncelli, and S. Mallat, “Generalization in diffusion models arises from geometry-adaptive harmonic representations,” in International Conference on Learning Representations, vol. 2024, pp. 46543–46567, 2024. [38] O. Ronneberger, P. Fischer, and T. Brox, “U-net: Convolutional networks for biomedical image segmentation,” in International Conference on Medical image computing and computer-assisted intervention, pp. 234–241, Springer, 2015. [39] T. Karras, M. Aittala, T. Aila, and S. Laine, “Elucidating the design space of diffusion-based generative models,” Advances in neural information processing systems, vol. 35, pp. 26565–26577, 2022.

15

[40] N. Ma, M. Goldstein, M. S. Albergo, N. M. Boffi, E. Vanden-Eijnden, and S. Xie, “Sit: Exploring flow and diffusion-based generative models with scalable interpolant transformers,” in European Conference on Computer Vision, pp. 23–40, Springer, 2024. [41] M. Fuest, P. Ma, M. Gui, J. Schusterbauer, V. T. Hu, and B. Ommer, “Diffusion models and representation learning: A survey,” IEEE Transactions on Pattern Analysis and Machine Intelligence, 2026. [42] K. He, H. Fan, Y. Wu, S. Xie, and R. Girshick, “Momentum contrast for unsupervised visual representation learning,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 9729–9738, 2020. [43] J.-B. Grill, F. Strub, F. Altché, C. Tallec, P. Richemond, E. Buchatskaya, C. Doersch, B. Avila Pires, Z. Guo, M. Gheshlaghi Azar, et al., “Bootstrap your own latent-a new approach to self-supervised learning,” Advances in neural information processing systems, vol. 33, pp. 21271–21284, 2020. [44] A. v. d. Oord, Y. Li, and O. Vinyals, “Representation learning with contrastive predictive coding,” arXiv preprint arXiv:1807.03748, 2018. [45] T. Wang and P. Isola, “Understanding contrastive representation learning through alignment and uniformity on the hypersphere,” in International conference on machine learning, pp. 9929–9939, PMLR, 2020. [46] K. Fukunaga, Introduction to Statistical Pattern Recognition. Academic Press, 2 ed., 1990. Second edition. [47] R. A. Horn and C. R. Johnson, Matrix analysis. Cambridge university press, 2012. [48] R. A. Fisher, “The use of multiple measurements in taxonomic problems,” Annals of Eugenics, vol. 7, no. 2, pp. 179–188, 1936. [49] A. Krizhevsky, “Learning multiple layers of features from tiny images,” tech. rep., University of Toronto, 2009. [50] J. Deng, W. Dong, R. Socher, L.-J. Li, K. Li, and L. Fei-Fei, “Imagenet: A large-scale hierarchical image database,” in 2009 IEEE conference on computer vision and pattern recognition, pp. 248–255, Ieee, 2009. [51] B. Wang and J. J. Vastola, “Diffusion models generate images like painters: an analytical theory of outline first, details later,” arXiv preprint arXiv:2303.02490, 2023. [52] M. Heusel, H. Ramsauer, T. Unterthiner, B. Nessler, and S. Hochreiter, “Gans trained by a two time-scale update rule converge to a local nash equilibrium,” Advances in neural information processing systems, vol. 30, 2017. [53] C. Szegedy, W. Liu, Y. Jia, P. Sermanet, S. Reed, D. Anguelov, D. Erhan, V. Vanhoucke, and A. Rabinovich, “Going deeper with convolutions,” in Proceedings of the IEEE conference on computer vision and pattern recognition, pp. 1–9, 2015. [54] Z. Zhang, X. Li, X. Li, L. Shi, M. Wu, M. Tao, and Q. Qu, “Generalization of diffusion models arises with a balanced representation space,” in International Conference on Learning Representations, 2026. [55] K. Deja, T. Trzciński, and J. M. Tomczak, “Learning data representations with joint diffusion models,” in Joint European Conference on Machine Learning and Knowledge Discovery in Databases, pp. 543–559, Springer, 2023. [56] J. Zhang, C. Herrmann, J. Hur, L. Polania Cabrera, V. Jampani, D. Sun, and M.-H. Yang, “A tale of two features: Stable diffusion complements dino for zero-shot semantic correspondence,” Advances in Neural Information Processing Systems, vol. 36, pp. 45533–45547, 2023.

16

[57] Y. Shi, C. Xue, J. H. Liew, J. Pan, H. Yan, W. Zhang, V. Y. Tan, and S. Bai, “Dragdiffusion: Harnessing diffusion models for interactive point-based image editing,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8839–8849, 2024. [58] C. S. Sastry, S. H. Dumpala, and S. Oore, “Diffaug: A diffuse-and-denoise augmentation for training robust classifiers,” Advances in Neural Information Processing Systems, 2024. [59] X. Yang and X. Wang, “Diffusion model as representation learner,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 18938–18949, 2023. [60] D. Li, H. Ling, A. Kar, D. Acuna, S. W. Kim, K. Kreis, A. Torralba, and S. Fidler, “Dreamteacher: Pretraining image backbones with deep generative models,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 16698–16708, 2023. [61] N. Stracke, S. A. Baumann, K. Bauer, F. Fundel, and B. Ommer, “Cleandift: Diffusion features without noise,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 117–127, 2025. [62] G. Luo, L. Dunlap, D. H. Park, A. Holynski, and T. Darrell, “Diffusion hyperfeatures: Searching through time and space for semantic correspondence,” Advances in Neural Information Processing Systems, vol. 36, pp. 47500–47510, 2023. [63] S. Mittal, K. Abstreiter, S. Bauer, B. Schölkopf, and A. Mehrjou, “Diffusion based representation learning,” in International conference on machine learning, pp. 24963–24982, PMLR, 2023. [64] Y. Wang, Y. Schiff, A. Gokaslan, W. Pan, F. Wang, C. De Sa, and V. Kuleshov, “Infodiffusion: Representation learning using information maximizing diffusion models,” in International Conference on Machine Learning, pp. 36336–36354, PMLR, 2023. [65] D. A. Hudson, D. Zoran, M. Malinowski, A. K. Lampinen, A. Jaegle, J. L. McClelland, L. Matthey, F. Hill, and A. Lerchner, “Soda: Bottleneck diffusion models for representation learning,” in Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 23115–23127, 2024. [66] K. Preechakul, N. Chatthee, S. Wizadwongsa, and S. Suwajanakorn, “Diffusion autoencoders: Toward a meaningful and decodable representation,” in Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, pp. 10619–10629, 2022. [67] Y. Han, A. Han, W. Huang, C. Lu, and D. Zou, “Can diffusion models learn hidden inter-feature rules behind images?,” in International Conference on Machine Learning, pp. 21704–21732, PMLR, 2025. [68] Y. Wang, P. Wang, H. Jiang, Z. Yang, Q. Huang, and Z. Wang, “Revisiting spectral representations in generative diffusion models,” 2026. [69] K. K. Agrawal, A. K. Mondal, A. Ghosh, and B. Richards, “𝛼-req : Assessing representation quality in self-supervised learning by measuring eigenspectrum decay,” Advances in Neural Information Processing Systems, vol. 35, pp. 17626–17638, 2022. [70] Q. Garrido, R. Balestriero, L. Najman, and Y. Lecun, “Rankme: Assessing the downstream performance of pretrained self-supervised representations by their rank,” in International conference on machine learning, pp. 10929–10974, PMLR, 2023. [71] V. Thilak, C. Huang, O. Saremi, L. Dinh, H. Goh, P. Nakkiran, J. Susskind, and E. Littwin, “Lidar: Sensing linear probing performance in joint embedding ssl architectures,” in International Conference on Learning Representations, vol. 2024, pp. 56726–56765, 2024. [72] M. Kamb and S. Ganguli, “An analytic theory of creativity in convolutional diffusion models,” in International Conference on Machine Learning, pp. 28795–28831, PMLR, 2025.

17

[73] L. Shi, M. Wu, H. Zhang, Z. Zhang, M. Tao, and Q. Qu, “A closer look at model collapse: From a generalization-to-memorization perspective,” Advances in Neural Information Processing Systems, vol. 38, pp. 40658–40691, 2026. [74] B. Achilli, E. Ventura, G. Silvestri, B. Pham, G. Raya, D. Krotov, C. Lucibello, and L. Ambrogioni, “Losing dimensions: Geometric memorization in generative diffusion,” arXiv preprint arXiv:2410.08727, 2024. [75] S. Buchanan, D. Pai, Y. Ma, and V. De Bortoli, “On the edge of memorization in diffusion models,” Advances in Neural Information Processing Systems, vol. 38, pp. 96113–96157, 2026. [76] B. Wang, E. L. B. Finn, and B. Liu, “The two clocks and the innovation window: When and how generative models learn rules,” arXiv preprint arXiv:2605.10019, 2026. [77] B. Wang and C. Pehlevan, “An analytical theory of spectral bias in the learning dynamics of diffusion models,” Advances in Neural Information Processing Systems, vol. 38, pp. 95865–95963, 2026. [78] A. Favero, A. Sclocchi, and M. Wyart, “Bigger isn’t always memorizing: Early stopping overparameterized diffusion models,” arXiv preprint arXiv:2505.16959, 2025. [79] M. Niedoba, B. Zwartsenberg, K. P. Murphy, and F. Wood, “Towards a mechanistic explanation of diffusion model generalization,” in Forty-second International Conference on Machine Learning, 2025. [80] A. Lukoianov, C. Yuan, J. Solomon, and V. Sitzmann, “Locality in image diffusion models emerges from data statistics,” Advances in Neural Information Processing Systems, vol. 38, pp. 95121–95157, 2025. [81] G. Biroli, T. Bonnaire, V. De Bortoli, and M. Mézard, “Dynamical regimes of diffusion models,” Nature Communications, 2024. [82] A. Sclocchi, A. Favero, and M. Wyart, “A phase transition in diffusion models reveals the hierarchical nature of data,” Proceedings of the National Academy of Sciences, 2025. [83] L. Ambrogioni, “In search of dispersed memories: Generative diffusion models are associative memory networks,” Entropy, 2024. [84] B. Pham, G. Raya, M. Negri, M. J. Zaki, L. Ambrogioni, and D. Krotov, “Memorization to generalization: Emergence of diffusion models from associative memory,” arXiv preprint arXiv:2505.21777, 2025. [85] S. Kornblith, T. Chen, H. Lee, and M. Norouzi, “Why do better loss functions lead to less transferable features?,” Advances in Neural Information Processing Systems, vol. 34, pp. 28648–28662, 2021. [86] X. Li, Y. Jia, X. Li, J. A. Fessler, R. Wang, and Q. Qu, “Mclr: Improving conditional modeling via interclass likelihood-ratio maximization and unifying classifier-free guidance with alignment objectives,” arXiv preprint arXiv:2603.22364, 2026. [87] V. Papyan, X. Han, and D. L. Donoho, “Prevalence of neural collapse during the terminal phase of deep learning training,” Proceedings of the National Academy of Sciences, vol. 117, no. 40, pp. 24652–24663, 2020. [88] Z. Zhu, T. Ding, J. Zhou, X. Li, C. You, J. Sulam, and Q. Qu, “A geometric analysis of neural collapse with unconstrained features,” Advances in Neural Information Processing Systems, vol. 34, pp. 29820–29834, 2021. [89] T. Galanti, A. György, and M. Hutter, “On the role of neural collapse in transfer learning,” in International Conference on Learning Representations, 2022. [90] Z. Wang, Y. Luo, L. Zheng, Z. Huang, and M. Baktashmotlagh, “How far pre-trained models are from neural collapse on the target dataset informs their transferability,” in Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 5549–5558, 2023.

18

[91] X. Li, S. Liu, J. Zhou, X. Lu, C. Fernandez-Granda, Z. Zhu, and Q. Qu, “Understanding and improving transfer learning of deep models via neural collapse,” Transactions on machine learning research, 2024. [92] P. Wang, X. Li, C. Yaras, Z. Zhu, L. Balzano, W. Hu, and Q. Qu, “Understanding deep representation learning via layerwise feature compression and discrimination,” Journal of Machine Learning Research, vol. 26, no. 220, pp. 1–61, 2025. [93] I. Loshchilov and F. Hutter, “Decoupled weight decay regularization,” in International Conference on Learning Representations, 2017.

A Related Work Diffusion-based representation learning. Many works treat a trained diffusion denoiser as a feature extractor and test its features on downstream tasks. These features work well for image classification [13, 14, 55], segmentation [12], correspondence [56, 16], and image editing [57], and recent surveys summarize this line of work [41]. Some articles also use diffusion models to generate augmentations and improve robustness under covariate shift [58]. Since representation quality often depends on the noise level used for feature extraction, several distillation and compression methods aim to reduce the need for expensive timestep search and to improve transfer [59, 60, 61, 62]. Other work changes the training objective or the network to better combine generation and representation learning, for example, by adding new information-based losses or by building an explicit autoencoding structure [63, 64, 65, 66]. [67] further studies whether diffusion models learn hidden dependencies among image features. In contrast to methods that mainly aim to improve downstream transfer or generation, we study how diffusion features change across noise levels and across training, and we link this behavior to self-supervised principles. Representation dynamics and links to self-supervised learning. Prior work [26] study why diffusion representations often peak at an intermediate noise level and explain this unimodal behavior through a low-dimensional data model. They further show that this unimodal pattern disappears when diffusion models transition from generalization to memorization [26]. More recently, Wang et al. [68] connect diffusion models and self-supervised learning through a shared perturbation-kernel perspective and propose a spectral alignment objective that improves diffusion training. This SSL perspective is also reflected in recent diffusion training methods such as REPA, which align diffusion representations with embeddings from pretrained self-supervised encoders (e.g., DINOv2) to improve generation quality and training efficiency [17, 19, 20]. Our work differs from these lines in its goal. Rather than modifying the training objective or introducing an external teacher encoder, we study diffusion representations through principles inspired by self-supervised learning. In particular, our work is closely related to a line of research that develops label-free metrics for predicting downstream representation quality and guiding model selection, including metrics based on covariance spectrum decay, effective rank, and Fisher-style covariance decompositions [69, 70, 71]. Similar to these works, we seek to evaluate representation quality directly from feature statistics without training downstream classifiers. However, unlike prior metrics designed primarily for representation selection in SSL, we leverage the invariant and residual decomposition induced by augmentations and diffusion noise to study diffusionspecific phenomena, including semantic windows, generation quality, memorization, and training dynamics.

19

0.5

0.5

Tr( S) Tr( )

0.3 0.2

0.3 0.2 0.1

0.1 0.0

Tr( S) Tr( )

0.4

Trace

Trace

0.4

0.0

0.01

0.02

0.06

0.14

0.3

0.59

Time step ( t)

1.09

1.92

0.0

3.26

0.0

(a) CIFAR10

0.01

0.02

0.06

0.14

0.3

0.59

Time step ( t)

1.09

1.92

3.26

(b) CIFAR100

Figure 9: Invariant and residual energy across diffusion noise levels. For pretrained EDM models on CIFAR10 and CIFAR100 in the data-rich regime, we plot the traces of the invariant and residual covariances, Tr(𝚺𝑠 (𝜎𝑡 )) and Tr(𝚺𝜉 (𝜎𝑡 )), as functions of the noise level 𝜎𝑡 . Invariant energy Tr(𝚺𝑠 ) increases from low noise, peaks at an intermediate scale, and then decreases in the high noise regime, whereas residual energy Tr(𝚺𝜉 ) grows monotonically with 𝜎𝑡 . Memorization and generalization in diffusion models. Our work contributes to the broad line of research aiming to understand memorization and generalization behaviors in diffusion models [37, 34, 72, 73, 29]. A large body of work has studied memorization in diffusion models from the perspectives of model complexity and data quantity [25, 74, 31, 75, 76], and has shown that memorization typically emerges after an initial generalization phase when diffusion models are trained with limited data [28, 77, 31, 78]. Other works seek to explain why diffusion models are able to recover the underlying score function from discrete empirical samples [79, 80]. The generation and generalization behaviors across the reverse diffusion process have also been studied in [81, 82]. In parallel, Ambrogioni et al. [83] established an asymptotic equivalence between generative diffusion models and modern Hopfield networks (associative memory networks), and a subsequent work [84] leveraged this associative memory perspective to identify spurious samples that emerge as diffusion models transition from generalization to memorization.

B Additional Discussions & Experiments B.1 Component Dynamics Across the Noise Schedule In Section 4, we show that there exists a strong correspondence between ICR and classification performance across noise levels. In this subsection, we also track the energy progression of the total variance of the invariant signal, Tr(𝚺𝑠 ), and the residual variation, Tr(𝚺𝜉 ). As shown in Figure 9, Tr(𝚺𝜉 ) grows monotonically with the noise level 𝜎𝑡 , reflecting the increased reconstructive uncertainty inherent in high-noise denoising. Conversely, Tr(𝚺𝑠 ) exhibits a unimodal trajectory, peaking at intermediate scales. This suggests a “semantic window” where the model’s invariant subspace is most expansive. Notably, peak classification accuracy (as shown in Figure 3) does not always correspond with the maximum of Tr(𝚺𝑠 ), but rather with the minimum of the ICR ratio. This indicates that representation utility is governed not by the absolute magnitude of the invariant signal, but by its strength relative to the contaminating residual variation.

20

CIFAR10, EDM, N = 4,096

Figure 10: Correspondence between ICR and classification accuracy across noise levels in the data-limited setting. We study the behavior of ICR across noise levels in the data-limited setting under prolonged training. In this regime, classification accuracy no longer exhibits the unimodal trend observed in the generalization phase, and instead decreases monotonically as noise increases. In contrast, ICR increases monotonically and maintains a clear negative correlation with classification accuracy. This demonstrates that ICR continues to track representation quality even when classification accuracy no longer follows the typical generalization pattern.

B.2 ICR and Classification Accuracy in the Data-limited Regime In Figure 3, we show that ICR exhibits a strong negative correlation with classification accuracy across noise levels when diffusion models are trained using the full training set. Interestingly, the classification accuracy follows a unimodal trend as the noise level increases, a phenomenon previously observed and analyzed in [13, 26]. Moreover, Li et al. [26] showed that this unimodal behavior is closely associated with the diffusion model correctly learning the underlying lowdimensional data distribution, and that it disappears once the model begins to memorize the training data, at which point the accuracy instead decreases monotonically with noise level. To further test the robustness of the correlation between ICR and classification accuracy, we train an EDM model on a subset of 4,096 CIFAR10 images for a prolonged period until the model substantially overfits and memorizes the training data. We then evaluate both ICR and classification accuracy across noise levels at this checkpoint, and report the results in Figure 10. As shown in the figure, classification accuracy no longer exhibits the unimodal trend observed during the generalization phase, and instead decreases monotonically as the noise level increases. Correspondingly, ICR exhibits a monotonic increasing trend across noise levels, further strengthening the correlation between ICR and representation quality even when the standard semantic window disappears due to memorization.

B.3 Discussion on the Alignment and Uniformity Metrics [45] Wang et al. [45] introduced two feature-level criteria for contrastive SSL encoders. In our notation, let 𝒉(𝑎(𝒙)) ∈ R𝑑 denote the feature of an augmented image, and let (𝒙, 𝒙 ′ ) be a positive pair obtained from two augmentations of the same image. The alignment loss is ℒalign (𝒉) B E(𝒙,𝒙 ′ ) ∥𝒉(𝑎1 (𝒙)) − 𝒉(𝑎 2 (𝒙))∥2𝛼 ,





21

(6)

ICR ( )

0.36

25 20

0.34

15

0.32 0.30

10

0.28 2.5m

7.5m

15m

25m

50m

Training Iter. (imgs)

25

0.300

5

0.26

30

0.325

Lalign(h) ( )

0.38

FID Score ( )

ICR FID Score

0.275

20

Alignment Loss FID Score

0.250 0.225

0.150

15 10

0.200

5

0.175

100m

FID Score ( )

30

0.40

2.5m

7.5m

15m

25m

50m

Training Iter. (imgs)

100m

(b) ℒalign (𝒉) and FID

(a) ICR and FID

Figure 11: Alignment versus ICR in data-rich diffusion training (CIFAR10, EDM). We track FID together with ICR and the alignment loss ℒalign over training on full CIFAR10. Both ICR (blue) and FID (brown) decrease monotonically, indicating improving representation invariance and generative quality, while ℒalign (green) increases despite being a lower is better metric. the expected squared distance between two views of the same sample. The uniformity loss is

h

ℒuniform (𝒉; 𝑡) B log E𝒙,𝒚 exp −𝑡∥𝒉(𝑎(𝒙)) − 𝒉(𝑎 (𝒚))∥22

i

,

𝑡 > 0,

(7)

which encourages features to be spread out on the unit hypersphere. In the contrastive setting, encoders that achieve low alignment together with good uniformity tend to have strong downstream classification accuracy. Our focus is slightly different. We are interested in how the representation space evolves across diffusion noise levels and along the training trajectory, and in particular in a relative notion of invariance that compares invariant structure to view specific variation while remaining stable under overall feature expansion. In this regime, the alignment loss becomes less informative. As shown in Figure 11, in the data-rich case the FID and ICR both decrease monotonically as training progresses, while ℒalign continues to increase, suggesting worse alignment. This apparent disagreement is largely due to the fact that ℒalign is an absolute squared distance: during diffusion training the overall feature variance grows (see Figure 8), so alignment can increase even when the relative invariant structure is improving.

B.4 Discussion on the Class Separation and Silhouette Score Metrics Besides the alignment and uniformity metrics discussed in [45], we additionally discuss two other representation metrics here: the class separation metric and the Silhouette score. Class separation is a metric proposed in [85], which measures the within-class variation of representations relative to the overall variation.5 Formally, based on cosine distances between normalized features, let 𝒉 𝑘,𝑚 denote the representation of sample 𝑚 from class 𝑘, where 𝐾 is the number of classes and 𝑁 𝑘 is the number of samples in class 𝑘. The class separation score 𝑅2 is defined as 5 The concept of maximizing separation between classes has recently been studied in improving diffusion model

generation without classifier-free guidance [86].

22

(a) ICR

(b) Silhouette score

(c) Class separation

Figure 12: ICR, Silhouette score, and class separation in data-limited diffusion training (CIFAR10, EDM). We revisit the experiment in Figure 4 (Training EDM on CIFAR10, with N=4,096 images) by incorporating two additional representation metrics: (i) the Silhouette Score, a partially unsupervised metric that relies on pseudo-labels (e.g., from k-means), and (ii) Class Separation, a supervised metric that depends on ground-truth labels. As shown in the figure, ICR exhibits a stronger and more consistent alignment with the memorization ratio, whereas the other two metrics saturate too early and fail to capture this trend as reliably.

𝑅2 = 1 −

𝑑¯ within , 𝑑¯ total

where

𝑑¯ within =

𝑁𝑘 Õ 𝑁𝑘 𝐾 Õ Õ 1 − sim(𝒉 𝑘,𝑚 , 𝒉 𝑘,𝑛 ) 𝑘=1 𝑚=1 𝑛=1 𝐾 𝐾 𝑁 𝑗 𝑁𝑘

𝑑¯ total =

𝐾𝑁 𝑘2

,

Õ Õ Õ Õ 1 − sim(𝒉 𝑗,𝑚 , 𝒉 𝑘,𝑛 ) 𝑗=1 𝑘=1 𝑚=1 𝑛=1

𝐾2 𝑁 𝑗 𝑁𝑘

.

The metric was shown in [85] to correlate with transferability of ImageNet pre-trained models on downstream classification tasks. However, as the name suggests, the metric requires ground-truth class labels and is therefore not suitable for unsupervised settings. In addition, being a global statistic, it does not explicitly capture directional structure in the representation space as ICR does. Silhouette score is a clustering-based metric that evaluates how well a representation separates data into groups (or classes when supervised labels are available). Given a feature representation 𝒉 𝑖 , define • 𝑎(𝑖): the average distance between 𝒉 𝑖 and samples within the same cluster, • 𝑏(𝑖): the minimum average distance between 𝒉 𝑖 and samples in other clusters. The Silhouette score is then defined as

𝑠(𝑖) =

𝑏(𝑖) − 𝑎(𝑖) , max(𝑎(𝑖), 𝑏(𝑖))

23

and the overall score is obtained by averaging 𝑠(𝑖) over all samples. The Silhouette score measures the balance between intra-cluster compactness and inter-cluster separation, where larger values indicate better separation. However, this metric is not fully unsupervised, since it still requires cluster assignments, typically obtained through algorithms such as k-means, making it sensitive to hyperparameters such as the number of clusters. In addition, it relies on pairwise distances, which can become unstable in high-dimensional feature spaces, and similarly does not explicitly capture directional structure. To compare these two metrics with ICR, we redo the experiments in Figure 5 and report the results in Figure 12. We observe that both the class separation score and the Silhouette score exhibit a unimodal trend during training, suggesting that they can partially capture the transition from learning the underlying data distribution to memorizing training samples. However, as shown in the figure, ICR exhibits a substantially stronger and more consistent alignment with the memorization ratio, whereas the other two metrics saturate much earlier and fail to reliably track the later-stage memorization behavior. We conjecture that this limitation is partly due to the global nature of these metrics.

B.5 Connection to Neural Collapse In the main text we showed that the generalized eigenvalues admit a direct SNR interpretation: for each Fisher direction 𝒗 𝑖 , the generalized Rayleigh quotient equals 𝜆 𝑖 , and ICR =

𝑑 𝑑+

−1/2

This follows by writing e 𝚺 𝑠 B 𝚺𝜉

Í𝑑

𝑖=1 𝜆 𝑖

−1/2

𝚺 𝑠 𝚺𝜉

𝑑 Õ

,

𝑖=1

𝜆 𝑖 = Tr 𝚺−1 𝜉 𝚺𝑠 .



1/2

and 𝒖𝑖 = 𝚺𝜉 𝒗 𝑖 , so that 𝒖𝑖 are the eigenvectors of e 𝚺𝑠

e with eigenvalues 𝜆 𝑖 and Tr(𝚺−1 𝑖 𝜆𝑖 . 𝜉 𝚺𝑠 ) = Tr(𝚺𝑠 ) = −1 The trace form Tr(𝚺𝜉 𝚺𝑠 ) could be linked to a familiar quantity in the Neural Collapse (𝒩 𝒞 ) literature [87, 88]. Neural Collapse refers to a phenomenon observed near the terminal phase of training in classification networks: penultimate features for each class collapse to a single mean vector, and these class means become maximally separated. A standard metric for quantifying this behavior is the 𝒩 𝒞 1 score, defined for a 𝐾-class classifier as Í

𝒩 𝒞1 =

 1 Tr 𝚺†𝐵 𝚺𝑊 , 𝐾

where 𝐾

𝒉𝐺 =

𝑛

1 ÕÕ 𝒉 𝑘,𝑖 , 𝑛𝐾

1 𝚺𝑊 B 𝑛𝐾

𝑘=1 𝑖=1 𝐾 Õ 𝑛 Õ

1 𝒉¯ 𝑘 = 𝑛

𝑛 Õ

𝒉 𝑘,𝑖 ,

(1 ≤ 𝑘 ≤ 𝐾),

𝑖=1 𝐾

⊤ 𝒉 𝑘,𝑖 − 𝒉¯ 𝑘 𝒉 𝑘,𝑖 − 𝒉¯ 𝑘 ,





𝑘=1 𝑖=1

 ⊤ 1Õ ¯ 𝚺𝐵 B 𝒉 𝑘 − 𝒉 𝐺 𝒉¯ 𝑘 − 𝒉 𝐺 . 𝐾 𝑘=1

Here 𝚺𝑊 and 𝚺𝐵 denote the within-class and between-class covariance matrices of the penultimate features. The design principle behind 𝒩 𝒞 1 is closely related to ours: both metrics compare two covariance structures via a trace of a generalized eigenvalue type object, effectively measuring a signal-to-noise ratio in feature space. The key difference is that 𝒩 𝒞 1 is label-based, contrasting between class 24

and within-class variation, whereas ICR is built from a self-supervised principle that contrasts perturbation-invariant and perturbation-sensitive components without using labels. Interestingly, several works have used 𝒩 𝒞 1 and related quantities as metrics for assessing the transferability of pretrained discriminative models to downstream tasks [89, 90, 91, 92]. Our results suggest that a similar trace-based viewpoint extends naturally to diffusion models, and that a unified representation-based evaluation principle may be possible that covers both discriminative and generative settings through appropriate choices of “signal” and “noise” covariances.

B.6 Technical Details on Calculating ICR In Section 3.2, we introduced the formal definition of the metric ICR and argue that it can be efficiently estimated using only two augmented views and a subset of training features6 . In this subsection, we briefly discuss alternative formulations of the metric, dive into more detail on the estimation of it and the robustness of the estimation regards different number of samples used. As noted in the main manuscript, the computation of ICR involves solving a generalized eigenvalue problem, one may argue that a simpler alternative is a trace-based statistic such as Tr(𝚺𝜉 )/Tr(𝚺𝑠 ), which aggregates all directions into a single global statistic. However, this aggregation loses directional information. For example, consider a representation where only a small number of directions carry strong invariant signal while the remaining directions are dominated by residual variation. In this case, the trace ratio averages over all directions and fails to reflect the presence of these highly informative directions. In contrast, the generalized eigenvalue formulation explicitly captures the signal-to-noise ratio along each direction and is therefore sensitive to such anisotropic structures. Two-view approximation and empirical estimation. The conditional expectation over all augmentations and noise realizations is not available in practice. Following augmentation based self supervised learning, we approximate it using two independent views per image. For each 𝒙, sample 𝑎 1 , 𝑎2 ∼ 𝒜 independently and set 𝒉 1 = 𝒉(𝑎1 (𝒙 0 )),

𝒉2 = 𝒉(𝑎2 (𝒙 0 )).

Under the decomposition above, this can be written as 𝒉 1 = 𝒔(𝒙 0 ) + 𝝃1 ,

𝒉2 = 𝒔(𝒙 0 ) + 𝝃2 ,

where 𝝃𝑣 = 𝝃(𝑎 𝑣 , 𝒙0 ) are zero mean residuals that are conditionally uncorrelated across views given 𝒙 and share the same covariance 𝚺𝜉 . We construct effective semantic and nuisance covariances from the sum and difference of the two views. Define 𝒅 B 𝒉1 − 𝒉2 .

𝒕 B 21 (𝒉1 + 𝒉 2 ),

A direct covariance calculation under the assumptions above yields Cov(𝒅) = 2 𝚺𝜉 ,

Cov(𝒕) = 𝚺𝑠 + 12 𝚺𝜉 ,

6We use all generalized eigenvalues, rather than only the top ones, in order to summarize the invariant-to-residual ratio across the full representation space. Since the metric is directional, if the spectrum is sparse, it indicates that only a few directions carry strong invariant signal, while others are dominated by residual variation.

25

ICR

0.50

ICR (all 50 000 samples)

0.45 0.40 0.5

ICR

0.4 N = 64 N = 128 N = 256 N = 512 N = 1024

0.3 0.2

N = 2048 N = 4096 N = 8192 N = 16384 N = 32768

0.1 0.0 2.5m 5m 7.5m 10m 15m 20m 25m 30m 50m 75m100m200m

Training Iter. (imgs)

Figure 13: Stability of ICR estimates under subsampling. We evaluate ICR on CIFAR10 using a pretrained EDM model (4095 training samples) and a fixed noise level, varying the number of training samples used to estimate the covariances from 𝑁 = 16 up to the full 50K images. As 𝑁 increases, the estimated ICR quickly showcase the similar trend close to the full data estimate. so that 𝚺𝑠 and 𝚺𝜉 can be recovered from the second moments of 𝒕 and 𝒅: 𝚺𝜉 = 21 Cov(𝒅),

𝚺𝑠 = Cov(𝒕) − 14 Cov(𝒅).

In our experiments we estimate Cov(𝒕) and Cov(𝒅) from paired augmentations within each dataset and obtain empirical covariances b 𝚺𝑠 and b 𝚺𝜉 via the same formulas. Estimating ICR from a subset of samples. In practice, ICR is computed from empirical covariances b 𝚺𝑠 and b 𝚺𝜉 estimated from a finite set of features, so the estimate may deviate from its population value depending on the number of samples used. Under standard covariance concentration results p for subgaussian features, the estimation error of b 𝚺𝑠 and b 𝚺𝜉 scales on the order of 𝑑/𝑁 in operator norm, where 𝑑 is the feature dimension and 𝑁 is the number of samples. Since ICR depends on these covariances only through the trace term Tr(𝚺−1 𝜉 𝚺𝑠 ), we expect its finite sample estimate to be relatively stable once 𝑁 is moderately larger than 𝑑. To verify this empirically, we perform a sample complexity study on CIFAR10. Fixing a pretrained EDM model and a representative noise level, we compute ICR using random subsets of training images with sizes 𝑁 ∈ {16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192, 16384, 32768, 50000}, where 50,000 is the full training set size. As shown in Figure 13, the estimated ICR stabilizes quickly: even with a few hundred to a few thousand samples, the ICR trends are already very close to the full dataset estimate. This robustness justifies our use of relatively small subsets of training features to monitor ICR throughout the experiments in the main text. Sensitivity of ICR to augmentation design. Since the computation of ICR relies on collecting representations under random augmentations, we further investigate its robustness to different

26

(a) CIFAR10, EDM, N = 50K

(b) CIFAR10, EDM, N = 4,096

Figure 14: Sensitivity of ICR to augmentation design. . We study the sensitivity of ICR to augmentation design by varying the strength of augmentations. We consider five augmentation levels and compute ICR under each setting. (a) In the data-abundant setting (50K training samples), the ICR curves across noise levels remain highly consistent across different augmentation strengths. (b) In the data-limited setting (4,096 training samples), the temporal evolution of ICR during training exhibits the same qualitative trend once sufficiently strong augmentations are applied. For clarity, we normalize each curve by its initial value to focus on the relative trend. augmentation strengths. Specifically, we consider five augmentation settings: Level 1 uses RandomCrop alone; level 2 adds horizontal flipping; level 3 further adds Color Jitter (the default setting used throughout the manuscript); level 4 additionally adds random rotation; and level 5 further adds CenterCrop. We denote these augmentation settings as Aug.1 through Aug.5. Using these augmentation pipelines, we redo the experiments of ICR across noise levels in the data-rich setting and across training dynamics in the data-limited setting, and report the results in Figure 14. As shown in the figure, once a reasonably rich augmentation pipeline is used (e.g., random crop + flip + color jitter and stronger variants), the resulting ICR trends remain highly consistent, indicating that our findings are robust to the specific choice of augmentations. In contrast, when overly weak augmentations are used (e.g., random crop alone), the behavior becomes noticeably less stable. We conjecture that this is because the invariant-residual decomposition requires sufficiently diverse perturbations in order to meaningfully separate invariant and residual components. Sensitivity of ICR to the choice of feature extraction layer. We evaluate the sensitivity of ICR to layer selection. Specifically, we extract features from multiple layers around the middle of the diffusion model and compute ICR for each choice. The results in Figure 15 show that ICR exhibits consistent behavior across layers in both data-abundant and data-limited settings. In particular, the overall trends and the location of the semantic window both remain stable regardless of layer choice.

C Experimental Details Unless stated otherwise, we apply mean pooling to obtain a single feature vector per sample. For EDM models, we average over spatial dimensions, mapping tensors of shape 𝑁 × 𝐶 × 𝐻 × 𝑊 to

27

(a) CIFAR10, EDM, N = 50K

(b) CIFAR10, EDM, N = 4,096

Figure 15: Sensitivity of ICR to the choice of feature extraction layer. We select multiple layers around the middle of the diffusion model and compute ICR using features from each layer. (a) In the data-abundant setting (50K training samples), the trend of the ICR curves across noise levels remain highly consistent across different layer choices, with the location of the semantic window largely unchanged. (b) In the data-limited setting (4,096 training samples), the temporal evolution of ICR during training exhibits the same qualitative trend across layers, including the U-shaped behavior associated with memorization. 𝑁 × 𝐶. For transformer-based models, we average over the token dimension, mapping 𝑁 × 𝑇 × 𝐷 to 𝑁 × 𝐷. In all experiments in Section 5, we extract features at a fixed intermediate noise scale, using 𝜎𝑡 = 0.29 for EDM based models [39] and 𝑡 = 0.2 for SiT based models [40]. Figure 2. We take a 2,000-image subset of ImageNet64 and extract representations from the dec.16x16_block1 layer of a publicly available pretrained EDM model. For each image, this gives a tensor of shape (576, 16, 16); we also extract representations for 14 augmented views ( hflip , shift±4x , shift±4y , cropc56 , croptl5 , cropbr54 , bright±0.08 , contrast0.85 , sat0.85 , cutout22 , blur3 ), for a total of 15 views. We flatten each tensor into a 147,456 = 576 × 16 × 16dimensional vector. For each image, we take the mean across the 15 view representations to obtain 𝒔 (shared by the image and its augmented views), and subtract 𝒔 from each view representation to obtain the nuisance component 𝝃 for that view. Finally, we pick a reference image (could be augmented) and retrieve its nearest neighbors under cosine similarity, in terms of 𝒔 or 𝝃. We then plot the retrieved neighbors (could be augmented) in the first and second rows, respectively. Figure 3. We evaluate ICR and linear probing accuracy using publicly available EDM models on CIFAR10 and a SiT-XL/2 model on ImageNet 256 × 256, together with a CIFAR100 EDM model that we train ourselves. For EDM backbones we extract features from the dec.16x16_block1 layer near the bottleneck; for SiT-XL/2 we use the output of transformer block 14, the midpoint of the 28-layer network. For EDM models we train a logistic regression classifier with scikit-learn on the full set of training features and report accuracy on test features. For SiT-XL/2, due to the larger feature set, we subsample 200K training features (200 images per class) and train a linear classifier with AdamW [93] for 100 epochs (batch size 8192, learning rate 10−2 , weight decay 10−4 ), reporting test accuracy at the final epoch. These hyperparameters are fixed across all noise levels and are not tuned, since 28

our goal is to capture trends rather than optimize absolute performance. For computing ICR we use a random subset of 𝑁 = 4096 training features for EDM models and 𝑁 = 20K for SiT-XL/2 at each noise level, with the subset held fixed across noise levels in each experiment. Figure 4. We train an EDM model on CIFAR10 and a SiT-B/2 model on ImageNet 256 × 256 using the full training datasets and report FID together with ICR. FID is computed from 50K generated images for each experiment. As above, ICR is estimated from a subset of training features, using 𝑁 = 4096 samples for the EDM model and 𝑁 = 20K for the SiT-B/2 model. Figure 5. We train an EDM model on CIFAR10 using 4096 training images and report ICR together with the memorization ratio over the course of training. For the nearest neighbor visualizations on the right, we take generated samples and find their nearest neighbors among the training images directly in pixel space. For the snapshot at Training iter = 200M, we slightly cherry pick a few generated samples that are clearly memorized to highlight this effect; all other visualizations use fixed random seeds aligned with this snapshot to keep the plots consistent across training iterations. Figure 6. We train an EDM model on ImageNet 64 × 64 using 10K training images (10 per class) and a SiT-B/2 model on ImageNet 256 × 256 using 20K training images (20 per class). Both models are trained with standard class conditional setups. When extracting features for ICR computation, to keep the procedure label-free, we encode all samples using the null class. Figure 7. We follow the same pipeline as in Figure 2, but use checkpoints from an EDM model trained on only 10K ImageNet 64 × 64 samples. We visualize three typical phases of limited-data training: (i) early learning (first row), after 0.6M images, where the model is still improving; (ii) the onset of overfitting (second row), after 7M images, where ICR is smallest and the nearest neighbors remain meaningful and share structure with the reference; and (iii) severe overfitting (last row), after 50M images, where ICR increases and nearest neighbors are no longer semantic. Figure 8. We reuse the EDM models trained on CIFAR10 with 4096 images and with the full 50K images to report the traces of 𝚺𝑠 and 𝚺𝜉 over training. To ensure that our notion of feature expansion is not conflated with simple growth in representation norms, we ℓ 2 normalize each representation before computing the covariances.

D Alignment between Optimal Test Loss and ICR In the main part of the paper, we discussed that ICR can be used as an early-stopping indicator when training diffusion models with limited data. In this section, we provide some preliminary theoretical insights for the underlying cause of such functionality. Specifically, through a simple Gaussian toy model, we show that ICR moves in the same direction as the Bayes optimal linear denoising loss. Toy two-layer linear model. the noisy input and

Fix a noise level 𝜎𝑡 and an encoder 𝑼 ∈ R𝑑×𝐷 . Let 𝒙 𝑡 ∈ R𝐷 denote 𝒉 = 𝑼 𝒙 𝑡 ∈ R𝑑

29

be the feature. As in Section 3, we consider the invariance-variance decomposition of the feature 𝒉 = 𝒔 + 𝝃 where 𝒔 is the invariant component and 𝝃 is the variant (residual) component induced by data augmentations and additive Gaussian noise. We denote their covariances by 𝚺𝑠 (𝑼 ) = Cov(𝒔),

𝚺𝜉 (𝑼 ) = Cov(𝝃),

We reconstruct the clean image from the feature using a linear decoder 𝑾 ∈ R𝐷×𝑑 ,

b 𝒙 0 = 𝑾 𝒉, 2

and define the population linear denoising loss ℒlin (𝑾 ; 𝑼 ) B E 𝒙0 − 𝑾 𝒉 . and thus ★ ℒlin (𝑼 ) B min ℒlin (𝑾 ; 𝑼 ) 𝑾

(8)

for the optimal linear denoising loss. For analytical clarity, we assume that (𝒙 0 , 𝒉) are jointly Gaussian. Proposition 1 (Monotonicity of Bayes optimal loss and ICR). Let 𝑼1 , 𝑼2 be two encoders at the same noise level 𝜎𝑡 , and denote their invariance and variance covariances by 𝚺𝑠 (𝑼 𝑘 ), 𝚺𝜉 (𝑼 𝑘 ) ,

𝑘 ∈ {1, 2}.



Assume the Gaussian model above holds for each encoder. (More variant energy hurts). If the invariant covariance is the same 𝚺𝑠 (𝑼1 ) = 𝚺𝑠 (𝑼2 ), and the variant covariance of 𝑼2 dominates that of 𝑼1 in PSD order, 𝚺𝜉 (𝑼1 ) ⪯ 𝚺𝜉 (𝑼2 ), then the optimal linear denoising loss and ICR both increase: ★ ★ ℒlin (𝑼1 ) ≤ ℒlin (𝑼2 ),

ICR(𝑼1 ) ≤ ICR(𝑼2 ).

(9)

(More invariant energy helps). If instead the variant covariance is the same 𝚺𝜉 (𝑼1 ) = 𝚺𝜉 (𝑼2 ), and the invariant covariance of 𝑼2 dominates that of 𝑼1 , 𝚺𝑠 (𝑼1 ) ⪯ 𝚺𝑠 (𝑼2 ), then the optimal linear denoising loss and ICR both decrease: ★ ★ ℒlin (𝑼1 ) ≥ ℒlin (𝑼2 ),

ICR(𝑼1 ) ≥ ICR(𝑼2 ).

(10)

Proof. We first derive the Bayes optimal loss in the original feature basis. Recall that for any measurable 𝑔 : R𝑑 → R𝐷 , ℒ(𝑔; 𝚺𝑠 , 𝚺𝜉 ) = E ∥𝒙0 − 𝑔(𝒉)∥2 . Let 𝒉 = 𝒔 + 𝝃 with 𝒔 ∼ 𝒩 (0, 𝚺𝑠 ) and 𝝃 ∼ 𝒩 (0, 𝚺𝜉 ) independent, and 𝒙 0 = 𝑨𝒔 + 𝑪𝝃 as in the statement. We now derive the covariance of 𝒙0 and 𝒉: 𝚺 ℎ ℎ B Cov(𝒉) = Cov(𝒔) + Cov(𝝃) = 𝚺𝑠 + 𝚺𝜉 , 𝚺𝑥𝑥 B Cov(𝒙 0 ) = Cov(𝑨𝒔 + 𝑪𝝃) = 𝑨𝚺𝑠 𝑨⊤ + 𝑪𝚺𝜉 𝑪 ⊤ , 𝚺𝑥 ℎ B Cov(𝒙 0 , 𝒉) = Cov(𝑨𝒔 + 𝑪𝝃, 𝒔 + 𝝃) = 𝑨𝚺𝑠 + 𝑪𝚺𝜉 . 30

Now consider 𝒙 0 − 𝑔(𝒉) = (𝒙 0 − E[𝒙0 | 𝒉]) + E[𝒙0 | 𝒉] − 𝑔(𝒉) .



|

{z

} |

𝒄1

{z

}

𝒄2

With this notation, we have

  E ∥𝒙0 − 𝑔(𝒉)∥2 = E (𝒄1 + 𝒄2 )⊤ (𝒄1 + 𝒄2 ) = E ∥𝒙 0 − E[𝒙0 | 𝒉]∥2 + E ∥ E[𝒙 0 | 𝒉] − 𝑔(𝒉)∥2 + 2 E (𝒙 0 − E[𝒙0 | 𝒉])⊤ (E[𝒙0 | 𝒉] − 𝑔(𝒉)) .





Then we can show the cross term vanishes:

 ⊤     ⊤ E[𝒄⊤ 1 𝒄 2 ] = E E[𝒄 1 𝒄 2 | 𝒉] = E E (𝒙 0 − E[𝒙 0 | 𝒉]) 𝒄 2 (𝒉) | 𝒉   = E E[𝒙0 − E[𝒙0 | 𝒉] | 𝒉]⊤ 𝒄 2 (𝒉)   = E (E[𝒙0 | 𝒉] − E[𝒙0 | 𝒉])⊤ 𝒄 2 (𝒉) = 0, where in the first equality we use the law of total expectation and in the third equality we use the fact that 𝒄 2 only depends on 𝒉 and is thus constant inside the conditional expectation. Hence

E ∥𝒙0 − 𝑔(𝒉)∥2 = E ∥𝒙0 − E[𝒙0 | 𝒉]∥2 + E ∥ E[𝒙0 | 𝒉] − 𝑔(𝒉)∥2 . Therefore E ∥𝒙 0 − 𝑔(𝒉)∥2 is minimized when 𝑔(𝒉) = E [𝒙 0 | 𝒉] with minimum value E ∥𝒙 0 − E[𝒙 0 | 𝒉]∥2 . Now with 𝒙 0 , 𝒉 jointly Gaussian, and 𝚺 ℎ ℎ invertible7 , we have

E [𝒙0 | 𝒉] = 𝚺𝑥 ℎ 𝚺−1 ℎ ℎ 𝒉. Write 𝑾 = 𝚺𝑥 ℎ 𝚺−1 ℎ ℎ , then ℒ★ (𝚺𝑠 , 𝚺𝜉 ) = E ∥𝒙 0 − E[𝒙0 | 𝒉]∥2 = E ∥𝒙 0 − 𝑾 𝒉∥2 = E (𝒙 0 − 𝑾 𝒉)⊤ (𝒙0 − 𝑾 𝒉)





= Tr (𝚺𝑥𝑥 ) − Tr 𝚺𝑥 ℎ 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 .



Now what’s left to show are the two co-monotonic facts. Consider a directional perturbation Δ𝚺𝜉 ⪰ 0. Then 𝛿𝚺𝑥𝑥 = 𝑪Δ𝚺𝜉 𝑪 ⊤ ,

𝛿𝚺𝑥 ℎ = 𝑪Δ𝚺𝜉 ,

𝛿𝚺 ℎ ℎ = Δ𝚺𝜉 .

−1 −1 Using 𝛿(𝚺−1 ℎ ℎ ) = −𝚺 ℎ ℎ (𝛿𝚺 ℎ ℎ )𝚺 ℎ ℎ and differentiating

ℒ★ = Tr(𝚺𝑥𝑥 ) − Tr(𝚺𝑥 ℎ 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 ), We can then calculate −1 −1 𝛿ℒ★ = Tr (𝛿𝚺𝑥𝑥 ) − Tr 𝛿𝚺𝑥 ℎ 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 − Tr 𝚺𝑥 ℎ 𝛿 𝚺 ℎ ℎ 𝚺 ℎ𝑥 − Tr 𝚺𝑥 ℎ 𝚺 ℎ ℎ 𝛿𝚺 ℎ𝑥





= Tr 𝑪 Δ𝚺𝜉 𝑪 ⊤ − Tr 𝑪 Δ𝚺𝜉 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥







−1 −1 ⊤ − Tr 𝚺𝑥 ℎ −𝚺−1 ℎ ℎ Δ𝚺𝜉 𝚺 ℎ ℎ 𝚺 ℎ𝑥 − Tr 𝚺𝑥 ℎ 𝚺 ℎ ℎ Δ𝚺𝜉 𝑪





= Tr 𝑪 Δ𝚺𝜉 𝑪 ⊤ − Tr 𝑪 Δ𝚺𝜉 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥





−1 −1 ⊤ + Tr 𝚺𝑥 ℎ 𝚺−1 . ℎ ℎ Δ𝚺𝜉 𝚺 ℎ ℎ 𝚺 ℎ𝑥 − Tr 𝚺𝑥 ℎ 𝚺 ℎ ℎ Δ𝚺𝜉 𝑪





7 In practice, we can enforce this by adding a small 𝜏𝑰 term to 𝚺 . 𝜉

31





Using cyclicity of the trace to move Δ𝚺𝜉 to the left in each term, 𝛿ℒ★ = Tr Δ𝚺𝜉 𝑪 ⊤ 𝑪 − Tr Δ𝚺𝜉 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 𝑪





−1 ⊤ −1 + Tr Δ𝚺𝜉 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 𝚺 𝑥 ℎ 𝚺 ℎ ℎ − Tr Δ𝚺𝜉 𝑪 𝚺𝑥 ℎ 𝚺 ℎ ℎ





−1 −1 ⊤ −1 = Tr Δ𝚺𝜉 𝑪 ⊤ 𝑪 − 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥 𝑪 + 𝚺 ℎ ℎ 𝚺 ℎ𝑥 𝚺𝑥 ℎ 𝚺 ℎ ℎ − 𝑪 𝚺𝑥 ℎ 𝚺 ℎ ℎ





.

Now define 𝑴 B 𝑪 − 𝚺𝑥 ℎ 𝚺−1 ℎℎ. Then

⊤

𝑪 − 𝚺𝑥 ℎ 𝚺−1 ℎℎ



= 𝑪 ⊤ − 𝚺−1 ℎ ℎ 𝚺 ℎ𝑥

𝑪 − 𝚺𝑥 ℎ 𝚺−1 ℎℎ



𝑴 ⊤ 𝑴 = 𝑪 − 𝚺𝑥 ℎ 𝚺−1 ℎℎ



−1 −1 −1 = 𝑪 ⊤ 𝑪 − 𝑪 ⊤ 𝚺𝑥 ℎ 𝚺−1 ℎ ℎ − 𝚺 ℎ ℎ 𝚺 ℎ𝑥 𝑪 + 𝚺 ℎ ℎ 𝚺 ℎ𝑥 𝚺𝑥 ℎ 𝚺 ℎ ℎ ,

which matches the bracket above. Hence 𝛿ℒ★ (𝚺𝑠 , 𝚺𝜉 ) = Tr Δ𝚺𝜉 𝑴 ⊤ 𝑴 .



Since Δ𝚺𝜉 ⪰ 0 and 𝑴 ⊤ 𝑴 ⪰ 0, the product inside the trace is positive semidefinite and hence 𝛿ℒ★ (𝚺𝑠 , 𝚺𝜉 ) = Tr Δ𝚺𝜉 𝑴 ⊤ 𝑴 ≥ 0.



Therefore, consider 𝚺𝜉,1 , 𝚺𝜉,2 ⪰ 0 with 𝚺𝜉,1 ⪯ 𝚺𝜉,2 , and define 𝚺𝜉 (𝑡) B 𝚺𝜉,1 + 𝑡 (𝚺𝜉,2 − 𝚺𝜉,1 ) ,

𝑡 ∈ [0, 1],

and 𝜙(𝑡) = ℒ★ (𝚺𝑠 , 𝚺𝜉 (𝑡)). Then 𝜙 ′ (𝑡) = Tr (𝚺𝜉,2 − 𝚺𝜉,1 ) 𝑴 ⊤ 𝑴 ≥ 0,



i.e., 𝜙(𝑡) is non-decreasing in 𝑡 and therefore ℒ★(𝚺𝑠 , 𝚺𝜉,1 ) ≤ ℒ★(𝚺𝑠 , 𝚺𝜉,2 ). Now we consider ICR. Fix 𝑽 as in the statement and define

e 𝚺𝑠 B 𝑽 ⊤ 𝚺𝑠 𝑽 ,

e 𝚺𝜉 B 𝑽 ⊤ 𝚺𝜉 𝑽 .

By cyclicity of the trace, ICR(𝚺𝑠 , 𝚺𝜉 ) =

Tr(e 𝚺𝜉 ) Tr(𝑽 ⊤ 𝚺𝜉 𝑽 ) = . ⊤ Tr(𝑽 (𝚺𝑠 + 𝚺𝜉 )𝑽 ) Tr(e 𝚺𝑠 + e 𝚺𝜉 )

Fix 𝚺𝑠 and consider the same path 𝚺𝜉 (𝑡) = 𝚺𝜉,1 + 𝑡(𝚺𝜉,2 − 𝚺𝜉,1 ), and set

e 𝚺𝜉 (𝑡) B 𝑽 ⊤ 𝚺𝜉 (𝑡)𝑽 = 𝑽 ⊤ 𝚺𝜉,1𝑽 + 𝑡 𝑽 ⊤ (𝚺𝜉,2 − 𝚺𝜉,1 )𝑽 , 𝛼(𝑡) B Tr(e 𝚺𝜉 (𝑡)),

𝛽 B Tr(e 𝚺𝑠 ) > 0.

32

Then ICR(𝑡) B ICR(𝚺𝑠 , 𝚺𝜉 (𝑡)) =

𝛼(𝑡) . 𝛼(𝑡) + 𝛽

Differentiating, 𝛼′ (𝑡) = Tr 𝑽 ⊤ (𝚺𝜉,2 − 𝚺𝜉,1 )𝑽 = Tr (𝚺𝜉,2 − 𝚺𝜉,1 ) 𝑽 𝑽 ⊤ ≥ 0,





because 𝚺𝜉,2 − 𝚺𝜉,1 ⪰ 0 and 𝑽 𝑽 ⊤ ≻ 0 imply





Tr (𝚺𝜉,2 − 𝚺𝜉,1 ) 𝑽 𝑽 ⊤ = Tr (𝑽 𝑽 ⊤ )1/2 (𝚺𝜉,2 − 𝚺𝜉,1 )(𝑽 𝑽 ⊤ )1/2 ≥ 0.



Hence ICR′ (𝑡) =

𝛽 𝛼′ (𝑡) ≥ 0, (𝛼(𝑡) + 𝛽)2

so ICR(𝑡) is non-decreasing in 𝑡 and we obtain ICR(𝚺𝑠 , 𝚺𝜉,1 ) ≤ ICR(𝚺𝑠 , 𝚺𝜉,2 ). which completes the proof.

33

Record · ID 267652 · SHA-256 1797c7c5a5dc82d3
Retrieved via Conceptio — every document is proof-bundled with source, license, and retrieval metadata.