About contrastive unsupervised representation learning for classification and its convergence

Abstract

Contrastive representation learning has been recently proved to be very efficient for self-supervised 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

References in corpus (8)

• Distributed Representations of Words and Phrases and their Compositionality
• A Simple Framework for Contrastive Learning of Visual Representations
• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Unsupervised Learning of Visual Features by Contrasting Cluster Assignments
• Data-Efficient Image Recognition with Contrastive Predictive Coding
• A Convergence Theory for Deep Learning via Over-Parameterization
• Learning Video Representations using Contrastive Bidirectional Transformer
• A Theoretical Analysis of Contrastive Unsupervised Representation Learning