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 条
  • [31] TRANSDUCTIVE PROTOTYPICAL NETWORK FOR FEW-SHOT CLASSIFICATION
    Liu, Xinyue
    Liu, Pengxin
    Zong, Linlin
    2020 IEEE INTERNATIONAL CONFERENCE ON IMAGE PROCESSING (ICIP), 2020, : 1671 - 1675
  • [32] Few-Shot Learning for Medical Image Classification
    Cai, Aihua
    Hu, Wenxin
    Zheng, Jun
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 : 441 - 452
  • [33] Generalized Few-Shot Classification with Knowledge Graph
    Liu, Dianqi
    Bai, Liang
    Yu, Tianyuan
    NEURAL PROCESSING LETTERS, 2023, 55 (06) : 7649 - 7666
  • [34] Generalized Few-Shot Classification with Knowledge Graph
    Dianqi Liu
    Liang Bai
    Tianyuan Yu
    Neural Processing Letters, 2023, 55 : 7649 - 7666
  • [35] Optimization model based on attention mechanism for few-shot image classification
    Liao, Ruizhi
    Zhai, Junhai
    Zhang, Feng
    MACHINE VISION AND APPLICATIONS, 2024, 35 (02)
  • [36] SELF-ADAPTIVE EMBEDDING FOR FEW-SHOT CLASSIFICATION BY HIERARCHICAL ATTENTION
    Wang, Xueliang
    Wu, Feng
    Wang, Jie
    2020 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2020,
  • [37] Multi-scale Attention-Based Few-Shot Hyperspectral Images Classification
    Ding, Lanwei
    Cao, Guo
    Xu, Ling
    Deng, Lindiao
    Xu, Hao
    Pan, Qikun
    Shang, Yanfeng
    FOURTEENTH INTERNATIONAL CONFERENCE ON GRAPHICS AND IMAGE PROCESSING, ICGIP 2022, 2022, 12705
  • [38] Reservoir computing based network for few-shot image classification
    Wang Bin
    Lan Hai
    Yu Hui
    Guo Jie-long
    Wei Xian
    CHINESE JOURNAL OF LIQUID CRYSTALS AND DISPLAYS, 2023, 38 (10) : 1399 - 1408
  • [39] Few-shot relation classification by context attention-based prototypical networks with BERT
    Bei Hui
    Liang Liu
    Jia Chen
    Xue Zhou
    Yuhui Nian
    EURASIP Journal on Wireless Communications and Networking, 2020
  • [40] Few-Shot Malware Classification via Attention-Based Transductive Learning Network
    Deng, Liting
    Yu, Chengli
    Wen, Hui
    Xin, Mingfeng
    Sun, Yue
    Sun, Limin
    Zhu, Hongsong
    MOBILE NETWORKS & APPLICATIONS, 2024, : 1690 - 1704