Analyzing the Benefits of Prototypes for Semi-Supervised Category Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zhang, Liyi, Nelson, Logan, Griffiths, Thomas L.
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866929373209165824
author Zhang, Liyi
Nelson, Logan
Griffiths, Thomas L.
author_facet Zhang, Liyi
Nelson, Logan
Griffiths, Thomas L.
contents Categories can be represented at different levels of abstraction, from prototypes focused on the most typical members to remembering all observed exemplars of the category. These representations have been explored in the context of supervised learning, where stimuli are presented with known category labels. We examine the benefits of prototype-based representations in a less-studied domain: semi-supervised learning, where agents must form unsupervised representations of stimuli before receiving category labels. We study this problem in a Bayesian unsupervised learning model called a variational auto-encoder, and we draw on recent advances in machine learning to implement a prior that encourages the model to use abstract prototypes to represent data. We apply this approach to image datasets and show that forming prototypes can improve semi-supervised category learning. Additionally, we study the latent embeddings of the models and show that these prototypes allow the models to form clustered representations without supervision, contributing to their success in downstream categorization performance.
format Preprint
id arxiv_https___arxiv_org_abs_2406_02268
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Analyzing the Benefits of Prototypes for Semi-Supervised Category Learning
Zhang, Liyi
Nelson, Logan
Griffiths, Thomas L.
Machine Learning
I.2; I.5
Categories can be represented at different levels of abstraction, from prototypes focused on the most typical members to remembering all observed exemplars of the category. These representations have been explored in the context of supervised learning, where stimuli are presented with known category labels. We examine the benefits of prototype-based representations in a less-studied domain: semi-supervised learning, where agents must form unsupervised representations of stimuli before receiving category labels. We study this problem in a Bayesian unsupervised learning model called a variational auto-encoder, and we draw on recent advances in machine learning to implement a prior that encourages the model to use abstract prototypes to represent data. We apply this approach to image datasets and show that forming prototypes can improve semi-supervised category learning. Additionally, we study the latent embeddings of the models and show that these prototypes allow the models to form clustered representations without supervision, contributing to their success in downstream categorization performance.
title Analyzing the Benefits of Prototypes for Semi-Supervised Category Learning
topic Machine Learning
I.2; I.5
url https://arxiv.org/abs/2406.02268