Attribute- and attention-guided few-shot classification

被引:0
|
作者
Wang, Ziquan [1 ,2 ]
Li, Hui [3 ]
Zhang, Zikai [1 ]
Chen, Feng [1 ]
Zhai, Jia [1 ]
机构
[1] Natl Key Lab Scattering & Radiat, Beijing, Peoples R China
[2] Tsinghua Univ, Beijing, Peoples R China
[3] Dalian Univ Technol, Dalian, Peoples R China
关键词
Few-shot learning; Attribute; Attention mechanism; Image classification;
D O I
10.1007/s00530-023-01236-1
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
The field of image classification faces significant challenges due to the scarcity of target samples, leading to model overfitting and difficult training. To address these issues, few-shot learning has emerged as a promising approach. However, current methods do not fully utilize the correlations among samples and external semantic information, resulting in poor recognition accuracy. To overcome these limitations, we propose a new few-shot classification method that incorporates both attributes and attention guided approach. The method leverages the attention mechanism to extract discriminative features from the images. By exploring regional correlations among samples, it assists in generating visual representations by utilizing predicted attribute features. As a result, accurate prototypes are generated. Extensive experiments were conducted on two attribute-labeled datasets, namely Caltech-UCSD Birds-200-2011(CUB) and SUN Attribute Database (SUN) Attribute Dataset. With the Resnet12 backbone, the method achieves remarkable accuracies of 79.95% and 89.34% for 1-shot and 5-shot, respectively, on the CUB dataset. Similarly, with the Conv4 backbone, the method achieves notable accuracies of 67.21% and 80.87% for 1-shot and 5-shot, respectively, on the SUN Attribute dataset. The achieved accuracies highlight the robustness and generalizability of our method, and show the capability of our method to accurately classify samples with limited training data, which is a significant advantage in real-world scenarios where labeled data are often scarce.
引用
收藏
页数:9
相关论文
共 50 条
  • [21] Transductive Graph-Attention Network for Few-shot Classification
    Pan, Lili
    Liu, Weifeng
    2022 16TH IEEE INTERNATIONAL CONFERENCE ON SIGNAL PROCESSING (ICSP2022), VOL 1, 2022, : 190 - 195
  • [22] Few-Shot Scene Classification with Attention Mechanism in Remote Sensing
    Zhang, Duona
    Zhao, Hongjia
    Lu, Yuanyao
    Cui, Jian
    Zhang, Baochang
    Computer Engineering and Applications, 2024, 60 (04) : 173 - 182
  • [23] Total Relation Network with Attention for Few-Shot Image Classification
    Li X.-X.
    Liu Z.-Y.
    Wu J.-J.
    Cao J.
    Ma Z.-Y.
    Jisuanji Xuebao/Chinese Journal of Computers, 2023, 46 (02): : 371 - 384
  • [24] Intra-inter channel attention for few-shot classification
    Yang L.
    Zhang T.
    Wang Y.
    Gu X.
    Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2023, 31 (21): : 3145 - 3155
  • [25] Hierarchical Attention Prototypical Networks for Few-Shot Text Classification
    Sun, Shengli
    Sun, Qingfeng
    Zhou, Kevin
    Lv, Tengchao
    2019 CONFERENCE ON EMPIRICAL METHODS IN NATURAL LANGUAGE PROCESSING AND THE 9TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (EMNLP-IJCNLP 2019): PROCEEDINGS OF THE CONFERENCE, 2019, : 476 - 485
  • [26] PANet: Pluralistic Attention Network for Few-Shot Image Classification
    Cao, Wenming
    Li, Tianyuan
    Liu, Qifan
    He, Zhiquan
    NEURAL PROCESSING LETTERS, 2024, 56 (04)
  • [27] Mask-guided BERT for few-shot text classification
    Liao, Wenxiong
    Liu, Zhengliang
    Dai, Haixing
    Wu, Zihao
    Zhang, Yiyang
    Huang, Xiaoke
    Chen, Yuzhong
    Jiang, Xi
    Liu, David
    Zhu, Dajiang
    Li, Sheng
    Liu, Wei
    Liu, Tianming
    Li, Quanzheng
    Cai, Hongmin
    Li, Xiang
    NEUROCOMPUTING, 2024, 610
  • [28] Knowledge Guided Metric Learning for Few-Shot Text Classification
    Sui, Dianbo
    Chen, Yubo
    Mao, Binjie
    Qiu, Delai
    Liu, Kang
    Zhao, Jun
    2021 CONFERENCE OF THE NORTH AMERICAN CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS: HUMAN LANGUAGE TECHNOLOGIES (NAACL-HLT 2021), 2021, : 3266 - 3271
  • [29] Query-Guided Prototype Optimization for Few-Shot Classification
    Sun, Jiaxing
    Sun, Yinghui
    Shen, Xiaobo
    Sun, Quansen
    IEEE TRANSACTIONS ON CIRCUITS AND SYSTEMS FOR VIDEO TECHNOLOGY, 2024, 34 (12) : 12197 - 12209
  • [30] Attribute-guided prototype network for few-shot molecular property prediction
    Hou, Linlin
    Xiang, Hongxin
    Zeng, Xiangxiang
    Cao, Dongsheng
    Zeng, Li
    Song, Bosheng
    BRIEFINGS IN BIOINFORMATICS, 2024, 25 (05)