Similarity contrastive estimation for image and video soft contrastive self-supervised learning - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue Machine Vision and Applications Année : 2023

Similarity contrastive estimation for image and video soft contrastive self-supervised learning

Résumé

Contrastive representation learning has proven to be an effective self-supervised learning method for images and videos. Most successful approaches are based on Noise Contrastive Estimation (NCE) and use different views of an instance as positives that should be contrasted with other instances, called negatives, that are considered as noise. However, several instances in a dataset are drawn from the same distribution and share underlying semantic information. A good data representation should contain relations between the instances, or semantic similarity and dissimilarity, that contrastive learning harms by considering all negatives as noise. To circumvent this issue, we propose a novel formulation of contrastive learning using semantic similarity between instances called Similarity Contrastive Estimation (SCE). Our training objective is a soft contrastive one that brings the positives closer and estimates a continuous distribution to push or pull negative instances based on their learned similarities. We validate empirically our approach on both image and video representation learning. We show that SCE performs competitively with the state of the art on the ImageNet linear evaluation protocol for fewer pretraining epochs and that it generalizes to several downstream image tasks. We also show that SCE reaches state-of-the-art results for pretraining video representation and that the learned representation can generalize to video downstream tasks. Source code is available here: https://github.com/juliendenize/eztorch.
Fichier principal
Vignette du fichier
s00138-023-01444-9.pdf (1.5 Mo) Télécharger le fichier
Origine : Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-04465089 , version 1 (19-02-2024)

Identifiants

Citer

Julien Denize, Jaonary Rabarisoa, Astrid Orcesi, Romain Hérault. Similarity contrastive estimation for image and video soft contrastive self-supervised learning. Machine Vision and Applications, 2023, 34 (6), pp.111. ⟨10.1007/s00138-023-01444-9⟩. ⟨hal-04465089⟩
16 Consultations
6 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More