Collapse, Entropy, and Sketching: SSL as Latent Distribution Matching

1. One problem, many fixes

Joint-embedding SSL starts with a simple wish: two related views π‘₯ and π‘₯β€² should have similar representations. If

𝑧 = 𝑓 πœ™ ( π‘₯ ) , 𝑧 β€² = 𝑓 πœ™ ( π‘₯ β€² ) ,

then we want π‘§β‰ˆπ‘§β€². By itself this collapses. The encoder can set π‘“πœ™(π‘₯)=𝑐 for every input, and every positive pair will be perfectly aligned.

The field has many ways to prevent this:

The usual summary is alignment and uniformity (Wang & Isola, 2020): pull positives together, then spread the representation out. Latent distribution matching (LDM) (Mikulasch & Zenke, 2026) makes that picture more precise. SSL is choosing a desired latent geometry, then learning an encoder whose induced latent distribution matches it.

2. The objective

Start with ordinary maximum likelihood:

𝔼 𝑃 data ( π‘₯ ) [ log 𝑃 πœƒ ( π‘₯ ) ] .

This says: draw real data π‘₯ from 𝑃data, ask how much probability the model π‘ƒπœƒ assigns to it, and average.

Now introduce an encoder:

𝑧 = 𝑓 πœ™ ( π‘₯ ) .

This creates a new distribution in latent space. If π‘₯βˆΌπ‘ƒdata, then 𝑧=π‘“πœ™(π‘₯) has an induced distribution π‘„πœ™(𝑧). The three objects are:

  1. 𝑃data(π‘₯): the real data distribution. We have samples, not a formula.
  2. π‘„πœ™(𝑧): the embedding distribution produced by the current encoder.
  3. π‘ƒπœƒ(𝑧): the latent model we want the embeddings to match.

For an invertible change of variables, likelihood in data space can be rewritten as latent likelihood plus latent entropy:

𝔼 𝑃 data ( π‘₯ ) [ log 𝑃 πœƒ ( π‘₯ ) ] ∝ 𝔼 𝑄 πœ™ [ log 𝑃 πœƒ ( 𝑧 ) ] ⏟ latent likelihood + 𝐻 𝑄 πœ™ [ 𝑧 ] ⏟ latent entropy = βˆ’ 𝐷 KL ( 𝑄 πœ™ ( 𝑧 ) βˆ₯ 𝑃 πœƒ ( 𝑧 ) ) .

The likelihood term says encoded samples should land where the latent model has high probability. The entropy term says the encoder should not fold many data points onto the same latent point.

This is why ICA is the right analogy (HyvΓ€rinen & Oja, 2000). Linear ICA observes mixtures π‘₯=𝐴𝑠 and learns

𝑧 = π‘Š π‘₯

so that 𝑧 looks like independent sources:

𝑃 πœƒ ( 𝑧 ) = ∏ 𝑖 𝑃 πœƒ ( 𝑧 𝑖 ) .

ICA chooses a source model and learns a transformation whose induced distribution matches it. SSL does the same thing with a deep encoder and paired views.

For SSL, the objects are pairs:

( π‘₯ , π‘₯ β€² ) β†’ ( 𝑧 , 𝑧 β€² ) = ( 𝑓 πœ™ ( π‘₯ ) , 𝑓 πœ™ ( π‘₯ β€² ) ) .

These pairs induce π‘„πœ™(𝑧,𝑧′). We choose a latent pair model

𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) = 𝑃 πœƒ ( 𝑧 ) 𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) ,

and train the encoder so that

𝑄 πœ™ ( 𝑧 , 𝑧 β€² ) β‰ˆ 𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) .

The LDM objective is

β„±οΈ€ LDM = βˆ’ 𝐷 KL ( 𝑄 πœ™ ( 𝑧 , 𝑧 β€² ) βˆ₯ 𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) ) = 𝔼 𝑄 πœ™ [ log 𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) ] ⏟ alignment + 𝐻 𝑄 πœ™ [ 𝑧 , 𝑧 β€² ] ⏟ uniformity .

The split is natural. Since

log 𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) = log 𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) + log 𝑃 πœƒ ( 𝑧 ) ,

the conditional π‘ƒπœƒ(𝑧′|𝑧) defines how related views should align, while the marginal π‘ƒπœƒ(𝑧) and entropy term define the global shape of the embedding cloud. This is the main idea: choose a latent geometry, then learn π‘“πœ™ to realize it.

