Long-Tailed Classification Based on Dual Branch Learning Network

Qiu Xinyi,Li Jun,Wei Yifei,Song Mei
DOI: https://doi.org/10.1007/978-3-031-06767-9_37
2022-01-01
Abstract:The phenomenon of long-tailed data distribution widely exists in every corner of life. And in fact, the fundamental solution to long-tailed distribution problem is actually the combination of imbalanced classification and few-shot learning. At present, the mainstream methods under the framework of deep learning are re-sampling, re-weighting and other strategies like transfer learning. This paper proves that when dealing with long-tailed data, the conventional deep learning methods turn to show a low accuracy of few-shot samples and many-shot samples. Therefore, this paper proposes a long-tailed recognition model based on Dual Branch Learning Network (DBLN). While the imbalanced learning branch is using class-balanced loss function to modify model preference for head data, DBLN leads in a data augmentation learning branch based on improved rebalanced Mixup algorithm. These two branches update the weights by sharing dynamic parameters. DBLN model shows an excellent performance in improving the tendency of the deep learning algorithm to prefer the head classes and the recognition ability of the tail data. The experiments on two benchmark datasets (imbalanced CIFAR-10/100 dataset and rail fastener dataset which contains 5,000 rail pictures about 4 types of fasteners) show that the proposed improved dual branch long-tailed recognition model is superior to other advanced methods.
What problem does this paper attempt to address?