When is Multicalibration Post-Processing Necessary?

Dutch Hansen, Siddartha Devic, Preetum Nakkiran, Vatsal Sharan
6/10/2024
cs.LG

No official code found. Best implementation: sid-devic/multicalibration (8 stars, MIT, last commit 2024-11-18). 1 more below.

Abstract

Calibration is a well-studied property of predictors which guarantees meaningful uncertainty estimates. Multicalibration is a related notion -- originating in algorithmic fairness -- which requires predictors to be simultaneously calibrated over a potentially complex and overlapping collection of protected subpopulations (such as groups defined by ethnicity, race, or income). We conduct the first comprehensive study evaluating the usefulness of multicalibration post-processing across a broad set of tabular, image, and language datasets for models spanning from simple decision trees to 90 million parameter fine-tuned LLMs. Our findings can be summarized as follows: (1) models which are calibrated out of the box tend to be relatively multicalibrated without any additional post-processing; (2) multicalibration post-processing can help inherently uncalibrated models and large vision and language models; and (3) traditional calibration measures may sometimes provide multicalibration implicitly. More generally, we also distill many independent observations which may be useful for practical and effective applications of multicalibration post-processing in real-world contexts. We also release a python package implementing multicalibration algorithms, available via `pip install multicalibration'.

View on arXivView PDF

Code Implementations(2)

MIT

A python package for multicalibration post-processing.

82PythonSep 21, 20241 years agoMIT
fairness-mlpost-processing
MIT

A benchmark repository for studying multicalibration post-processing algorithms.

41PythonOct 14, 20241 years agoMIT

Cite this paper

@article{hansen2024when,
  title  = {When is Multicalibration Post-Processing Necessary?},
  author = {Dutch Hansen and Siddartha Devic and Preetum Nakkiran and Vatsal Sharan},
  year   = {2024},
  eprint = {2406.06487},
  archivePrefix = {arXiv},
  doi    = {10.48550/arXiv.2406.06487},
  url    = {http://arxiv.org/abs/2406.06487v2},
  journal = {NEURIPS 2024 2024}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2406.06487.svg)](https://codeofpaper.com/paper/2406.06487)

Discussion