arXiv · 2501.13031
A Probabilistic Model for Non-Contrastive Learning
Abstract
Self-supervised learning (SSL) aims to find meaningful representations from unlabeled data by encoding semantic similarities through data augmentations. Despite its current popularity, theoretical insights about SSL are still scarce. For example, it is not yet known whether commonly used SSL loss functions can be related to a statistical model, much in the same as OLS, generalized linear models or PCA naturally emerge as maximum likelihood estimates of an underlying generative process. In this short paper, we consider a latent variable statistical model for SSL that exhibits an interesting property: Depending on the informativeness of the data augmentations, the MLE of the model either reduces to PCA, or approaches a simple non-contrastive loss. We analyze the model and also empirically illustrate our findings.
Explore related subjects
Keep this discovery
Maximilian Fleissner, Pascal Esser, Debarghya Ghoshdastidar. 2025-01-22. A Probabilistic Model for Non-Contrastive Learning. https://arxiv.org/abs/2501.13031
Cite the original work for its findings. Save a collection to share your selection of sources.