Unsupervised Meta-Learning via In-Context Learning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Vettoruzzo, Anna, Braccaioli, Lorenzo, Vanschoren, Joaquin, Nowaczyk, Marlena
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909484614418432
author Vettoruzzo, Anna
Braccaioli, Lorenzo
Vanschoren, Joaquin
Nowaczyk, Marlena
author_facet Vettoruzzo, Anna
Braccaioli, Lorenzo
Vanschoren, Joaquin
Nowaczyk, Marlena
contents Unsupervised meta-learning aims to learn feature representations from unsupervised datasets that can transfer to downstream tasks with limited labeled data. In this paper, we propose a novel approach to unsupervised meta-learning that leverages the generalization abilities of in-context learning observed in transformer architectures. Our method reframes meta-learning as a sequence modeling problem, enabling the transformer encoder to learn task context from support images and utilize it to predict query images. At the core of our approach lies the creation of diverse tasks generated using a combination of data augmentations and a mixing strategy that challenges the model during training while fostering generalization to unseen tasks at test time. Experimental results on benchmark datasets showcase the superiority of our approach over existing unsupervised meta-learning baselines, establishing it as the new state-of-the-art. Remarkably, our method achieves competitive results with supervised and self-supervised approaches, underscoring its efficacy in leveraging generalization over memorization.
format Preprint
id arxiv_https___arxiv_org_abs_2405_16124
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Unsupervised Meta-Learning via In-Context Learning
Vettoruzzo, Anna
Braccaioli, Lorenzo
Vanschoren, Joaquin
Nowaczyk, Marlena
Machine Learning
Unsupervised meta-learning aims to learn feature representations from unsupervised datasets that can transfer to downstream tasks with limited labeled data. In this paper, we propose a novel approach to unsupervised meta-learning that leverages the generalization abilities of in-context learning observed in transformer architectures. Our method reframes meta-learning as a sequence modeling problem, enabling the transformer encoder to learn task context from support images and utilize it to predict query images. At the core of our approach lies the creation of diverse tasks generated using a combination of data augmentations and a mixing strategy that challenges the model during training while fostering generalization to unseen tasks at test time. Experimental results on benchmark datasets showcase the superiority of our approach over existing unsupervised meta-learning baselines, establishing it as the new state-of-the-art. Remarkably, our method achieves competitive results with supervised and self-supervised approaches, underscoring its efficacy in leveraging generalization over memorization.
title Unsupervised Meta-Learning via In-Context Learning
topic Machine Learning
url https://arxiv.org/abs/2405.16124