Normalizing Batch Normalization for Long-Tailed Recognition
In real-world scenarios, the number of training samples across classes usually subjects to a long-tailed distribution. The conventionally trained network may achieve unexpected inferior performance on the rare class compared to the frequent class. Most previous works attempt to rectify the network bias from the data-level or from the classifier-level. Differently, in this paper, we identify that the bias towards the frequent class may be encoded into features, i.e., the rare-specific features which play a key role in discriminating the rare class are much weaker than the frequent-specific features. Based on such an observation, we introduce a simple yet effective approach, normalizing the parameters of Batch Normalization (BN) layer to explicitly rectify the feature bias. To achieve this end, we represent theWeight/Bias parameters of a BN layer as a vector, normalize it into a unit one and multiply the unit vector by a scalar learnable parameter. Through decoupling the direction and magnitude of parameters in BN layer to learn, the Weight/Bias exhibits a more balanced distribution and thus the strength of features becomes more even. Extensive experiments on various long-tailed recognition benchmarks (i.e., CIFAR-10/100-LT, ImageNet-LT and iNaturalist 2018) show that our method outperforms previous state-of-the-arts remarkably.
Authors
- Guoliang Kang (ORCID: https://orcid.org/0000-0003-1978-2025)
- Baochang Zhang (ORCID: https://orcid.org/0000-0001-7396-6218)
- Xiaoyue Duan (ORCID: https://orcid.org/0000-0002-5792-2335)
- Yuxiang Bao (ORCID: https://orcid.org/0000-0001-9376-4656)
- Linlin Yang (ORCID: https://orcid.org/0000-0001-6752-0252)
- Bo Zhao (ORCID: https://orcid.org/0000-0002-2120-2571)
Institutions
- Shanghai Jiao Tong University (CN)
- Communication University of China (CN)
- Beihang University (CN)
Publication Details
- Journal
- IEEE Transactions on Image Processing
- Published
- 2024-12-27
- DOI
- https://doi.org/10.1109/tip.2024.3518099
- Citations
- 3
- Primary Topic
- Machine Learning and Algorithms
- Type
- article
- Field-Weighted Citation Impact
- 1.04
Funders
- National Natural Science Foundation of China
- National Key Research and Development Program of China
- Natural Science Foundation of Zhejiang Province