SCL: Self-supervised contrastive learning for few-shot image classification


Lim, Jit Yan and Lim, Kian Ming and Lee, Chin Poo and Tan, Yong Xuan (2023) SCL: Self-supervised contrastive learning for few-shot image classification. Neural Networks, 165. pp. 19-30. ISSN 0893-6080

[img] Text
1.pdf - Published Version
Restricted to Repository staff only

Download (1MB)


Few-shot learning aims to train a model with a limited number of base class samples to classify the novel class samples. However, to attain generalization with a limited number of samples is not a trivial task. This paper proposed a novel few-shot learning approach named Self-supervised Contrastive Learning (SCL) that enriched the model representation with multiple self-supervision objectives. Given the base class samples, the model is trained with the base class loss. Subsequently, contrastive-based self-supervision is introduced to minimize the distance between each training sample with their augmented variants to improve the sample discrimination. To recognize the distant sample, rotation-based self-supervision is proposed to enable the model to learn to recognize the rotation degree of the samples for better sample diversity. The multitask environment is introduced where each training sample is assigned with two class labels: base class label and rotation class label. Complex augmentation is put forth to help the model learn a deeper understanding of the object. The image structure of the training samples are augmented independent of the base class information. The proposed SCL is trained to minimize the base class loss, contrastive distance loss, and rotation class loss simultaneously to learn the generic features and improve the novel class performance. With the multiple self-supervision objectives, the proposed SCL outperforms state-of-the-art few-shot approaches on few-shot image classification benchmark datasets.

Item Type: Article
Uncontrolled Keywords: Few-shot learning, Self-supervised learning, Meta-learning, Contrastive learning
Subjects: L Education > LB Theory and practice of education > LB1060 Learning
Divisions: Faculty of Information Science and Technology (FIST)
Depositing User: Ms Nurul Iqtiani Ahmad
Date Deposited: 03 Jul 2023 03:26
Last Modified: 31 Oct 2023 03:27


Downloads per month over past year

View ItemEdit (login required)