Transductive Episodic-Wise Adaptive Metric for Few-Shot Learning

被引:163
作者
Qiao, Limeng [1 ,4 ]
Shi, Yemin [2 ,4 ]
Li, Jia [3 ,4 ]
Wang, Yaowei [4 ]
Huang, Tiejun [2 ,4 ]
Tian, Yonghong [2 ,4 ]
机构
[1] Peking Univ, AAIS, Ctr Data Sci, Beijing, Peoples R China
[2] Peking Univ, Sch EE&CS, Natl Engn Lab Video Technol, Beijing, Peoples R China
[3] Beihang Univ, SCSE, State Key Lab Virtual Real Technol & Syst, Beijing, Peoples R China
[4] Peng Cheng Lab, Shenzhen, Peoples R China
来源
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019) | 2019年
基金
中国国家自然科学基金; 国家重点研发计划;
关键词
D O I
10.1109/ICCV.2019.00370
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot learning, which aims at extracting new concepts rapidly from extremely few examples of novel classes, has been featured into the meta-learning paradigm recently. Yet, the key challenge of how to learn a generalizable classifier with the capability of adapting to specific tasks with severely limited data still remains in this domain. To this end, we propose a Transductive Episodic-wise Adaptive Metric (TEAM) framework for few-shot learning, by integrating the meta-learning paradigm with both deep metric learning and transductive inference. With exploring the pairwise constraints and regularization prior within each task, we explicitly formulate the adaptation procedure into a standard semi-definite programming problem. By solving the problem with its closed-form solution on the fly with the setup of transduction, our approach efficiently tailors an episodic-wise metric for each task to adapt all features from a shared task-agnostic embedding space into a more discriminative task-specific metric space. Moreover, we further leverage an attention-based bi-directional similarity strategy for extracting the more robust relationship between queries and prototypes. Extensive experiments on three benchmark datasets show that our framework is superior to other existing approaches and achieves the state-of-the-art performance in the few-shot literature.
引用
收藏
页码:3602 / 3611
页数:10
相关论文
共 40 条
[1]  
[Anonymous], P 3 INT C LEARNING R
[2]  
[Anonymous], 2019, ICLR
[3]  
[Anonymous], 2018, PR MACH LEARN RES
[4]  
[Anonymous], 2018, CVPR
[5]  
[Anonymous], 2018, ICLR
[6]  
[Anonymous], 2018, ICLR
[7]  
[Anonymous], 2018, ARXIV180204376
[8]  
[Anonymous], 2018, ICLR
[9]  
[Anonymous], 2017, ICML
[10]  
Bellet Aurelien, 2013, research report