Joint Feature Disentanglement and Hallucination for Few-Shot Image Classification

被引:12
作者
Lin, Chia-Ching [1 ]
Chu, Hsin-Li [2 ]
Wang, Yu-Chiang Frank [1 ,3 ]
Lei, Chin-Laung [1 ]
机构
[1] Natl Taiwan Univ, Dept Elect Engn, Taipei 10617, Taiwan
[2] Natl Taiwan Univ, Dept Engn Sci & Ocean Engn, Taipei 10617, Taiwan
[3] Natl Taiwan Univ, Grad Inst Commun Engn, Taipei 10617, Taiwan
关键词
Task analysis; Feature extraction; Visualization; Training; Data models; Data mining; Birds; Few-shot learning (FSL); image classification; data hallucination; feature disentanglement;
D O I
10.1109/TIP.2021.3124322
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Few-shot learning (FSL) refers to the learning task that generalizes from base to novel concepts with only few examples observed during training. One intuitive FSL approach is to hallucinate additional training samples for novel categories. While this is typically done by learning from a disjoint set of base categories with sufficient amount of training data, most existing works did not fully exploit the intra-class information from base categories, and thus there is no guarantee that the hallucinated data would represent the class of interest accordingly. In this paper, we propose Feature Disentanglement and Hallucination Network (FDH-Net), which jointly performs feature disentanglement and hallucination for FSL purposes. More specifically, our FDH-Net is able to disentangle input visual data into class-specific and appearance-specific features. With both data recovery and classification constraints, hallucination of image features for novel categories using appearance information extracted from base categories can be achieved. We perform extensive experiments on two fine-grained datasets (CUB and FLO) and two coarse-grained ones (mini-ImageNet and CIFAR-100). The results confirm that our framework performs favorably against state-of-the-art metric-learning and hallucination-based FSL models.
引用
收藏
页码:9245 / 9258
页数:14
相关论文
共 61 条
[1]  
Belongie, 2011, CNS T 2011 001
[2]   Representation Learning: A Review and New Perspectives [J].
Bengio, Yoshua ;
Courville, Aaron ;
Vincent, Pascal .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2013, 35 (08) :1798-1828
[3]  
Chen LB, 2017, IEEE INT SYMP NANO, P1, DOI 10.1109/NANOARCH.2017.8053709
[4]  
Chen MT, 2020, AAAI CONF ARTIF INTE, V34, P10559
[5]  
Chen W, 2019, INT C LEARN REPR
[6]  
Chen X, 2016, ADV NEUR IN, V29
[7]   Image Deformation Meta-Networks for One-Shot Learning [J].
Chen, Zitian ;
Fu, Yanwei ;
Wang, Yu-Xiong ;
Ma, Lin ;
Liu, Wei ;
Hebert, Martial .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :8672-8681
[8]   Multi-Level Semantic Feature Augmentation for One-Shot Learning [J].
Chen, Zitian ;
Fu, Yanwei ;
Zhang, Yinda ;
Jiang, Yu-Gang ;
Xue, Xiangyang ;
Sigal, Leonid .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2019, 28 (09) :4594-4605
[9]   EFFICIENT FACE MANIPULATION VIA DEEP FEATURE DISENTANGLEMENT AND REINTEGRATION NET [J].
Cheng, Bin ;
Dai, Tao ;
Chen, Bin ;
Xia, Shutao ;
Li, Xiu .
2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, :1990-1994
[10]  
Chu WH, 2018, IEEE IMAGE PROC, P2979, DOI 10.1109/ICIP.2018.8451350