Papers
Topics
Authors
Recent
Search
2000 character limit reached

Covariance Collapse in Representation Learning

Updated 4 July 2026
  • Covariance collapse is the degeneration of second-order feature statistics, where many eigenvalues become near-zero and the effective rank is significantly reduced.
  • Regularization techniques like VICReg and VCReg counteract collapse by maintaining high variance and reducing redundant feature correlations, thus improving transfer performance.
  • This phenomenon is observed across multiple settings—from self-supervised models and supervised transfer to text embeddings and stochastic processes—with distinct diagnostic criteria for each case.

Covariance collapse denotes a family of degeneracy phenomena in which second-order structure becomes low-rank, highly correlated, or otherwise uninformative. In joint-embedding and supervised representation learning, it refers to embeddings whose sample covariance matrix has many near-zero eigenvalues or whose features occupy only a tiny subspace of the available representation space (Mialon et al., 2022, Zhu et al., 2023). Related formulations quantify within-class covariance collapse through the NC1 ratio in neural collapse (Kothapalli et al., 2024), second-order information loss under mean pooling in text encoders (Hara et al., 30 Apr 2026), and rank-one concentration of quadratic covariation matrices in high-dimensional semimartingales (Petkevicius, 24 Jun 2026). These usages are not identical, but they all concern a loss of informative diversity in second-order statistics.

1. Definitions and diagnostic criteria

Recent work uses closely related but non-equivalent notions of covariance collapse across several settings.

Setting Second-order object Collapse criterion
Joint-embedding SSL Cov(Z)\operatorname{Cov}(Z) rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D or many λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 0 (Mialon et al., 2022)
Supervised penultimate features Feature covariance across a batch Many dimensions have near-zero variance; surviving dimensions become highly correlated (Zhu et al., 2023)
Neural collapse ΣW(H)\Sigma_W(H) and ΣB(H)\Sigma_B(H) NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 0 (Kothapalli et al., 2024)
Mean-pooled text embeddings Token covariance Σ\Sigma μ1μ2\mu_1\approx\mu_2 but Σ1Σ2\Sigma_1\neq\Sigma_2, yielding high SOCM (Hara et al., 30 Apr 2026)
High-dimensional semimartingales Quadratic covariation QnQ_n rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D0 and rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D1 (Petkevicius, 24 Jun 2026)

For a batch of features rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D2, covariance collapse in SSL is formalized by the conditions rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D3 or many eigenvalues rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D4 (Mialon et al., 2022). In supervised training, a more refined decomposition distinguishes variance collapse, where many feature dimensions acquire near-zero variance across a batch, from covariance collapse, where the remaining non-degenerate dimensions become highly correlated with one another (Zhu et al., 2023).

The neural-collapse literature isolates within-class covariance collapse through the metric

rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D5

where rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D6 is the within-class covariance and rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D7 is the between-class covariance (Kothapalli et al., 2024). In the original neural-collapse description, within-class features collapse to their class mean, while class means become equi-angular and maximal-distance apart; this structure yields zero training error but destroys feature diversity and harms transfer (Zhu et al., 2023).

A distinct but related definition appears in text embedding models that use mean pooling. There, second-order collapse means that two token clouds with nearly identical means but different covariances map to nearly identical text embeddings, because mean pooling discards the covariance structure (Hara et al., 30 Apr 2026). In stochastic-process settings, collapse is instead spectral: a quadratic covariation matrix becomes effectively rank one when its top eigenvalue dominates the trace and the effective rank converges to one (Petkevicius, 24 Jun 2026).

2. Regularization mechanisms: VICReg and VCReg

The principal anti-collapse construction in self-supervised learning is VICReg, which combines an invariance loss with separate variance and covariance penalties. For a batch rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D8, the covariance term is

rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D9

where

λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 00

The variance term is a hinge on each coordinate’s standard deviation with target λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 01. Ablations show that “Inv only” collapses, “Inv+Cov” collapses, “Inv+Var” reaches λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 02 top-1, and “Inv+Var+Cov” reaches λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 03 top-1, establishing that covariance regularization alone does not prevent norm-collapse and variance preservation alone does not prevent rank-1 informational collapse (Bardes et al., 2021).

