Calibrating for Class Weights by Modeling Machine Learning
Not on the map. Why?
Abstract (arXiv)
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).
Key facts. Cited by 4 papers in the map (data through 14 Oct 2026). Builds on 12 of its 44 references that are in the map. First posted on arXiv 10 May 2022.
Builds on
12 of 44 references.
References · 12
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
- 2016Densely Connected Convolutional Networkscited by 6,720
- 2017On Calibration of Modern Neural Networkscited by 4,960
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
- 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,446
- 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 461
- 2021Revisiting the Calibration of Modern Neural Networkscited by 390
- 2019The Algorithmic Automation Problem: Prediction, Triage, and Human Effortcited by 101
- 2021Don't Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classificationcited by 39
- 2021Deep Probability Estimationcited by 14
Built on by
Citing papers per year: 2024: 1; 2025: 1; 2026: 2.
4 papers.
- 2024Instance-wise Uncertainty for Class Imbalance in Semantic Segmentationcited by 0
- 2025Misaligned by Design: Incentive Failures in Machine Learningcited by 0
- 2026L-FNO: Lorentzian Fourier Neural Operator for Stochastic Event Dynamicscited by 0
- 2026Odds-Shift Slippage in One-vs-Rest Rankers: Diagnosing and Repairing Reweighting-Induced Top-K Errorscited by 0
- 2026Odds-Shift Slippage in One-vs-Rest Rankers: Diagnosing and Repairing Reweighting-Induced Top-K Errors12 Sep 2026
- 2026L-FNO: Lorentzian Fourier Neural Operator for Stochastic Event Dynamics7 May 2026
- 2025Misaligned by Design: Incentive Failures in Machine Learning10 Nov 2025
- 2024Instance-wise Uncertainty for Class Imbalance in Semantic Segmentation17 Jul 2024