mlsanityData to 14 Oct 2026

Calibrating for Class Weights by Modeling Machine Learning

arXiv 2205.04613 ↗First posted 10 May 2022cs.LG econ.THcited by 4

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

  1. 2014Adam: A Method for Stochastic Optimizationcited by 37,240
  2. 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
  3. 2015Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shiftcited by 7,005
  4. 2016Densely Connected Convolutional Networkscited by 6,720
  5. 2017On Calibration of Modern Neural Networkscited by 4,960
  6. 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,926
  7. 2021MLP-Mixer: An all-MLP Architecture for Visioncited by 1,446
  8. 2017CheXNet: Radiologist-Level Pneumonia Detection on Chest X-Rays with Deep Learningcited by 461
  9. 2021Revisiting the Calibration of Modern Neural Networkscited by 390
  10. 2019The Algorithmic Automation Problem: Prediction, Triage, and Human Effortcited by 101
  11. 2021Don't Just Blame Over-parametrization for Over-confidence: Theoretical Analysis of Calibration in Binary Classificationcited by 39
  12. 2021Deep Probability Estimationcited by 14

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2024: 1; 2025: 1; 2026: 2.

4 papers.