Hybrid routing transformer for zero-shot learning

De Cheng, Gerong Wang, Bo Wang, Qiang Zhang*, Jungong Han, Dingwen Zhang

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

61 Citations (Scopus)

Abstract

Zero-shot learning (ZSL) aims to learn models that can recognize unseen image semantics based on the training of data with seen semantics. Recent studies either leverage the global image features or mine discriminative local patch features to associate the extracted visual features to the semantic attributes. However, due to the lack of the necessary top-down guidance and semantic alignment for ensuring the model attend to the real attribute-correlation regions, these methods still encounter a significant semantic gap between the visual modality and the attribute modality, which makes their prediction on unseen semantics unreliable. To solve this problem, this paper establishes a novel transformer encoder-decoder model, called hybrid routing transformer (HRT). In HRT encoder, we embed an active attention, which is constructed by both the bottom-up and the top-down dynamic routing pathways to generate the attribute-aligned visual feature. While in HRT decoder, we use static routing to calculate the correlation among the attribute-aligned visual features, the corresponding attribute semantics, and the class attribute vectors to generate the final class label predictions. This design makes the presented transformer model a hybrid of 1) top-down and bottom-up attention pathways and 2) dynamic and static routing pathways. Comprehensive experiments on three widely-used benchmark datasets, namely CUB, SUN, and AWA2, are conducted. The obtained experimental results demonstrate the effectiveness of the proposed method.

Original languageEnglish
Article number109270
JournalPattern Recognition
Volume137
Early online date28 Dec 2022
DOIs
Publication statusPublished - 01 May 2023

Keywords

  • Attention
  • Hybrid routing
  • Transformer
  • Zero-shot learning

Fingerprint

Dive into the research topics of 'Hybrid routing transformer for zero-shot learning'. Together they form a unique fingerprint.

Cite this