The supervised adaptation is VCReg, which omits the invariance term and applies the two statistical penalties directly to supervised representations. Writing a batch of representations as λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 04 with covariance matrix λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 05,

λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 06

and the supervised objective is

λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 07

or, when applied to multiple layers,

λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 08

Empirically, VCReg layers are inserted after each block or residual rather than only on the final feature; for convolutional features of shape λi(Cov(Z))0\lambda_i(\operatorname{Cov}(Z))\to 09, each channel is demeaned over batchΣW(H)\Sigma_W(H)0spatial locations and each spatial location is treated as an independent sample; the squared-entry penalty ΣW(H)\Sigma_W(H)1 is replaced by a smooth-L1 variant to guard against unstable gradients from rare large covariances; and a backward hook computes covariance and variance only in the backward pass. This optimized implementation is only ΣW(H)\Sigma_W(H)2 slower than identity and comparable to a BatchNorm in wall-clock time (Zhu et al., 2023).

The underlying mechanism is explicit. The high-variance term keeps each axis alive by pushing each dimension’s standard deviation above ΣW(H)\Sigma_W(H)3, while the low-covariance term discourages redundant features and forces the model to spread information across dimensions. In the VICReg formulation, this blocks rank-1 solutions in which all coordinates track the same scalar; in the VCReg formulation, it is proposed as a general regularization framework for supervised transfer learning (Bardes et al., 2021, Zhu et al., 2023).

3. Neural collapse, gradient starvation, and transferability

In supervised deep networks, covariance collapse is treated as one component of a broader representational degeneracy that includes neural collapse and gradient starvation. The account developed for VCReg states that a network trained only to minimize cross-entropy can focus greedily on a minimal set of source-task features, causing within-class collapse, covariance collapse, and poor transfer to downstream tasks. VCReg is reported to counter this by enforcing high-variance, low-covariance embeddings at intermediate layers, thereby keeping new inter-class directions available and ensuring that gradients exist for each feature dimension rather than allowing one “win” dimension to shut off the rest (Zhu et al., 2023).

The empirical record in that study is extensive. On ImageNet-supervised pretraining followed by linear probing across 9 datasets, ResNet-50 with VCReg improves the average by ΣW(H)\Sigma_W(H)4 points versus the baseline and beats DeCov and WLD-Reg by ΣW(H)\Sigma_W(H)5 points. ConvNeXt-T improves by ΣW(H)\Sigma_W(H)6 point and ViT-Base improves by ΣW(H)\Sigma_W(H)7 point. On HMDB51 video finetuning for VideoMAE and ViViT backbones, VCReg adds ΣW(H)\Sigma_W(H)8 point consistently. Collapse-oriented metrics move in the same direction: Class-Distance Normalized Variance rises from ΣW(H)\Sigma_W(H)9, Nearest-Class-Center accuracy falls from ΣB(H)\Sigma_B(H)0, and Mutual Information estimated by MINE increases from ΣB(H)\Sigma_B(H)1 bits. In the synthetic two-moon setting of Pezeshki et al., VCReg is reported as the only regularizer that learns the curved decision boundary rather than starving out all but a linear feature. Additional gains appear in long-tail CIFAR10-LT/CIFAR100-LT, with ΣB(H)\Sigma_B(H)2 and ΣB(H)\Sigma_B(H)3 points, in hierarchical subclass probing for ConvNeXt, where CIFAR100 superclass-to-subclass linear probing rises from ΣB(H)\Sigma_B(H)4 to ΣB(H)\Sigma_B(H)5, and in noisy HMDB51 finetuning under Gaussian corruption ΣB(H)\Sigma_B(H)6 (Zhu et al., 2023).

These results are presented as evidence for a three-way link: collapse, gradient starvation, and feature transferability. The paper states that neural collapse and gradient starvation both reflect a network’s greedy focus on a minimal feature set, that this produces a degenerate representation space, and that reversing the collapse metrics aligns nearly one-for-one with improved downstream transfer across images, videos, long-tail learning, and hierarchical tasks (Zhu et al., 2023).

