The Community for Technology Leaders
2013 IEEE 13th International Conference on Data Mining (2012)
Brussels, Belgium Belgium
Dec. 10, 2012 to Dec. 13, 2012
ISSN: 1550-4786
ISBN: 978-1-4673-4649-8
pp: 695-704
ABSTRACT
Obtaining good probability estimates is imperative for many applications. The increased uncertainty and typically asymmetric costs surrounding rare events increases this need. Experts (and classification systems) often rely on probabilities to inform decisions. However, we demonstrate that class probability estimates attained via supervised learning in imbalanced scenarios systematically underestimate the probabilities for minority class instances, despite ostensibly good overall calibration. To our knowledge, this problem has not previously been explored. Motivated by our exposition of this issue, we propose a simple, effective and theoretically motivated method to mitigate the bias of probability estimates for imbalanced data that bags estimators calibrated over balanced bootstrap samples. This approach drastically improves performance on the minority instances without greatly affecting overall calibration. We show that additional uncertainty can be exploited via a Bayesian approach by considering posterior distributions over bagged probability estimates.
INDEX TERMS
class imbalance, probability estimates
CITATION
Byron C. Wallace, Issa J. Dahabreh, "Class Probability Estimates are Unreliable for Imbalanced Data (and How to Fix Them)", 2013 IEEE 13th International Conference on Data Mining, vol. 00, no. , pp. 695-704, 2012, doi:10.1109/ICDM.2012.115
86 ms
(Ver 3.3 (11022016))