Cargando…

Optimizing Few-Shot Learning Based on Variational Autoencoders

Despite the importance of few-shot learning, the lack of labeled training data in the real world makes it extremely challenging for existing machine learning methods because this limited dataset does not well represent the data variance. In this research, we suggest employing a generative approach u...

Descripción completa

Detalles Bibliográficos
Autores principales: Wei, Ruoqi, Mahmood, Ausif
Formato: Online Artículo Texto
Lenguaje:English
Publicado: MDPI 2021
Materias:
Acceso en línea:https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8618453/
https://www.ncbi.nlm.nih.gov/pubmed/34828088
http://dx.doi.org/10.3390/e23111390
_version_ 1784604751398174720
author Wei, Ruoqi
Mahmood, Ausif
author_facet Wei, Ruoqi
Mahmood, Ausif
author_sort Wei, Ruoqi
collection PubMed
description Despite the importance of few-shot learning, the lack of labeled training data in the real world makes it extremely challenging for existing machine learning methods because this limited dataset does not well represent the data variance. In this research, we suggest employing a generative approach using variational autoencoders (VAEs), which can be used specifically to optimize few-shot learning tasks by generating new samples with more intra-class variations on the Labeled Faces in the Wild (LFW) dataset. The purpose of our research is to increase the size of the training dataset using various methods to improve the accuracy and robustness of the few-shot face recognition. Specifically, we employ the VAE generator to increase the size of the training dataset, including the basic and the novel sets while utilizing transfer learning as the backend. Based on extensive experimental research, we analyze various data augmentation methods to observe how each method affects the accuracy of face recognition. The face generation method based on VAEs with perceptual loss can effectively improve the recognition accuracy rate to 96.47% using both the base and the novel sets.
format Online
Article
Text
id pubmed-8618453
institution National Center for Biotechnology Information
language English
publishDate 2021
publisher MDPI
record_format MEDLINE/PubMed
spelling pubmed-86184532021-11-27 Optimizing Few-Shot Learning Based on Variational Autoencoders Wei, Ruoqi Mahmood, Ausif Entropy (Basel) Article Despite the importance of few-shot learning, the lack of labeled training data in the real world makes it extremely challenging for existing machine learning methods because this limited dataset does not well represent the data variance. In this research, we suggest employing a generative approach using variational autoencoders (VAEs), which can be used specifically to optimize few-shot learning tasks by generating new samples with more intra-class variations on the Labeled Faces in the Wild (LFW) dataset. The purpose of our research is to increase the size of the training dataset using various methods to improve the accuracy and robustness of the few-shot face recognition. Specifically, we employ the VAE generator to increase the size of the training dataset, including the basic and the novel sets while utilizing transfer learning as the backend. Based on extensive experimental research, we analyze various data augmentation methods to observe how each method affects the accuracy of face recognition. The face generation method based on VAEs with perceptual loss can effectively improve the recognition accuracy rate to 96.47% using both the base and the novel sets. MDPI 2021-10-24 /pmc/articles/PMC8618453/ /pubmed/34828088 http://dx.doi.org/10.3390/e23111390 Text en © 2021 by the authors. https://creativecommons.org/licenses/by/4.0/Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (https://creativecommons.org/licenses/by/4.0/).
spellingShingle Article
Wei, Ruoqi
Mahmood, Ausif
Optimizing Few-Shot Learning Based on Variational Autoencoders
title Optimizing Few-Shot Learning Based on Variational Autoencoders
title_full Optimizing Few-Shot Learning Based on Variational Autoencoders
title_fullStr Optimizing Few-Shot Learning Based on Variational Autoencoders
title_full_unstemmed Optimizing Few-Shot Learning Based on Variational Autoencoders
title_short Optimizing Few-Shot Learning Based on Variational Autoencoders
title_sort optimizing few-shot learning based on variational autoencoders
topic Article
url https://www.ncbi.nlm.nih.gov/pmc/articles/PMC8618453/
https://www.ncbi.nlm.nih.gov/pubmed/34828088
http://dx.doi.org/10.3390/e23111390
work_keys_str_mv AT weiruoqi optimizingfewshotlearningbasedonvariationalautoencoders
AT mahmoodausif optimizingfewshotlearningbasedonvariationalautoencoders