4. Independence and kernel-based explanations

A more theoretical line of work analyzes covariance collapse through independence criteria. In self-supervised joint-embedding models, VCReg applied to the output of a wide MLP projector is shown to enforce pairwise independence between the features of the learned representation. The central result links the covariance penalty on the projector output ΣB(H)\Sigma_B(H)7 to Hilbert–Schmidt Independence Criterion terms on the projector input ΣB(H)\Sigma_B(H)8, implying that driving ΣB(H)\Sigma_B(H)9 toward the identity forces NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 00 for all NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 01 in the large-width limit. The construction relies on alternating random linear maps and elementwise nonlinearities such as BatchNorm and ReLU, with projector width NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 02 and weights remaining near random. Empirically, increasing projector width improves pairwise independence, adding depth beyond 3 layers can hurt HSIC, keeping the projector random or resampling it each step lowers HSIC relative to fully learned projectors, and the same mechanism suffices to solve linear ICA but fails in post-nonlinear ICA, consistent with the claim that VCReg enforces pairwise rather than higher-order independence (Mialon et al., 2022).

Kernel methods provide a complementary analysis of within-class covariance collapse. The NC1 ratio can be written entirely in terms of the Gram matrix NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 03, so that

NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 04

is computed from kernel evaluations alone. Specializing this expression to the NNGP and NTK of shallow infinite-width networks, one study shows that the NTK does not represent more collapsed features than the NNGP for Gaussian data of arbitrary dimensions. In the prototypical 1-D two-class Gaussian mixture model, the expected NC1 values coincide for the ReLU NNGP and the ReLU NTK, indicating that lazy training does not further decrease NC1 beyond initialization. The same study reports that a data-aware Gaussian Process kernel based on Equations of State yields lower NC1 than NNGP at smaller effective widths, and that the choice of nonlinearity matters: ERF yields lower NC1 than ReLU, both theoretically and empirically (Kothapalli et al., 2024).

Taken together, these analyses separate two questions that are often conflated. One concerns whether covariance regularization decorrelates features strongly enough to approach pairwise independence; the other concerns whether simplified kernel regimes can explain the data dependence of neural collapse. The available results support the first claim for wide random projectors and qualify the second by showing clear limitations of data-independent kernels such as NTK (Mialon et al., 2022, Kothapalli et al., 2024).

5. Structured manifestations in graphs and text embeddings

In graph neural networks, covariance collapse is partial rather than exact in the regimes studied so far. For node-wise classification on sparse stochastic block model graphs, the trace of the within-class covariance, NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 05, decreases steadily during the terminal phase of training but plateaus at a positive level, reported as approximately NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 06, rather than collapsing nearly to zero as in plain image classifiers. The same plateau behavior appears for both the raw features NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 07 and the neighborhood-aggregated features NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 08. An “optimistic” graph-UFM shows that exact collapse, NC1(H)=tr[ΣW(H)]/tr[ΣB(H)]0NC_1(H)=\operatorname{tr}[\Sigma_W(H)]/\operatorname{tr}[\Sigma_B(H)]\to 09, occurs only under a strict structural requirement: for every class Σ\Sigma0, the vector of neighbor fractions Σ\Sigma1 must be independent of node index Σ\Sigma2 within that class. Under exact collapse, the KKT conditions force the class means to form a simplex ETF. Gradient-flow analysis for Σ\Sigma3 and a single graph yields Σ\Sigma4 and Σ\Sigma5, but finite graph noise and Σ\Sigma6 make the collapse stall at a positive plateau (Kothapalli et al., 2023).

Text embedding models expose a different second-order failure mode. If a text encoder produces token embeddings Σ\Sigma7 with mean Σ\Sigma8 and covariance Σ\Sigma9, then mean pooling discards μ1μ2\mu_1\approx\mu_20 entirely. To quantify the resulting second-order collapse, the SOCM metric is defined by

μ1μ2\mu_1\approx\mu_21

