A metric-based meta-learning approach combined attention mechanism and ensemble learning for few-shot learning

被引:17
|
作者
Guo, Nan [1 ,2 ,3 ,4 ,5 ]
Di, Kexin [1 ]
Liu, Hongyan [1 ,2 ,3 ,4 ,5 ]
Wang, Yifei [1 ]
Qiao, Junfei [1 ,2 ,3 ,4 ,5 ]
机构
[1] Beijing Univ Technol, Fac Informat Technol, Beijing, Peoples R China
[2] Minist Educ, Engn Res Ctr Intelligence Percept & Autonomous Co, Beijing, Peoples R China
[3] Beijing Lab Smart Environm Protect, Beijing, Peoples R China
[4] Beijing Key Lab Computat Intelligence & Intellige, Beijing, Peoples R China
[5] Beijing Artificial Intelligence Inst, Beijing, Peoples R China
基金
中国国家自然科学基金;
关键词
Meta-learning; Ensemble learning; Metric-learning; Attention module; Few-shot learning; NETWORK;
D O I
10.1016/j.displa.2021.102065
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
Meta-learning is one of the latest research directions in machine learning, which is considered to be one of the most probably ways to realize strong artificial intelligence. Meta-learning focuses on seeking solutions for machines to learn like human beings do - to recognize things through only few sample data and quickly adapt to new tasks. Challenges occur in how to train an efficient machine model with limited labeled data, since the model is easily over-fitted. In this paper, we address this obvious but important problem and propose a metric-based metalearning model, which combines attention mechanisms and ensemble learning method. In our model, we first design a dual path attention module which considers both channel attention and spatial attention module, and the attention modules have been stacked to conduct a meta-learner for few shot meta-learning. Then, we apply an ensemble method called snap-shot ensemble to the attention-based meta-learner in order to generate more models in a single episode. Features abstracted from the models are put into the metric-based architecture to compute a prototype for each class. Our proposed method intensifies the feature extracting ability of backbone network in meta-learner and reduces over-fitting through ensemble learning and metric learning method. Experimental results toward several meta-learning datasets show that our approach is effective.
引用
收藏
页数:8
相关论文
共 50 条
  • [1] META-LEARNING WITH ATTENTION FOR IMPROVED FEW-SHOT LEARNING
    Hou, Zejiang
    Walid, Anwar
    Kung, Sun-Yuan
    [J]. 2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 2725 - 2729
  • [2] Prior-knowledge and attention based meta-learning for few-shot learning
    Qin, Yunxiao
    Zhang, Weiguo
    Zhao, Chenxu
    Wang, Zezheng
    Zhu, Xiangyu
    Shi, Jingping
    Qi, Guojun
    Lei, Zhen
    [J]. KNOWLEDGE-BASED SYSTEMS, 2021, 213
  • [3] Metric-Based Meta-Learning for Cross-Domain Few-Shot Identification of Welding Defect
    Xie, Tingli
    Huang, Xufeng
    Choi, Seung-Kyum
    [J]. JOURNAL OF COMPUTING AND INFORMATION SCIENCE IN ENGINEERING, 2023, 23 (03)
  • [4] Unsupervised meta-learning for few-shot learning
    Xu, Hui
    Wang, Jiaxing
    Li, Hao
    Ouyang, Deqiang
    Shao, Jie
    [J]. PATTERN RECOGNITION, 2021, 116
  • [5] Adversarially Robust Few-Shot Learning: A Meta-Learning Approach
    Goldblum, Micah
    Fowl, Liam
    Goldstein, Tom
    [J]. ADVANCES IN NEURAL INFORMATION PROCESSING SYSTEMS (NEURIPS 2020), 2020, 33
  • [6] MetaFSCEL A Meta-Learning Approach for Few-Shot Class Incremental Learning
    Chi, Zhixiang
    Gu, Li
    Liu, Huan
    Wang, Yang
    Yu, Yuanhao
    Tang, Jin
    [J]. 2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2022, : 14146 - 14155
  • [7] Ensemble Meta-Learning for Few-Shot Soot Density Recognition
    Gu, Ke
    Zhang, Yonghui
    Qiao, Junfei
    [J]. IEEE TRANSACTIONS ON INDUSTRIAL INFORMATICS, 2021, 17 (03) : 2261 - 2270
  • [8] Task Agnostic Meta-Learning for Few-Shot Learning
    Jamal, Muhammad Abdullah
    Qi, Guo-Jun
    [J]. 2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 11711 - 11719
  • [9] Ensemble-Based Deep Metric Learning for Few-Shot Learning
    Zhou, Meng
    Li, Yaoyi
    Lu, Hongtao
    [J]. ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2020, PT I, 2020, 12396 : 406 - 418
  • [10] Few-Shot Rolling Bearing Fault Diagnosis with Metric-Based Meta Learning
    Wang, Sihan
    Wang, Dazhi
    Kong, Deshan
    Wang, Jiaxing
    Li, Wenhui
    Zhou, Shuai
    [J]. SENSORS, 2020, 20 (22) : 1 - 15