Hybrid routing transformer for zero-shot learning

被引:70
作者
Cheng, De [1 ]
Wang, Gerong [5 ]
Wang, Bo [2 ]
Zhang, Qiang [1 ]
Han, Jungong [3 ]
Zhang, Dingwen [4 ,6 ]
机构
[1] Xidian Univ, Xian, Peoples R China
[2] Tsinghua Univ, Beijing, Peoples R China
[3] Aberystwyth Univ, Aberystwyth, Wales
[4] Northwestern Polytech Univ, Xian, Peoples R China
[5] Beijing Inst Remote Sensing Equipment, Beijing 100854, Peoples R China
[6] Hefei Comprehens Natl Sci Ctr, Inst Artificial Intelligence, Hefei, Peoples R China
关键词
Zero -shot learning; Hybrid routing; Transformer; Attention;
D O I
10.1016/j.patcog.2022.109270
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
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 se-mantic 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. Our code is released in https://github.com/KORIYN/HRT . (c) 2022 Elsevier Ltd. All rights reserved.
引用
收藏
页数:9
相关论文
共 54 条
[1]  
Akata Z, 2015, PROC CVPR IEEE, P2927, DOI 10.1109/CVPR.2015.7298911
[2]   Label-Embedding for Attribute-Based Classification [J].
Akata, Zeynep ;
Perronnin, Florent ;
Harchaoui, Zaid ;
Schmid, Cordelia .
2013 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2013, :819-826
[3]  
[Anonymous], 2014, INT C LEARN REPR
[4]   Adaptive Confidence Smoothing for Generalized Zero-Shot Learning [J].
Atzmon, Yuval ;
Chechik, Gal .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :11663-11672
[5]   End-to-End Object Detection with Transformers [J].
Carion, Nicolas ;
Massa, Francisco ;
Synnaeve, Gabriel ;
Usunier, Nicolas ;
Kirillov, Alexander ;
Zagoruyko, Sergey .
COMPUTER VISION - ECCV 2020, PT I, 2020, 12346 :213-229
[6]   Synthesized Classifiers for Zero-Shot Learning [J].
Changpinyo, Soravit ;
Chao, Wei-Lun ;
Gong, Boqing ;
Sha, Fei .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :5327-5336
[7]  
Chen KH, 2018, AAAI CONF ARTIF INTE, P4792
[8]   Zero-Shot Visual Recognition using Semantics-Preserving Adversarial Embedding Networks [J].
Chen, Long ;
Zhang, Hanwang ;
Xiao, Jun ;
Liu, Wei ;
Chang, Shih-Fu .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :1043-1052
[9]  
Chen SM, 2022, Arxiv, DOI arXiv:2112.08643
[10]   A Boundary Based Out-of-Distribution Classifier for Generalized Zero-Shot Learning [J].
Chen, Xingyu ;
Lan, Xuguang ;
Sun, Fuchun ;
Zheng, Nanning .
COMPUTER VISION - ECCV 2020, PT XXIV, 2020, 12369 :572-588