Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Towards Calibrated Model for Long-Tailed Visual Recognition from Prior Perspective

About

Real-world data universally confronts a severe class-imbalance problem and exhibits a long-tailed distribution, i.e., most labels are associated with limited instances. The na\"ive models supervised by such datasets would prefer dominant labels, encounter a serious generalization challenge and become poorly calibrated. We propose two novel methods from the prior perspective to alleviate this dilemma. First, we deduce a balance-oriented data augmentation named Uniform Mixup (UniMix) to promote mixup in long-tailed scenarios, which adopts advanced mixing factor and sampler in favor of the minority. Second, motivated by the Bayesian theory, we figure out the Bayes Bias (Bayias), an inherent bias caused by the inconsistency of prior, and compensate it as a modification on standard cross-entropy loss. We further prove that both the proposed methods ensure the classification calibration theoretically and empirically. Extensive experiments verify that our strategies contribute to a better-calibrated model, and their combination achieves state-of-the-art performance on CIFAR-LT, ImageNet-LT, and iNaturalist 2018.

Zhengzhuo Xu, Zenghao Chai, Chun Yuan• 2021

Related benchmarks

TaskDatasetResultRank
Image ClassificationCIFAR-10 long-tailed (test)--
201
Image ClassificationiNaturalist 2018 (val)
Top-1 Accuracy69.15
116
Image ClassificationCIFAR-100 LT (val)
Top-1 Accuracy61.25
69
Image ClassificationCIFAR-10-LT (val)
Top-1 Acc89.66
65
Confidence calibrationCIFAR-100-LT (test)
ECE0.45
53
Image ClassificationImageNet-LT 1.0 (test)
Top-1 Accuracy48.4
37
Image ClassificationCIFAR-10-LT Imbalance Ratio 100 (test)
Top-1 Accuracy82.7
32
Image ClassificationImageNet-LT IR=256 (test)
Accuracy48.4
32
Image ClassificationCIFAR-100-LT IR=100 (test)
Accuracy45.5
25
Image ClassificationCIFAR-100-LT IR=50 (test)
Top-1 Acc (IR 50)51.1
23
Showing 10 of 21 rows

Other info

Follow for update