28 June 2022 SIM: an improved few-shot image classification model with multi-task learning
Jin Guo, Wengen Li, Jihong Guan, Hang Gao, Baobo Liu, Lili Gong
Author Affiliations +
Abstract

Few-shot learning (FSL) has been widely used for image classification when the training samples are limited and can effectively avoid overfitting. Most FSL methods are based on metric learning and tend to learn features that are conducive to classifying known classes of images. However, these methods cannot capture the semantic features that are important for classifying new classes of images. To address this issue, we proposed an improved few-shot image classification model based on multi-task learning termed SIM. It combines the self-supervised image representation learning task with the supervised image classification task, thus utilizing the complementarity of these two tasks. SIM has two stages, i.e., the pre-training stage and the meta-training stage. In pre-training stage, we learned the representation of training images via supervised learning and self-supervised learning. Then, in meta-training stage, we trained a linear classifier based on the learned representation. According to the experiments on four data sets, including three natural image data sets and one marine plankton image data set, SIM outperformed existing methods and achieved quite good performance in complex application scenarios.

© 2022 SPIE and IS&T 1017-9909/2022/$28.00 © 2022 SPIE and IS&T
Jin Guo, Wengen Li, Jihong Guan, Hang Gao, Baobo Liu, and Lili Gong "SIM: an improved few-shot image classification model with multi-task learning," Journal of Electronic Imaging 31(3), 033044 (28 June 2022). https://doi.org/10.1117/1.JEI.31.3.033044
Received: 22 January 2022; Accepted: 10 June 2022; Published: 28 June 2022
Lens.org Logo
CITATIONS
Cited by 2 scholarly publications.
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Image classification

Data modeling

Performance modeling

Machine learning

Statistical modeling

Solid state lighting

Ocean optics

Back to Top