2.1. Mutual information is not the main point

Many SSL papers describe the goal as maximizing mutual information between views (Oord et al., 2018) (Shwartz-Ziv et al., 2023). LDM says that is not the central force. If we add MI to the objective, we get

β„±οΈ€ MI = β„±οΈ€ LDM + 𝐼 𝑄 πœ™ [ 𝑧 , 𝑧 β€² ] = 𝔼 𝑄 πœ™ [ log 𝑃 πœƒ ( 𝑧 , 𝑧 β€² ) ] + 2 𝐻 𝑄 πœ™ [ 𝑧 ] ,

assuming both views have the same marginal entropy.

MI cannot choose the representation geometry. If 𝑧̃=𝑔(𝑧) is an invertible reparameterization, then

𝐼 ( 𝑧 , 𝑧 β€² ) = 𝐼 ( 𝑧 Μƒ , 𝑧 β€² Μƒ ) .

So MI can say information is preserved, but not whether classes, positions, or dynamics are linearly exposed. In (Mikulasch & Zenke, 2026), turning MI on or off barely changes linear-probe accuracy, while changing the entropy estimator moves accuracy by much more.

3. Axis 1: the latent model

The latent model is the inductive bias. Different choices of π‘ƒπœƒ(𝑧′|𝑧) give different alignment losses.

  • Gaussian conditional. Choose

    𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) = 𝒩︀ ( 𝑧 β€² ; 𝑧 , 𝜎 2 𝐼 ) .

    Then

    log 𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) = βˆ’ 1 2 𝜎 2 β€– 𝑧 β€² βˆ’ 𝑧 β€– 2 + 𝐢 ,

    which gives the MSE invariance loss of VICReg.

  • von Mises-Fisher conditional. Put latents on the unit sphere and choose

    𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) ∝ exp ( 𝛽 𝑧 ⊀ 𝑧 β€² ) .

    The log-likelihood is cosine similarity, which gives SimCLR’s positive-pair term.

  • Dirichlet conditional. Put latents on the simplex and choose

    𝑃 πœƒ ( 𝑧 β€² | 𝑧 ) = Dir ( 𝑧 β€² ; 𝜏 𝑧 + 1 ) .

    The log-likelihood contains πœβˆ‘π‘˜π‘§π‘˜logπ‘§π‘˜β€², which gives the teacher-student cross-entropy used by DINO-like methods.

  • Predictive conditional. For time or masked prediction, choose

    𝑃 πœƒ ( 𝑧 𝑑 | 𝑧 : 𝑑 ) = 𝒩︀ ( 𝑧 𝑑 ; 𝑝 πœƒ ( 𝑧 : 𝑑 ) , Ξ£ ) .

    This gives the JEPA family (Assran et al., 2023) (Bardes et al., 2024): predict the next or missing latent, not the next or missing pixels.

So SSL is not mainly compression. It is a way to choose a coordinate system on the data manifold. The latent model says which geometry we want.

4. Axis 2: the uniformity mechanism

Alignment alone collapses, so each method also needs a way to make π‘„πœ™(𝑧) spread out. In LDM this is the entropy or marginal-matching side.

Uniformity mechanism What it does Methods
KDE / contrastive density estimate βˆ’1π‘›βˆ‘π‘–logβˆ‘π‘—exp(π›½π‘§π‘–βŠ€π‘§π‘—) repels crowded samples SimCLR, CPC
Gaussian entropy via log|Σ𝑧| increase variance and reduce covariance VICReg, Barlow Twins
Conditional entropy plug-in use predictor plus stop-gradient BYOL, SimSiam, JEPA
Sliced distribution test match random projections to 𝒩︀(0,𝐼) SIGReg, VISReg

SimCLR’s negatives can be read as a kernel density entropy estimator. If embeddings crowd together, the local density estimate is high and entropy is low, so the loss pushes points apart.

VICReg approximates Gaussian entropy:

𝐻 𝑄 πœ™ [ 𝑧 ] ∝ log | Ξ£ 𝑧 | .

Near a diagonal covariance,

log | Ξ£ 𝑧 | β‰ˆ βˆ‘ 𝑖 log Ξ£ 𝑖 𝑖 βˆ’ 1 2 βˆ‘ 𝑖 β‰  𝑗 Ξ£ 𝑖 𝑗 Ξ£ 𝑗 𝑖 Ξ£ 𝑖 𝑖 Ξ£ 𝑗 𝑗 .

