Prototype-based classifier learning for long-tailed visual recognition

被引:10
作者
Wei, Xiu-Shen [1 ,2 ,3 ,4 ]
Xu, Shu-Lin [1 ,3 ,4 ]
Chen, Hao [1 ]
Xiao, Liang [1 ]
Peng, Yuxin [5 ]
机构
[1] Nanjing Univ Sci & Technol, Sch Comp Sci & Engn, Nanjing 210094, Peoples R China
[2] Xidian Univ, State Key Lab Integrated Serv Networks, Xian 710071, Peoples R China
[3] Minist Educ, Key Lab Intelligent Percept & Syst High Dimens In, Nanjing 210094, Peoples R China
[4] Jiangsu Key Lab Image & Video Understanding Socia, Nanjing 210094, Peoples R China
[5] Peking Univ, Wangxuan Inst Comp Technol, Beijing 100871, Peoples R China
基金
国家重点研发计划; 中国国家自然科学基金;
关键词
long-tailed distribution; categorical prototype; classifier generation; classifier calibration; class imbalance; CLASS IMBALANCE;
D O I
10.1007/s11432-021-3489-1
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In this paper, we tackle the long-tailed visual recognition problem from the categorical prototype perspective by proposing a prototype-based classifier learning (PCL) method. Specifically, thanks to the generalization ability and robustness, categorical prototypes reveal their advantages of representing the category semantics. Coupled with their class-balance characteristic, categorical prototypes also show potential for handling data imbalance. In our PCL, we propose to generate the categorical classifiers based on the prototypes by performing a learnable mapping function. To further alleviate the impact of imbalance on classifier generation, two kinds of classifier calibration approaches are designed from both prototype-level and example-level aspects. Extensive experiments on five benchmark datasets, including the large-scale iNaturalist, Places-LT, and ImageNet-LT, justify that the proposed PCL can outperform state-of-the-arts. Furthermore, validation experiments can demonstrate the effectiveness of tailored designs in PCL for long-tailed problems.
引用
收藏
页数:15
相关论文
共 62 条
[1]  
[Anonymous], 2019, ADV NEUR IN
[2]  
Banerjee A, 2005, J MACH LEARN RES, V6, P1705
[3]   A systematic study of the class imbalance problem in convolutional neural networks [J].
Buda, Mateusz ;
Maki, Atsuto ;
Mazurowski, Maciej A. .
NEURAL NETWORKS, 2018, 106 :249-259
[4]   SMOTE: Synthetic minority over-sampling technique [J].
Chawla, Nitesh V. ;
Bowyer, Kevin W. ;
Hall, Lawrence O. ;
Kegelmeyer, W. Philip .
2002, American Association for Artificial Intelligence (16)
[5]  
Chu P., 2020, COMPUTER VISION ECCV
[6]  
Clevert D.-A., 2016, 4 INT C LEARN REPR I
[7]   Class-Balanced Loss Based on Effective Number of Samples [J].
Cui, Yin ;
Jia, Menglin ;
Lin, Tsung-Yi ;
Song, Yang ;
Belongie, Serge .
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, :9260-9269
[8]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[9]  
Edwards H, 2017, P INT C LEARN REPR, P114
[10]  
Fort Stanislav, 2017, Gaussian prototypical networks for few-shot learning on omniglot