Collect and Select: Semantic Alignment Metric Learning for Few-Shot Learning

被引:129
作者
Hao, Fusheng [1 ,2 ]
He, Fengxiang [3 ]
Cheng, Jun [1 ,2 ]
Wang, Lei [1 ,2 ]
Cao, Jianzhong [4 ]
Tao, Dacheng [3 ]
机构
[1] Chinese Acad Sci, Shenzhen Inst Adv Technol, CAS Key Lab Human Machine Intelligence Synergy Sy, Beijing, Peoples R China
[2] Chinese Univ Hong Kong, Hong Kong, Peoples R China
[3] Univ Sydney, Sch Comp Sci, Fac Engn, UBTECH Sydney AI Ctr, Darlington, NSW 2008, Australia
[4] Chinese Acad Sci, Xian Inst Opt & Precis Mech, Beijing, Peoples R China
来源
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019) | 2019年
基金
中国国家自然科学基金; 澳大利亚研究理事会;
关键词
D O I
10.1109/ICCV.2019.00855
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot learning aims to learn latent patterns from few training examples and has shown promises in practice. However, directly calculating the distances between the query image and support image in existing methods may cause ambiguity because dominant objects can locate anywhere on images. To address this issue, this paper proposes a Semantic Alignment Metric Learning (SAML) method for few-shot learning that aligns the semantically relevant dominant objects through a "collect-and-select" strategy. Specifically, we first calculate a relation matrix (RM) to "collect" the distances of each local region pairs of the 3D tensor extracted from a query image and the mean tensor of the support images. Then, the attention technique is adapted to "select" the semantically relevant pairs and put more weights on them. Afterwards, a multi-layer perceptron (MLP) is utilized to map the reweighted RMs to their corresponding similarity scores. Theoretical analysis demonstrates the generalization ability of SAML and gives a theoretical guarantee. Empirical results demonstrate that semantic alignment is achieved. Extensive experiments on benchmark datasets validate the strengths of the proposed approach and demonstrate that SAML significantly outperforms the current state-of-the-art methods. The source code is available at https://github.com/haofusheng/SAML.
引用
收藏
页码:8459 / 8468
页数:10
相关论文
共 45 条
[1]  
[Anonymous], 2012, FDN MACHINE LEARNING
[2]  
[Anonymous], 2018, ARXIV181007218
[3]  
[Anonymous], 2017, ARXIV170308033
[4]  
[Anonymous], 2018, ARXIV180105599
[5]  
[Anonymous], 2017, ARXIV170600326
[6]  
[Anonymous], 2017, OPTIMIZATION MODEL F
[7]  
[Anonymous], BRIT MACH VIS C
[8]  
[Anonymous], 2019, P INT C LEARN REPR
[9]  
[Anonymous], 2016, P BRIT MACHINE VISIO
[10]  
BARTLETT P. L., 2017, NIPS, V30, P6240