arrow
返回

Self-Supervised Learning by Estimating Twin Class Distribution

delete2023-01-01
delete6
PRE
AI
F
Feng Wang
T
Tao Kong *
R
Rufeng Zhang
刘华坪 封面图
刘华坪 (Huaping Liu) *
H
Hang Li
DOI:10.1109/TIP.2023.3266169delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
We present Twist, a simple and theoretically explainable self-supervised representation learning method by classifying large-scale unlabeled datasets in an end-to-end way. We employ a siamese network terminated by a softmax operation to produce twin class distributions of two augmented images. Without supervision, we enforce the class distributions of different augmentations to be consistent. However, simply minimizing the divergence between augmentations will generate collapsed solutions, i.e., outputting the same class distribution for all images. In this case, little information about the input images is preserved. To solve this problem, we propose to maximize the mutual information between the input image and the output class predictions. Specifically, we minimize the entropy of the distribution for each sample to make the class prediction assertive, and maximize the entropy of the mean distribution to make the predictions of different samples diverse. In this way, Twist can naturally avoid the collapsed solutions without specific designs such as asymmetric network, stop-gradient operation, or momentum encoder. As a result, Twist outperforms previous state-of-the-art methods on a wide range of tasks. Specifically on the semi-supervised classification task, Twist achieves 61.2% top-1 accuracy with 1% ImageNet labels using a ResNet-50 as backbone, surpassing previous best results by an improvement of 6.2%. Codes and pre-trained models are available at https://github.com/bytedance/TWIST
Keyword:
Task analysis
Mutual information
Entropy
Probability distribution
Self-supervised learning
Classification algorithms
Neural networks
Unsupervised learning
image classification

期刊

IEEE Transactions on Image Processing 封面图
IEEE Transactions on Image Processing
IF:
13.7
论文数:
1.0W
被引数:
8.4W

机构

T
tsinghua university
学者数:
11.9W
论文数: 10.0W
被引数: 137
T
tongji university
学者数:
7.9W
论文数: 6.0W
被引数: 98
引用论文

引用论文

High prevalence of malaria in a non-endemic setting among febrile episodes in travellers and migrants coming from endemic areas: a retrospective analysis of a 2013–2018 cohort
err2021-11-27
err0
errOAAI
errAlejandro Garcia-Ruiz de Morales; Covadonga Morcate; Elena Isaba-Ares; Ramon Perez-Tanoira; Jose A. Perez-Molina
err分享
err收藏
A glove-based system for studying hand-object manipulation via joint pose and force sensing
err2017-09-01
err0
PREAI
errHangxin Liu; Xu Xie; Matt Millar; Mark Edmonds; Feng Gao; Yixin Zhu; Veronica J. Santos; Brandon Rothrock; Song-Chun Zhu
err分享
err收藏
Intra-Abdominal Splenosis Mimicking Metastatic Cancer
err2011-03-01
err0
PREAI
errNicholas J. Short; Teresa G. Hayes; Peeyush Bhargava
err分享
err收藏
Ingestion of corrosive acids
err1989-09-01
err0
PREAI
errShowkat Ali Zargar; Rakesh Kochhar; Birender Nagi; Saroj Mehta; Satish Kumar Mehta
err分享
err收藏
Role of simple descriptors and applicability domain in predicting change in protein thermostability
err2018-09-07
err0
errOAAI
errKenneth N. McGuinness; Weilan Pan; Robert P. Sheridan; Grant Murphy; Alejandro Crespo
err分享
err收藏
学者 查看更多内容