This is why VICReg has a variance term and a covariance term (Shwartz-Ziv et al., 2023). The weakness is also visible: covariance only sees second-order statistics.

BYOL, SimSiam, and JEPA use predictor plus stop-gradient. In predictive LDM, the target contains

𝔼 [ log 𝑃 πœƒ ( 𝑧 𝑑 | 𝑧 : 𝑑 ) ] + 𝐻 𝑄 πœ™ [ 𝑧 𝑑 | 𝑧 : 𝑑 ] .

The stop-gradient objective approximates the conditional entropy term:

𝔼 [ log 𝑃 πœƒ ( 𝑧 𝑑 | 𝑧 : 𝑑 ) ] βˆ’ 𝔼 [ log 𝑃 SG [ πœƒ ] ( 𝑧 𝑑 | SG [ 𝑧 : 𝑑 ] ) ] .

This makes the usual predictor, target network, and stop-gradient less mysterious. They are not just asymmetry tricks. They make a conditional entropy estimate usable.

DINO fits the alignment side cleanly through teacher-student cross-entropy on the simplex. Its anti-collapse side, centering plus sharpening, is less clearly an entropy estimator. This is why the LDM paper treats it as more heuristic.

5. From implicit entropy to explicit sketching

If uniformity means matching the marginal π‘„πœ™(𝑧), then we can skip entropy estimation and directly test the distribution. LeJEPA (Balestriero & LeCun, 2025) chooses

𝑃 ( 𝑧 ) = 𝒩︀ ( 0 , 𝐼 ) .

SIGReg checks this with random one-dimensional projections and an Epps-Pulley normality test (Epps & Pulley, 1983). By the Cramer-Wold theorem (CramΓ©r & Wold, 1936), matching all one-dimensional projections matches the full distribution.

VISReg (Wu et al., 2026) keeps the Gaussian target but separates scale, center, and shape. With centered embeddings π‘Μ‚βˆˆβ„π‘Γ—π·:

β„’οΈ€ scale = 1 𝐷 βˆ‘ 𝑗 = 1 𝐷 ( 1 βˆ’ 𝜎 𝑗 ( 𝑍 Μ‚ ) ) 2 , β„’οΈ€ center = β€– πœ‡ β€– 2 2 .

Then it removes scale before testing shape:

𝑍 Μƒ = 𝑍 Μ‚ / ( sg ( 𝜎 ) + πœ€ ) ,

and matches random projections to standard-normal quantiles:

β„’οΈ€ shape = 1 𝐾 βˆ‘ π‘˜ = 1 𝐾 β€– sort ( 𝑍 Μƒ 𝑀 π‘˜ ) βˆ’ π‘ž 𝒩︀ β€– 2 2 .

The full loss is

β„’οΈ€ = ( 1 βˆ’ πœ† ) β„’οΈ€ pred + πœ† ( πœ† scale β„’οΈ€ scale + πœ† shape β„’οΈ€ shape + πœ† center β„’οΈ€ center ) .

Compared with SIGReg, VISReg keeps a stronger gradient near collapse and lets scale and shape be reweighted separately. Compared with VICReg, it tests more than covariance.

6. What this buys

The first payoff is identifiability. With a predictive Gaussian or vMF conditional, LDM can recover the true latent variables up to an affine transformation under the paper’s assumptions (Mikulasch & Zenke, 2026). The source of identifiability is the conditional noise model, not the architecture.

The second payoff is robustness. VISReg reports that explicit regularization methods remain stable on long-tailed ImageNet-LT and low-rank Galaxy10, while DINO can collapse on the long-tailed setting (Wu et al., 2026). VISReg also benefits from its scale-shape split: hard data can be handled by increasing the shape weight.

The tradeoff is that global marginal matching does not directly constrain spatial token structure. That likely explains why dense prediction is still a gap compared with methods such as MoCoV3 and iBOT.

7. A design recipe

The LDM view turns SSL design into a short checklist:

  1. Choose π‘ƒπœƒ(𝑧′|𝑧), which fixes the alignment loss and the inductive bias.
  2. Choose a uniformity mechanism, and ask whether it estimates only covariance, whether its gradient survives collapse, and what it costs.
  3. Separate scale from shape when possible.
  4. Do not expect MI to choose the geometry. Entropy and explicit marginal matching are doing that work.

