Towards Prior Gap and Representation Gap for Long-Tailed Recognition

Ming-Liang Zhang,Xu-Yao Zhang,Chuang Wang,Cheng-Lin Liu
DOI: https://doi.org/10.1016/j.patcog.2022.109012
IF: 8
2022-08-31
Pattern Recognition
Abstract:Most deep learning models are elaborately designed for balanced datasets, and thus they inevitably suffer performance degradation in practical long-tailed recognition tasks, especially to the minority classes. There are two crucial issues in learning from imbalanced datasets: skew decision boundary and unrepresentative feature space. In this work, we establish a theoretical framework to analyze the sources of these two issues from Bayesian perspective, and find that they are closely related to the prior gap and the representation gap, respectively. Under this framework, we show that existing long-tailed recognition methods manage to remove either the prior gap or the presentation gap. Different from these methods, we propose to simultaneously remove the two gaps to achieve more accurate long-tailed recognition. Specifically, we propose the prior calibration strategy to remove the prior gap and introduce three strategies (representative feature extraction, optimization strategy adjustment and effective sample modeling) to mitigate the representation gap. Extensive experiments on five benchmark datasets validate the superiority of our method against the state-of-the-art competitors.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?