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

Self-Supervised Learning For Few-Shot Image Classification

About

Few-shot image classification aims to classify unseen classes with limited labelled samples. Recent works benefit from the meta-learning process with episodic tasks and can fast adapt to class from training to testing. Due to the limited number of samples for each task, the initial embedding network for meta-learning becomes an essential component and can largely affect the performance in practice. To this end, most of the existing methods highly rely on the efficient embedding network. Due to the limited labelled data, the scale of embedding network is constrained under a supervised learning(SL) manner which becomes a bottleneck of the few-shot learning methods. In this paper, we proposed to train a more generalized embedding network with self-supervised learning (SSL) which can provide robust representation for downstream tasks by learning from the data itself. We evaluate our work by extensive comparisons with previous baseline methods on two few-shot classification datasets ({\em i.e.,} MiniImageNet and CUB) and achieve better performance over baselines. Tests on four datasets in cross-domain few-shot learning classification show that the proposed method achieves state-of-the-art results and further prove the robustness of the proposed model. Our code is available at \hyperref[https://github.com/phecy/SSL-FEW-SHOT.]{https://github.com/phecy/SSL-FEW-SHOT.}

Da Chen, Yuefeng Chen, Yuhong Li, Feng Mao, Yuan He, Hui Xue• 2019

Related benchmarks

TaskDatasetResultRank
Few-shot classificationminiImageNet standard (test)
5-way 1-shot Acc76.82
138
Few-shot classificationCUB-200-2011 (test)
5-way 1-shot Acc77.09
56
Few-shot Image ClassificationCropDiseases CDFSL (test)--
45
Few-shot Image ClassificationChestX CDFSL (test)
Accuracy (5-way 5-shot)28.5
25
Few-shot Image ClassificationEuroSat CDFSL (test)
Accuracy (5-way 5-shot)83.44
8
5-way ClassificationISIC
Accuracy (5-way, 5-shot)44.15
8
Showing 6 of 6 rows

Other info

Code

Follow for update