Skip to content
/ SAE Public

Semantic Autoencoder for Zero-shot Learning (Spotlight), CVPR 2017

Notifications You must be signed in to change notification settings

Elyorcv/SAE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

87 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Semantic Autoencoder for Zero-shot Learning

Elyor Kodirov, Tao Xiang, and Shaogang Gong, Spotlight.

Abstract

Existing zero-shot learning (ZSL) models typically learn a projection function from a visual feature space to a semantic embedding space (e.g.~attribute space). However, such a projection function is only concerned with predicting the training seen class semantic representation (e.g.~attribute prediction) or classification. When applied to test data, which in the context of ZSL contains different (unseen) classes without training data, a ZSL model typically suffers from the project domain shift problem. In this work, we present a novel solution to ZSL based on learning a Semantic AutoEncoder (SAE). Taking the encoder-decoder paradigm, an encoder aims to project a visual feature vector into the semantic space as in the existing ZSL models. However, the decoder exerts an additional constraint, that is, the projection/code must be able to reconstruct the original visual feature. We show that with this additional reconstruction constraint, the learned projection function from the seen classes is able to generalise better to the new unseen classes. Importantly, the encoder and decoder are linear and symmetric which enable us to develop an extremely efficient learning algorithm. Extensive experiments on six benchmark datasets demonstrate that the proposed SAE outperforms significantly the existing ZSL models with the additional benefit of lower computational cost. Furthermore, when the SAE is applied to supervised clustering problem, it also beats the state-of-the-art.

An implementation of SAE in MATLAB

In Python: https://github.com/hoseong-kim/sae-pytorch/blob/master/sae.py (Thanks to hoseong-kim)

Download Paper

Paper

Citation

@ARTICLE{ekodirov_cvpr2017,
   author = {Elyor Kodirov, Tao Xiang, and Shagong Gong},
   title = "{Semantic Autoencoder for Zero-shot Learning}",
   journal = {IEEE CVPR 2017},
   year = 2017,
   month = July
}

About

Semantic Autoencoder for Zero-shot Learning (Spotlight), CVPR 2017

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages