arXiv · 2104.00466
Improving Calibration for Long-Tailed Recognition
Abstract
Deep neural networks may perform poorly when training datasets are heavily class-imbalanced. Recently, two-stage methods decouple representation learning and classifier learning to improve performance. But there is still the vital issue of miscalibration. To address it, we design two methods to improve calibration and performance in such scenarios. Motivated by the fact that predicted probability distributions of classes are highly related to the numbers of class instances, we propose label-aware smoothing to deal with different degrees of over-confidence for classes and improve classifier learning. For dataset bias between these two stages due to different samplers, we further propose shifted batch normalization in the decoupling framework. Our proposed methods set new records on multiple popular long-tailed recognition benchmark datasets, including CIFAR-10-LT, CIFAR-100-LT, ImageNet-LT, Places-LT, and iNaturalist 2018. Code will be available at https://github.com/Jia-Research-Lab/MiSLAS.
Explore related subjects
Keep this discovery
Zhisheng Zhong, Jiequan Cui, Shu Liu, Jiaya Jia. 2021-04-01. Improving Calibration for Long-Tailed Recognition. https://arxiv.org/abs/2104.00466
Cite the original work for its findings. Save a collection to share your selection of sources.