Prior-Knowledge and Attention-based Meta-Learning for Few-Shot Learning
Yunxiao Qin,Weiguo Zhang,Chenxu Zhao,Zezheng Wang,Xiangyu Zhu,Jingping Shi,Guojun Qi,Zhen Lei
DOI: https://doi.org/10.1016/j.knosys.2020.106609
IF: 8.139
2020-01-01
Knowledge-Based Systems
Abstract:Recently, meta-learning has been shown as a promising way to solve few-shotlearning. In this paper, inspired by the human cognition process which utilizesboth prior-knowledge and vision attention in learning new knowledge, we presenta novel paradigm of meta-learning approach with three developments to introduceattention mechanism and prior-knowledge for meta-learning. In our approach,prior-knowledge is responsible for helping meta-learner expressing the inputdata into high-level representation space, and attention mechanism enablesmeta-learner focusing on key features of the data in the representation space.Compared with existing meta-learning approaches that pay little attention toprior-knowledge and vision attention, our approach alleviates themeta-learner's few-shot cognition burden. Furthermore, a Task-Over-Fitting(TOF) problem, which indicates that the meta-learner has poor generalization ondifferent K-shot learning tasks, is discovered and we propose a Cross-Entropyacross Tasks (CET) metric to model and solve the TOF problem. Extensiveexperiments demonstrate that we improve the meta-learner with state-of-the-artperformance on several few-shot learning benchmarks, and at the same time theTOF problem can also be released greatly.