BBN: Bilateral-Branch Network with Cumulative Learning for Long-Tailed Visual Recognition

被引:612
作者
Zhou, Boyan [1 ]
Cui, Quan [1 ,2 ,4 ]
Wei, Xiu-Shen [1 ]
Chen, Zhao-Min [1 ,3 ,4 ]
机构
[1] Megvii Technol, Nanjing, Peoples R China
[2] Waseda Univ, Tokyo, Japan
[3] Nanjing Univ, Nanjing, Peoples R China
[4] Megvii Technol, Megvii Res Nanjing, Nanjing, Peoples R China
来源
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020) | 2020年
基金
国家重点研发计划;
关键词
D O I
10.1109/CVPR42600.2020.00974
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Our work focuses on tackling the challenging but natural visual recognition task of long-tailed data distribution (i.e., a few classes occupy most of the data, while most classes have rarely few samples). In the literature, class re-balancing strategies (e.g., re-weighting and re-sampling) are the prominent and effective methods proposed to alleviate the extreme imbalance for dealing with long-tailed problems. In this paper, we firstly discover that these rebalancing methods achieving satisfactory recognition accuracy owe to that they could significantly promote the classifier learning of deep networks. However, at the same time, they will unexpectedly damage the representative ability of the learned deep features to some extent. Therefore, we propose a unified Bilateral-Branch Network (BBN) to take care of both representation learning and classifier learning simultaneously, where each branch does perform its own duty separately. In particular, our BBN model is further equipped with a novel cumulative learning strategy, which is designed to first learn the universal patterns and then pay attention to the tail data gradually. Extensive experiments on four benchmark datasets, including the large-scale iNaturalist ones, justify that the proposed BBN can significantly outperform state-of-the-art methods. Furthermore, validation experiments can demonstrate both our preliminary discovery and effectiveness of tailored designs in BBN for long-tailed problems. Our method won the first place in the iNaturalist 2019 large scale species classification competition, and our code is open-source and available at https://github.com/Megvii- Nanjing/BBN.
引用
收藏
页码:9716 / 9725
页数:10
相关论文
共 37 条
[1]  
[Anonymous], 2019, ICML
[2]  
[Anonymous], 2019, ARXIV
[3]  
[Anonymous], 1987, Kendall's Advanced Theory of Statistics
[4]  
[Anonymous], 2019, CVPR, DOI DOI 10.1109/CVPR.2019.00800
[5]   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
[6]  
Byrd J, 2019, PR MACH LEARN RES, V97
[7]  
Cao Kaidi, 2019, NEURIPS, P1
[8]   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)
[9]  
Cui Yin, 2019, CVPR, P9268
[10]  
Cui Yin, 2018, PROC CVPR IEEE, P4109, DOI DOI 10.1109/CVPR.2018.00432