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

Self-Supervision Can Be a Good Few-Shot Learner

About

Existing few-shot learning (FSL) methods rely on training with a large labeled dataset, which prevents them from leveraging abundant unlabeled data. From an information-theoretic perspective, we propose an effective unsupervised FSL method, learning representations with self-supervision. Following the InfoMax principle, our method learns comprehensive representations by capturing the intrinsic structure of the data. Specifically, we maximize the mutual information (MI) of instances and their representations with a low-bias MI estimator to perform self-supervised pre-training. Rather than supervised pre-training focusing on the discriminable features of the seen classes, our self-supervised model has less bias toward the seen classes, resulting in better generalization for unseen classes. We explain that supervised pre-training and self-supervised pre-training are actually maximizing different MI objectives. Extensive experiments are further conducted to analyze their FSL performance with various training settings. Surprisingly, the results show that self-supervised pre-training can outperform supervised pre-training under the appropriate conditions. Compared with state-of-the-art FSL methods, our approach achieves comparable performance on widely used FSL benchmarks without any labels of the base classes.

Yuning Lu, Liangjian Wen, Jianzhuang Liu, Yajing Liu, Xinmei Tian• 2022

Related benchmarks

TaskDatasetResultRank
Few-shot classificationtieredImageNet (test)--
282
Few-shot Image ClassificationminiImageNet (test)--
111
Few-shot classificationMiniImagenet
5-way 5-shot Accuracy83.22
98
Few-shot Image ClassificationtieredImageNet--
90
5-way ClassificationEuroSAT
Average Accuracy95.34
51
5-way ClassificationISIC BSCD-FSL
Average Accuracy62.27
45
5-way ClassificationChestX BSCD-FSL
Average Accuracy39.48
45
Few-shot Image ClassificationCropDiseases CDFSL (test)
Accuracy98.14
45
5-way 1-shot ClassificationMini-Imagenet (test)--
43
5-way 5-shot Classificationtiered-ImageNet (test)
Accuracy86.51
32
Showing 10 of 11 rows

Other info

Follow for update