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] SEGA: Semantic Guided Attention on Visual Prototype for Few-Shot Learning
    Yang, Fengyuan
    Wang, Ruiping
    Chen, Xilin
    2022 IEEE WINTER CONFERENCE ON APPLICATIONS OF COMPUTER VISION (WACV 2022), 2022, : 1586 - 1596
  • [32] VSA: Adaptive Visual and Semantic Guided Attention on Few-Shot Learning
    Chai, Jin
    Chen, Yisheng
    Shen, Weinan
    Zhang, Tong
    Chen, C. L. Philip
    ARTIFICIAL INTELLIGENCE, CICAI 2022, PT I, 2022, 13604 : 280 - 292
  • [33] A review of few-shot classification
    Lim, Jia Min
    Lim, Kian Ming
    Lee, Chin Poo
    Lim, Jit Yan
    EXPERT SYSTEMS WITH APPLICATIONS, 2025, 275
  • [34] Psanet: prototype-guided salient attention for few-shot segmentation
    Li, Hao
    Huang, Guoheng
    Yuan, Xiaochen
    Zheng, Zewen
    Chen, Xuhang
    Zhong, Guo
    Pun, Chi-Man
    VISUAL COMPUTER, 2025, 41 (04): : 2987 - 3001
  • [35] An approach based on performer-attention-guided few-shot learning model for plant disease classification
    Boulila, Wadii
    EARTH SCIENCE INFORMATICS, 2024, 17 (04) : 3797 - 3809
  • [36] Inferring Prototypes for Multi-Label Few-Shot Image Classification with Word Vector Guided Attention
    Yan, Kun
    Zhang, Chenbin
    Hou, Jun
    Wang, Ping
    Bouraoui, Zied
    Jameel, Shoaib
    Schockaert, Steven
    THIRTY-SIXTH AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTY-FOURTH CONFERENCE ON INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE / THE TWELVETH SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2022, : 2991 - 2999
  • [37] A Greedy Strategy Guided Graph Self-Attention Network for Few-Shot Hyperspectral Image Classification
    Zhu, Fei
    Shi, Cuiping
    Wang, Liguo
    Shi, Kaijie
    IEEE Transactions on Geoscience and Remote Sensing, 2024, 62
  • [38] Graph Few-shot Learning with Attribute Matching
    Wang, Ning
    Luo, Minnan
    Ding, Kaize
    Zhang, Lingling
    Li, Jundong
    Zheng, Qinghua
    CIKM '20: PROCEEDINGS OF THE 29TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, 2020, : 1545 - 1554
  • [39] Cross Attention with Deep Local Features for Few-Shot Image Classification
    Chu, Tengfei
    Shen, Hao
    Lv, Jing
    Yang, Ming
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PART X, 2023, 14263 : 98 - 110
  • [40] Optimization model based on attention mechanism for few-shot image classification
    Liao, Ruizhi
    Zhai, Junhai
    Zhang, Feng
    MACHINE VISION AND APPLICATIONS, 2024, 35 (02)