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

TransZero: Attribute-guided Transformer for Zero-Shot Learning

About

Zero-shot learning (ZSL) aims to recognize novel classes by transferring semantic knowledge from seen classes to unseen ones. Semantic knowledge is learned from attribute descriptions shared between different classes, which act as strong priors for localizing object attributes that represent discriminative region features, enabling significant visual-semantic interaction. Although some attention-based models have attempted to learn such region features in a single image, the transferability and discriminative attribute localization of visual features are typically neglected. In this paper, we propose an attribute-guided Transformer network, termed TransZero, to refine visual features and learn attribute localization for discriminative visual embedding representations in ZSL. Specifically, TransZero takes a feature augmentation encoder to alleviate the cross-dataset bias between ImageNet and ZSL benchmarks, and improves the transferability of visual features by reducing the entangled relative geometry relationships among region features. To learn locality-augmented visual features, TransZero employs a visual-semantic decoder to localize the image regions most relevant to each attribute in a given image, under the guidance of semantic attribute information. Then, the locality-augmented visual features and semantic vectors are used to conduct effective visual-semantic interaction in a visual-semantic embedding network. Extensive experiments show that TransZero achieves the new state of the art on three ZSL benchmarks. The codes are available at: \url{https://github.com/shiming-chen/TransZero}.

Shiming Chen, Ziming Hong, Yang Liu, Guo-Sen Xie, Baigui Sun, Hao Li, Qinmu Peng, Ke Lu, Xinge You• 2021

Related benchmarks

TaskDatasetResultRank
Generalized Zero-Shot LearningCUB
H Score68.8
250
Generalized Zero-Shot LearningSUN
H40.8
184
Generalized Zero-Shot LearningAWA2
S Score82.3
165
Zero-shot LearningCUB
Top-1 Accuracy76.8
144
Zero-shot LearningSUN
Top-1 Accuracy65.6
114
Zero-shot LearningAWA2
Top-1 Accuracy0.701
95
Zero-shot Image ClassificationAWA2 (test)
Metric U61.3
46
Zero-shot Image ClassificationCUB
U Score69.3
34
ClassificationAWA2 (test)
MCA (unseen)62
22
Image ClassificationSUN Attribute (test)
U Score52.6
19
Showing 10 of 14 rows

Other info

Follow for update