A Two-Stream Network with Image-to-Class Deep Metric for Few-Shot Classification

被引:1
作者
Gu, Qinghua [1 ]
Luo, Zhengding [1 ]
Zhu, Yuesheng [1 ]
机构
[1] Peking Univ, Shenzhen Grad Sch, Commun & Informat Secur Lab, Beijing, Peoples R China
来源
ECAI 2020: 24TH EUROPEAN CONFERENCE ON ARTIFICIAL INTELLIGENCE | 2020年 / 325卷
关键词
D O I
10.3233/FAIA200409
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot learning in image classification aims to learn classifiers for new classes when few examples are available for each class. Though recent work has greatly advanced promising classification performance, they mainly focus on the feature maps extracted from RGB images and the task-invariant image-to-image metrics. In this paper, we argue that richer features need to be learned and the general metrics are not effective enough due to the scarcity of examples in few-shot learning. Specifically, we propose a Two-Stream Neural Network (TSNN) with a learnable Image-to-Class Deep Metric (ICDM) for few-shot learning, which is trained end-to-end from scratch upon the recent episodic training mechanism. We not only extract features from RGB images to find contrast differences in semantic information, but also leverage the steganalysis features extracted from a steganalysis rich model filter layer to discover the local inconsistencies between different categories. Meanwhile, we extend our model to fine-grained few-shot classification, which is benefit from the proposed novel ICDM. The experimental results on three benchmark datasets show that our approach attains superior performance, with the largest improvement of 6.01% in classification accuracy over related competitive baselines.
引用
收藏
页码:2704 / 2711
页数:8
相关论文
共 32 条
[1]  
[Anonymous], 2017, P 5 INT C LEARNING R
[2]  
[Anonymous], 2010, Caltech-UCSD birds 200
[3]  
[Anonymous], 2016, OPTIMIZATION MODEL F
[4]  
[Anonymous], 2017, ARXIV170703141
[5]  
Bengio Y., 2012, P ICML WORKSH UNS TR, V7, P19
[6]   Memory Matching Networks for One-Shot Image Recognition [J].
Cai, Qi ;
Pan, Yingwei ;
Yao, Ting ;
Yan, Chenggang ;
Mei, Tao .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :4080-4088
[7]   Spot and Learn: A Maximum-Entropy Patch Sampler for Few-Shot Image Classification [J].
Chu, Wen-Hsuan ;
Li, Yu-Jhe ;
Chang, Jing-Cheng ;
Wang, Yu-Chiang Frank .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :6244-6253
[8]  
Collobert R, 2011, J MACH LEARN RES, V12, P2493
[9]  
Finn C, 2017, PR MACH LEARN RES, V70
[10]   Rich Models for Steganalysis of Digital Images [J].
Fridrich, Jessica ;
Kodovsky, Jan .
IEEE TRANSACTIONS ON INFORMATION FORENSICS AND SECURITY, 2012, 7 (03) :868-882