mlsanityData to 14 Oct 2026

Correcting Performance Estimation Bias in Imbalanced Classification with Minority Subconcepts

arXiv 2604.26024 ↗First posted 28 Apr 2026cs.LG cs.AIcited by 0

Not on the map. Why?

Abstract (arXiv)

Class-level evaluation can conceal substantial performance disparities across subconcepts within the same class, causing models that perform well on average to fail on specific subpopulations. Prior work has shown that common evaluation measures for imbalanced classification are biased toward larger minority subconcepts and that utility-based reweighting using true subconcept labels can mitigate this bias; however, such labels are rarely available at test time. We introduce a practical utility-weighted evaluation that replaces unavailable subconcept labels with predicted posterior probabilities from a multiclass subconcept model. Evaluation weights are defined as the expected utility under this posterior, yielding a soft, uncertainty-aware metric we call predicted-weighted balanced accuracy (pBA). Experiments on tabular benchmarks as well as medical-imaging and text datasets show that unweighted scores can be misleading under within-class heterogeneity, while pBA provides more stable and interpretable assessments when subconcept distributions are uneven but not pathological. Our code is available at: https://anonymous.4open.science/r/correcting-bias-imbalance-9C6C/.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 11 of its 24 references that are in the map. First posted on arXiv 28 Apr 2026.

Builds on

11 of 24 references.

References · 11

  1. 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,375
  2. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,247
  3. 2019Class-Balanced Loss Based on Effective Number of Samplescited by 1,245
  4. 2019Learning Imbalanced Datasets with Label-Distribution-Aware Margin Losscited by 989
  5. 2017ChestX-ray8: Hospital-scale Chest X-ray Database and Benchmarks on Weakly-Supervised Classification and Localization of Common Thorax Diseasescited by 972
  6. 2019Decoupling Representation and Classifier for Long-Tailed Recognitioncited by 809
  7. 2021Just Train Twice: Improving Group Robustness without Training Group Informationcited by 523
  8. 2020Balanced Meta-Softmax for Long-Tailed Visual Recognitioncited by 400
  9. 2019Hidden Stratification Causes Clinically Meaningful Failures in Machine Learning for Medical Imagingcited by 216
  10. 2017PMLB: A Large Benchmark Suite for Machine Learning Evaluation and Comparisoncited by 140
  11. 2019Exploring Hate Speech Detection in Multimodal Publicationscited by 103

Built on by

Not cited yet.