Calibrating for Class Weights by Modeling Machine Learning
Machine Learning
2022-08-02 v2 Theoretical Economics
Abstract
A much studied issue is the extent to which the confidence scores provided by machine learning algorithms are calibrated to ground truth probabilities. Our starting point is that calibration is seemingly incompatible with class weighting, a technique often employed when one class is less common (class imbalance) or with the hope of achieving some external objective (cost-sensitive learning). We provide a model-based explanation for this incompatibility and use our anthropomorphic model to generate a simple method of recovering likelihoods from an algorithm that is miscalibrated due to class weighting. We validate this approach in the binary pneumonia detection task of Rajpurkar, Irvin, Zhu, et al. (2017).
Cite
@article{arxiv.2205.04613,
title = {Calibrating for Class Weights by Modeling Machine Learning},
author = {Andrew Caplin and Daniel Martin and Philip Marx},
journal= {arXiv preprint arXiv:2205.04613},
year = {2022}
}
Comments
14 pages, 4 figures