ReSSL: Relational Self-Supervised Learning with Weak Augmentation
arXiv:2107.09282
Abstract
Self-supervised Learning (SSL) including the mainstream contrastive learning has achieved great success in learning visual representations without data annotations. However, most of methods mainly focus on the instance level information (\ie, the different augmented images of the same instance should have the same feature or cluster into the same class), but there is a lack of attention on the relationships between different instances. In this paper, we introduced a novel SSL paradigm, which we term as relational self-supervised learning (ReSSL) framework that learns representations by modeling the relationship between different instances. Specifically, our proposed method employs sharpened distribution of pairwise similarities among different instances as \textit{relation} metric, which is thus utilized to match the feature embeddings of different augmentations. Moreover, to boost the performance, we argue that weak augmentations matter to represent a more reliable relation, and leverage momentum strategy for practical efficiency. Experimental results show that our proposed ReSSL significantly outperforms the previous state-of-the-art algorithms in terms of both performance and training efficiency. Code is available at \url{https://github.com/KyleZheng1997/ReSSL}.
arXiv commentfixed several typos
References in corpus (19)
- Distilling the Knowledge in a Neural Network
- A Simple Framework for Contrastive Learning of Visual Representations
- Large Scale GAN Training for High Fidelity Natural Image Synthesis
- Bootstrap your own latent: A new approach to self-supervised Learning
- Improved Baselines with Momentum Contrastive Learning
- Unsupervised Learning of Visual Features by Contrasting Cluster Assignments
- FitNets: Hints for Thin Deep Nets
- SGDR: Stochastic Gradient Descent with Warm Restarts
- Momentum Contrast for Unsupervised Visual Representation Learning
- AutoAugment: Learning Augmentation Policies from Data
- Contrastive Multiview Coding
- Big Self-Supervised Models are Strong Semi-Supervised Learners
- Exploring Simple Siamese Representation Learning
- Understanding Contrastive Representation Learning through Alignment and Uniformity on the Hypersphere
- Self-supervised Label Augmentation via Input Transformations
- Contrastive Learning with Stronger Augmentations
- DenseNet Models for Tiny ImageNet Classification
- Contrastive Predictive Coding Based Feature for Automatic Speaker Verification
- AdCo: Adversarial Contrast for Efficient Learning of Unsupervised Representations from Self-Trained Negative Adversaries