Variance Covariance Regularization Enforces Pairwise Independence in Self-Supervised Representations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Mialon, Grégoire, Balestriero, Randall, LeCun, Yann
Natura: Preprint
Pubblicazione: 2022
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929242834468864
author Mialon, Grégoire
Balestriero, Randall
LeCun, Yann
author_facet Mialon, Grégoire
Balestriero, Randall
LeCun, Yann
contents Self-Supervised Learning (SSL) methods such as VICReg, Barlow Twins or W-MSE avoid collapse of their joint embedding architectures by constraining or regularizing the covariance matrix of their projector's output. This study highlights important properties of such strategy, which we coin Variance-Covariance regularization (VCReg). More precisely, we show that {\em VCReg combined to a MLP projector enforces pairwise independence between the features of the learned representation}. This result emerges by bridging VCReg applied on the projector's output to kernel independence criteria applied on the projector's input. We empirically validate our findings where (i) we put in evidence which projector's characteristics favor pairwise independence, (ii) we demonstrate pairwise independence to be beneficial for out-of-domain generalization, (iii) we demonstrate that the scope of VCReg goes beyond SSL by using it to solve Independent Component Analysis. This provides the first theoretical motivation and explanation of MLP projectors in SSL.
format Preprint
id arxiv_https___arxiv_org_abs_2209_14905
institution arXiv
publishDate 2022
record_format arxiv
spellingShingle Variance Covariance Regularization Enforces Pairwise Independence in Self-Supervised Representations
Mialon, Grégoire
Balestriero, Randall
LeCun, Yann
Machine Learning
Self-Supervised Learning (SSL) methods such as VICReg, Barlow Twins or W-MSE avoid collapse of their joint embedding architectures by constraining or regularizing the covariance matrix of their projector's output. This study highlights important properties of such strategy, which we coin Variance-Covariance regularization (VCReg). More precisely, we show that {\em VCReg combined to a MLP projector enforces pairwise independence between the features of the learned representation}. This result emerges by bridging VCReg applied on the projector's output to kernel independence criteria applied on the projector's input. We empirically validate our findings where (i) we put in evidence which projector's characteristics favor pairwise independence, (ii) we demonstrate pairwise independence to be beneficial for out-of-domain generalization, (iii) we demonstrate that the scope of VCReg goes beyond SSL by using it to solve Independent Component Analysis. This provides the first theoretical motivation and explanation of MLP projectors in SSL.
title Variance Covariance Regularization Enforces Pairwise Independence in Self-Supervised Representations
topic Machine Learning
url https://arxiv.org/abs/2209.14905