HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Generalized Zero- and Few-Shot Learning via Aligned Variational Autoencoders

Edgar Schönfeld; Sayna Ebrahimi; Samarth Sinha; Trevor Darrell; Zeynep Akata

Generalized Zero- and Few-Shot Learning via Aligned Variational Autoencoders

Abstract

Many approaches in generalized zero-shot learning rely on cross-modal mapping between the image feature space and the class embedding space. As labeled images are expensive, one direction is to augment the dataset by generating either images or image features. However, the former misses fine-grained details and the latter requires learning a mapping associated with class embeddings. In this work, we take feature generation one step further and propose a model where a shared latent space of image features and class embeddings is learned by modality-specific aligned variational autoencoders. This leaves us with the required discriminative information about the image and classes in the latent features, on which we train a softmax classifier. The key to our approach is that we align the distributions learned from images and from side-information to construct latent features that contain the essential multi-modal information associated with unseen classes. We evaluate our learned latent features on several benchmark datasets, i.e. CUB, SUN, AWA1 and AWA2, and establish a new state of the art on generalized zero-shot as well as on few-shot learning. Moreover, our results on ImageNet with various zero-shot splits show that our latent features generalize well in large-scale settings.

Code Repositories

edgarschnfld/CADA-VAE-PyTorch
Official
pytorch
Mentioned in GitHub
sanixa/CADA-VAE-pytorch
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
generalized-few-shot-learning-on-awa2CADA-VAE
Per-Class Accuracy (1-shot): 69.6
Per-Class Accuracy (10-shots): 80.2
Per-Class Accuracy (2-shots): 73.7
Per-Class Accuracy (20-shots): 80.9
Per-Class Accuracy (5-shots): 78.1
zero-shot-skeletal-action-recognition-on-ntuCADA-VAE
Accuracy (12 unseen classes): 28.96
Accuracy (5 unseen classes): 76.84
Random Split Accuracy: 60.74
zero-shot-skeletal-action-recognition-on-ntu-1CADA-VAE
Accuracy (10 unseen classes): 59.53
Accuracy (24 unseen classes): 35.77
Random Split Accuracy: 45.14
zero-shot-skeletal-action-recognition-on-pkuCADA-VAE
Random Split Accuracy: 60.74

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Generalized Zero- and Few-Shot Learning via Aligned Variational Autoencoders | Papers | HyperAI