Sorry, you need to enable JavaScript to visit this website.

SELF-SUPERVISED LEARNING FOR FEW-SHOT IMAGE CLASSIFICATION

Citation Author(s):
Da Chen, Yuefeng Chen, Yuhong Li, Feng Mao, Yuan He, Hui Xue
Submitted by:
DA CHEN
Last updated:
25 June 2021 - 4:17am
Document Type:
Poster
Event:
 

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 learn- ing(SL) manner which becomes a bottleneck of the few-shot learn- ing 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 (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 https://github.com/phecy/SSL-FEW-SHOT.

up
0 users have voted: