arrow
Return

Augmenting Few-Shot Learning With Supervised Contrastive Learning

delete2021-01-01
delete17
delete
OA
AI
T
Taemin Lee *
S
Sungjoo Yoo
DOI:10.1109/ACCESS.2021.3074525delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Few-shot learning deals with a small amount of data which incurs insufficient performance with conventional cross-entropy loss. We propose a pretraining approach for few-shot learning scenarios. That is, considering that the feature extractor quality is a critical factor in few-shot learning, we augment the feature extractor using a contrastive learning technique. It is reported that supervised contrastive learning applied to base class training in transductive few-shot training pipeline leads to improved results, outperforming the state-of-the-art methods on Mini-ImageNet and CUB. Furthermore, our experiment shows that a much larger dataset is needed to retain few-shot classification accuracy when domain-shift degradation exists, and if our method is applied, the need for a large dataset is eliminated. The accuracy gain can be translated to a runtime reduction of 3.87x in a resource-constrained environment.
Keywords:
Few-shot learning
contrastive learning
information maximization
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

IEEE Access cover
IEEE Access
IF:
3.6
Papers:
9.8W
Citations:
29.4W

Organization

S
seoul national university (snu)
Scholars:
7.2W
Papers: 6.6W
Citations: 86