Selective Classification Can Magnify Disparities Across Groups
By Erik Jones, Shiori Sagawa, Pang Wei Koh, Ananya Kumar, Percy Liang. 2020
53 citations recorded by the source corpus.
Selective classification, in which models are allowed to abstain on uncertain predictions, is a natural approach to improving accuracy in settings where errors are costly but abstentions are manageable. In this paper, we find that while selective classification can improve average accuracies, it can simultaneously magnify existing accuracy disparities between various groups within a population, especially in the presence of spurious correlations. We observe this behavior consistently across five datasets from computer vision and NLP. Surprisingly, increasing the abstention rate can even decrease accuracies on some groups. To better understand when selective classification improves or worsens accuracy on a group, we study its margin distribution, which captures the model's confidences over all predictions. For example, when the margin distribution is symmetric, we prove that whether selective classification monotonically improves or worsens accuracy is fully determined by the accuracy at full coverage (i.e., without any abstentions) and whether the distribution satisfies a property we term left-log-concavity. Our analysis also shows that selective classification tends to magnify accuracy disparities that are present at full coverage. Fortunately, we find that it uniformly improves each group when applied to distributionally-robust models that achieve similar full-coverage accuracies across groups. Altogether, our results imply selective classification should be used with care and underscore the importance of models that perform equally well across groups at full coverage.
Subfields: Artificial Intelligence, Safety, Robustness & Interpretability
How researchers see this paper
The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.
In Artificial Intelligence
- Stefano Ermon
- Percy Liang
- Tengyu Ma
- Jiaming Song
- Chenlin Meng
- Sergey Levine
- Aditi Raghunathan
- Ananya Kumar
In Safety, Robustness & Interpretability
- Tengyu Ma
- Aditi Raghunathan
- Percy Liang
- Ananya Kumar
- Stefano Ermon
- Abhishek Sinha
- Jiaming Song
- J. Kolter
Citations in the corpus
- D2C: Diffusion-Decoding Models for Few-Shot Conditional Generation (2021, background): “This kind of dataset bias in CelebA is well-documented in previous work [46]; among all the images with the “blond” attribute, only 5.”
- Calibrated ensembles can mitigate accuracy tradeoffs under distribution shift (2022, methodology): “Jones et al. [2021] use the group DRO intervention.”