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 条
  • [41] Hybrid Adjacency Representation for Few-Shot Classification
    Zhao, Zhihao
    Liang, Xiaoyu
    Wu, Lu
    Xu, Ning
    Guo, Han
    2024 13TH INTERNATIONAL CONFERENCE ON COMMUNICATIONS, CIRCUITS AND SYSTEMS, ICCCAS 2024, 2024, : 514 - 518
  • [42] Optimization model based on attention mechanism for few-shot image classification
    Ruizhi Liao
    Junhai Zhai
    Feng Zhang
    Machine Vision and Applications, 2024, 35
  • [43] Few-shot relation classification by context attention-based prototypical networks with BERT
    Hui, Bei
    Liu, Liang
    Chen, Jia
    Zhou, Xue
    Nian, Yuhui
    EURASIP JOURNAL ON WIRELESS COMMUNICATIONS AND NETWORKING, 2020, 2020 (01)
  • [44] Few-Shot Learning Based on Double Pooling Squeeze and Excitation Attention
    Xu, Qiuyu
    Su, Jie
    Wang, Ying
    Zhang, Jing
    Zhong, Yixin
    ELECTRONICS, 2023, 12 (01)
  • [45] DUAL-ATTENTION NETWORK FOR FEW-SHOT SEGMENTATION
    Chen, Zhikui
    Wang, Han
    Zhang, Suhua
    Zhong, Fangming
    2022 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP), 2022, : 2210 - 2214
  • [46] Attribute-guided Dynamic Routing Graph Network for Transductive Few-shot Learning
    Chen, Chaofan
    Yang, Xiaoshan
    Yan, Ming
    Xu, Changsheng
    PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON MULTIMEDIA, MM 2022, 2022, : 6259 - 6268
  • [47] Quantum Few-Shot Image Classification
    Huang, Zhihao
    Shi, Jinjing
    Li, Xuelong
    IEEE TRANSACTIONS ON CYBERNETICS, 2025, 55 (01) : 194 - 206
  • [48] Few-Shot Classification with Contrastive Learning
    Yang, Zhanyuan
    Wang, Jinghua
    Zhu, Yingying
    COMPUTER VISION, ECCV 2022, PT XX, 2022, 13680 : 293 - 309
  • [49] Global and Local Attention Embedding Network for Few-Shot Fine-Grained Image Classification
    Hu, Jiayuan
    Own, Chung-Ming
    Tao, Wenyuan
    WEB AND BIG DATA, PT I, APWEB-WAIM 2020, 2020, 12317 : 740 - 747
  • [50] Adaptive attribute distribution similarity for few-shot learning
    Cai, Anping
    Chen, Leiting
    Chen, Yongqi
    He, Ziyu
    Tao, Shuqing
    Zhou, Chuan
    IMAGE AND VISION COMPUTING, 2024, 148