Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

ReSSL: Relational Self-Supervised Learning with Weak Augmentation

About

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}.

Mingkai Zheng, Shan You, Fei Wang, Chen Qian, Changshui Zhang, Xiaogang Wang, Chang Xu• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-100 (test)
Accuracy63.79
3518
Image ClassificationImageNet-1k (val)
Top-1 Accuracy74.7
1453
Image ClassificationCIFAR-10 (test)
Accuracy90.2
906
Image ClassificationImageNet--
429
Image ClassificationTinyImageNet (test)
Accuracy46.6
366
Image ClassificationSTL-10 (test)
Accuracy88.25
357
Image ClassificationImageNet
Top-1 Accuracy74.7
324
ClassificationPASCAL VOC 2007 (test)
mAP (%)86.31
217
Image ClassificationImageNet (val)
Top-1 Accuracy74.7
118
Image ClassificationImageNet 20 Dp: CIFAR10 downstream (test)
Balanced Accuracy71.2
44
Showing 10 of 24 rows

Other info

Code

Follow for update