Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Attribute Prototype Network for Any-Shot Learning

About

Any-shot image classification allows to recognize novel classes with only a few or even zero samples. For the task of zero-shot learning, visual attributes have been shown to play an important role, while in the few-shot regime, the effect of attributes is under-explored. To better transfer attribute-based knowledge from seen to unseen classes, we argue that an image representation with integrated attribute localization ability would be beneficial for any-shot, i.e. zero-shot and few-shot, image classification tasks. To this end, we propose a novel representation learning framework that jointly learns discriminative global and local features using only class-level attributes. While a visual-semantic embedding layer learns global features, local features are learned through an attribute prototype network that simultaneously regresses and decorrelates attributes from intermediate features. Furthermore, we introduce a zoom-in module that localizes and crops the informative regions to encourage the network to learn informative features explicitly. We show that our locality augmented image representations achieve a new state-of-the-art on challenging benchmarks, i.e. CUB, AWA2, and SUN. As an additional benefit, our model points to the visual evidence of the attributes in an image, confirming the improved attribute localization ability of our image representation. The attribute localization is evaluated quantitatively with ground truth part annotations, qualitatively with visualizations, and through well-designed user studies.

Wenjia Xu, Yongqin Xian, Jiuniu Wang, Bernt Schiele, Zeynep Akata• 2022

Related benchmarks

TaskDatasetResultRank
Generalized Zero-Shot LearningCUB
H Score70.6
250
Generalized Zero-Shot LearningSUN
H56.7
184
Generalized Zero-Shot LearningAWA2
S Score81
165
Zero-shot LearningCUB
Top-1 Accuracy75
144
Zero-shot LearningSUN
Top-1 Accuracy65.9
114
Few-shot classificationCUB
Accuracy87.1
96
Zero-shot LearningAWA2
Top-1 Accuracy0.739
95
Low-shot Image ClassificationImageNet 1k (novel classes)
Top-5 Acc85.5
57
Few-shot classificationCUB-200-2011 (test)--
56
Generalized Few-Shot LearningAWA2
Accuracy96
48
Showing 10 of 21 rows

Other info

Follow for update