Seen this way, SSL is close in spirit to ICA. Pick a latent model, keep the encoder informative on the data manifold, and match distributions. The old anti-collapse toolbox, negatives, EMA teachers, stop-gradients, covariance penalties, is a collection of entropy estimators of uneven quality. VISReg replaces part of that implicit machinery with an explicit test of the embedding distribution.

References

  1. Chen, T., Kornblith, S., Norouzi, M., and Hinton, G. E. A Simple Framework for Contrastive Learning of Visual Representations. International Conference on Machine Learning (ICML), 2020.
  2. HyvΓ€rinen, A., and Oja, E. Independent component analysis: algorithms and applications. Neural Networks, 2000.
  3. Bardes, A., Ponce, J., and LeCun, Y. VICReg: Variance-Invariance-Covariance Regularization for Self-Supervised Learning. International Conference on Learning Representations (ICLR), 2022.
  4. Zbontar, J., Jing, L., Misra, I., LeCun, Y., and Deny, S. Barlow Twins: Self-Supervised Learning via Redundancy Reduction. International Conference on Machine Learning (ICML), 2021.
  5. HyvΓ€rinen, A., Sasaki, H., and Turner, R. E. Nonlinear ICA Using Auxiliary Variables and Generalized Contrastive Learning. International Conference on Artificial Intelligence and Statistics (AISTATS), 2019.
  6. Chen, X., and He, K. Exploring Simple Siamese Representation Learning. IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2021.
  7. Zimmermann, R. S., Sharma, Y., Schneider, S., Bethge, M., and Brendel, W. Contrastive Learning Inverts the Data Generating Process. International Conference on Machine Learning (ICML), 2021.
  8. Mikulasch, F. A., and Zenke, F. Understanding Self-Supervised Learning via Latent Distribution Matching, 2026.
  9. Shwartz-Ziv, R., Balestriero, R., Kawaguchi, K., Rudner, T. G. J., and LeCun, Y. An Information Theory Perspective on Variance-Invariance-Covariance Regularization. Advances in Neural Information Processing Systems (Neurips), 2023.
  10. Balestriero, R., and LeCun, Y. LeJEPA: Provable and Scalable Self-Supervised Learning Without the Heuristics, 2025.
  11. Grill, J.-B., Strub, F., Altché, F., Tallec, C., Richemond, P. H., Buchatskaya, E., Doersch, C., Pires, B. Á., Guo, Z., Azar, M. G., Piot, B., Kavukcuoglu, K., Munos, R., and Valko, M. Bootstrap Your Own Latent - A New Approach to Self-Supervised Learning. Advances in Neural Information Processing Systems (Neurips), 2020.
  12. Assran, M., Duval, Q., Misra, I., Bojanowski, P., Vincent, P., Rabbat, M., LeCun, Y., and Ballas, N. Self-Supervised Learning from Images with a Joint-Embedding Predictive Architecture. IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2023.
  13. Caron, M., Touvron, H., Misra, I., Jegou, H., Mairal, J., Bojanowski, P., and Joulin, A. Emerging Properties in Self-Supervised Vision Transformers. IEEE/CVF International Conference on Computer Vision (ICCV), 2021.
  14. Wang, T., and Isola, P. Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere. International Conference on Machine Learning (ICML), 2020.
  15. Wu, H., Balestriero, R., and Levine, M. VISReg: Variance-Invariance-Sketching Regularization for JEPA training, 2026.
  16. Tschannen, M., Djolonga, J., Rubenstein, P. K., Gelly, S., and Lucic, M. On Mutual Information Maximization for Representation Learning. International Conference on Learning Representations (ICLR), 2020.
  17. CramΓ©r, H., and Wold, H. Some Theorems on Distribution Functions. Journal of the London Mathematical Society, 1936.
  18. Bardes, A., Garrido, Q., Ponce, J., Chen, X., Rabbat, M., LeCun, Y., Assran, M., and Ballas, N. Revisiting Feature Prediction for Learning Visual Representations from Video. Transactions on Machine Learning Research (TMLR), 2024.
  19. Oord, A. van den, Li, Y., and Vinyals, O. Representation Learning with Contrastive Predictive Coding, 2018.
  20. Epps, T. W., and Pulley, L. B. A test for normality based on the empirical characteristic function. Biometrika, 1983.