On the duality between contrastive and non-contrastive self-supervised learning - Archive ouverte HAL Accéder directement au contenu
Pré-Publication, Document De Travail Année : 2022

On the duality between contrastive and non-contrastive self-supervised learning

Résumé

Recent approaches in self-supervised learning of image representations can be categorized into different families of methods and, in particular, can be divided into contrastive and non-contrastive approaches. While differences between the two families have been thoroughly discussed to motivate new approaches, we focus more on the theoretical similarities between them. By designing contrastive and covariance based non-contrastive criteria that can be related algebraically and shown to be equivalent under limited assumptions, we show how close those families can be. We further study popular methods and introduce variations of them, allowing us to relate this theoretical result to current practices and show the influence (or lack thereof) of design choices on downstream performance. Motivated by our equivalence result, we investigate the low performance of SimCLR and show how it can match VICReg's with careful hyperparameter tuning, improving significantly over known baselines. We also challenge the popular assumptions that contrastive and non-contrastive methods, respectively, need large batch sizes and output dimensions. Our theoretical and quantitative results suggest that the numerical gaps between contrastive and non-contrastive methods in certain regimes can be closed given better network design choices and hyperparameter tuning. The evidence shows that unifying different SOTA methods is an important direction to build a better understanding of self-supervised learning.
Fichier principal
Vignette du fichier
main.pdf (619.86 Ko) Télécharger le fichier
Origine : Fichiers produits par l'(les) auteur(s)

Dates et versions

hal-03685169 , version 1 (02-06-2022)
hal-03685169 , version 2 (02-10-2022)
hal-03685169 , version 3 (19-06-2023)

Identifiants

Citer

Quentin Garrido, Yubei Chen, Adrien Bardes, Laurent Najman, Yann Lecun. On the duality between contrastive and non-contrastive self-supervised learning. 2022. ⟨hal-03685169v2⟩
296 Consultations
977 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More