where μ1μ2\mu_1\approx\mu_22 and μ1μ2\mu_1\approx\mu_23 are normalized first- and second-order components of the squared μ1μ2\mu_1\approx\mu_24-Wasserstein distance between Gaussian approximations to two token clouds. By construction, SOCM is maximal when means coincide and covariances differ, and vanishes when the means differ maximally or the covariances coincide. On 1,000 Wikipedia texts, covering all 499,500 pairwise comparisons, average SOCM drops from μ1μ2\mu_1\approx\mu_25 for BERT to μ1μ2\mu_1\approx\mu_26 for Unsup-SimCSE-mean, μ1μ2\mu_1\approx\mu_27 for E5_base, and μ1μ2\mu_1\approx\mu_28 for GTE_base; for MiniLM it drops from μ1μ2\mu_1\approx\mu_29 to Σ1Σ2\Sigma_1\neq\Sigma_20 for E5_small and Σ1Σ2\Sigma_1\neq\Sigma_21 for GTE_small, but rises to Σ1Σ2\Sigma_1\neq\Sigma_22 for all-MiniLM-L12-v2; for MPNet it drops from Σ1Σ2\Sigma_1\neq\Sigma_23 to Σ1Σ2\Sigma_1\neq\Sigma_24 for all-mpnet-base-v2; and for nomic-bert-2048 it drops from Σ1Σ2\Sigma_1\neq\Sigma_25 to Σ1Σ2\Sigma_1\neq\Sigma_26 for nomic-embed-text-v1.5. The same work proves that if each text’s normalized spread Σ1Σ2\Sigma_1\neq\Sigma_27, then Σ1Σ2\Sigma_1\neq\Sigma_28, and reports that average SOCM correlates with MTEB score with Spearman’s Σ1Σ2\Sigma_1\neq\Sigma_29 and QnQ_n0 (Hara et al., 30 Apr 2026).

A plausible implication is that “collapse” in structured domains is highly architecture-dependent. In GNNs, graph heterogeneity and finite-sample randomness obstruct exact within-class collapse except under a strict neighbor-fraction condition. In contrastive text encoders, mean pooling remains effective because token embeddings within each text become sufficiently concentrated that little second-order information remains to be lost (Kothapalli et al., 2023, Hara et al., 30 Apr 2026).

6. Spectral collapse in stochastic systems and terminological boundaries

Outside representation learning, covariance collapse appears as a spectral property of high-dimensional semimartingales. Let

QnQ_n1

be the quadratic covariation matrix, with diffusion component QnQ_n2 and jump component

QnQ_n3

Two diagnostics are used:

QnQ_n4

Spectral collapse means that both converge to QnQ_n5. The main theorem states that this occurs if and only if the jump directions are geometrically aligned in a weighted sense and the background diffusion is asymptotically negligible, specifically QnQ_n6 and QnQ_n7. In the stochastic extension, the same conclusion holds in probability under natural conditions, and a scalar diagnostic

QnQ_n8

tends to QnQ_n9 exactly when rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D00 grows unboundedly. The proposed operational interpretation is that values of rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D01 or rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D02 near rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D03 signal a rank-one extreme-event phase (Petkevicius, 24 Jun 2026).

A separate terminological issue arises in gravitational-collapse models inspired by loop quantum gravity. There, “covariance” refers to closure of the deformed constraint algebra and underlying rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D04 diffeomorphism invariance, not to covariance-matrix degeneracy. In the effective spherically symmetric dust model, one has

rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D05

which closes with the diffeomorphism constraint rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D06, and the theory remains fully rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D07 diffeo-invariant even when one later imposes the areal gauge rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D08. The same analysis argues that shock solutions proposed in some earlier phenomenological models are absent because the dynamics depend on rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D09 rather than rankCov(Z)D\operatorname{rank}\operatorname{Cov}(Z)\ll D10, so coordinate changes cannot create discontinuous shock layers across the bounce (Giesel et al., 2023).

This distinction is important because the phrase combines two overloaded terms. In machine learning and stochastic-process theory, covariance collapse concerns degeneration of second-order structure. In canonical gravity, covariance is a statement about symmetry and constraint closure during gravitational collapse, and the relevant question is whether an effective theory remains covariant across a quantum bounce (Giesel et al., 2023).

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Covariance Collapse.