Class Distribution Shifts in Zero-Shot Learning: Learning Robust Representations

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Slavutsky, Yuli, Benjamini, Yuval
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913603527901184
author Slavutsky, Yuli
Benjamini, Yuval
author_facet Slavutsky, Yuli
Benjamini, Yuval
contents Zero-shot learning methods typically assume that the new, unseen classes encountered during deployment come from the same distribution as the the classes in the training set. However, real-world scenarios often involve class distribution shifts (e.g., in age or gender for person identification), posing challenges for zero-shot classifiers that rely on learned representations from training classes. In this work, we propose and analyze a model that assumes that the attribute responsible for the shift is unknown in advance. We show that in this setting, standard training may lead to non-robust representations. To mitigate this, we develop an algorithm for learning robust representations in which (a) synthetic data environments are constructed via hierarchical sampling, and (b) environment balancing penalization, inspired by out-of-distribution problems, is applied. We show that our algorithm improves generalization to diverse class distributions in both simulations and experiments on real-world datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2311_18575
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Class Distribution Shifts in Zero-Shot Learning: Learning Robust Representations
Slavutsky, Yuli
Benjamini, Yuval
Machine Learning
Zero-shot learning methods typically assume that the new, unseen classes encountered during deployment come from the same distribution as the the classes in the training set. However, real-world scenarios often involve class distribution shifts (e.g., in age or gender for person identification), posing challenges for zero-shot classifiers that rely on learned representations from training classes. In this work, we propose and analyze a model that assumes that the attribute responsible for the shift is unknown in advance. We show that in this setting, standard training may lead to non-robust representations. To mitigate this, we develop an algorithm for learning robust representations in which (a) synthetic data environments are constructed via hierarchical sampling, and (b) environment balancing penalization, inspired by out-of-distribution problems, is applied. We show that our algorithm improves generalization to diverse class distributions in both simulations and experiments on real-world datasets.
title Class Distribution Shifts in Zero-Shot Learning: Learning Robust Representations
topic Machine Learning
url https://arxiv.org/abs/2311.18575