Thumbnail

About contrastive unsupervised representation learning for classification and its convergence

Merad, Ibrahim; Yu, Yiyang; Bacry, Emmanuel; Gaïffas, Stéphane (2021), About contrastive unsupervised representation learning for classification and its convergence. https://basepub.dauphine.psl.eu/handle/123456789/22718

View/Open
Type
Document de travail / Working paper
External document link
https://hal.archives-ouvertes.fr/hal-03438767
Date
2021
Series title
Cahier de recherche CEREMADE, Université Paris Dauphine-PSL
Published in
Paris
Pages
17
Metadata
Show full item record
Author(s)
Merad, Ibrahim
Yu, Yiyang
Bacry, Emmanuel cc
CEntre de REcherches en MAthématiques de la DEcision [CEREMADE]
Gaïffas, Stéphane
Centre de Mathématiques Appliquées - Ecole Polytechnique [CMAP]
Laboratoire de Probabilités, Statistique et Modélisation [LPSM (UMR_8001)]
Abstract (EN)
Contrastive representation learning has been recently proved to be very efficient for selfsupervised training. These methods have been successfully used to train encoders which perform comparably to supervised training on downstream classification tasks. A few works have started to build a theoretical framework around contrastive learning in which guarantees for its performance can be proven. We provide extensions of these results to training with multiple negative samples and for multiway classification. Furthermore, we provide convergence guarantees for the minimization of the contrastive training error with gradient descent of an overparametrized deep neural encoder, and provide some numerical experiments that complement our theoretical findings.
Subjects / Keywords
unsupervised learning; constrative learning; deep neural networks; theoretical guarantees