Skip to main content

Utilities to calibrate model uncertainties and measure calibration.

Project description

Uncertainty Calibration Library

This repository contains library code to measure the calibration error of models, including confidence intervals computed by Bootstrap resampling, and code to recalibrate models.

Motivating example for uncertainty calibration: Calster and Vickers 2015 train a random forest that takes in features like tumor size, presence of teratoma and tries to predict the probability of testicular cancer. For a large number of patients, the model predicts ~20% chance of cancer. In reality, ~40% of these people had cancer. This underestimation can lead to doctors prescribing the wrong treatment in a situation where many lives are at stake. The high level point here is that the uncertainties that models output matter, not just the accuracy. Calibration is a popular way to measure the quality of a model's uncertainties, and recalibration is a way to take an existing model and correct its probabilities to make them better. See Verified Uncertainty Calibration for background on this.


pip3 install uncertainty-calibration

The calibration library requires python 3.6 or higher at the moment. If your project requires an earlier of version of python, and you wish to use our library, please contact us.


Measuring the calibration error of a model is as simple as:

import calibration as cal
calibration_error = cal.get_calibration_error(logits, labels)

Recalibrating a model is very simple as well. Recalibration requires a small labeled dataset, on which we train a recalibrator:

calibrator = cal.PlattBinnerCalibrator(num_points, num_bins=10)
calibrator.train_calibration(logits, labels)

Now whenever the model outputs a prediction, we pass it through the calibrator to produce better probabilities.

calibrated_logits = cal.calibrate(test_logits)

Our library makes it very easy to measure confidence intervals on the calibration error as well, using bootstrap resamples.

[lower, _, upper] = cal.get_calibration_error_uncertainties(logits, labels)


You can find complete code examples in the examples folder. Refer to:

  • examples/ for a simple example in the binary classification setting.
  • examples/ for the multiclass (more than 2 classes) setting.
  • examples/ --- our calibration library also exposes a more customizable interface for advanced users.


If you find this library useful please consider citing our paper:

  author = {A. Kumar and P. Liang and T. Ma},
  booktitle = {Advances in Neural Information Processing Systems (NeurIPS)},
  title = {Verified Uncertainty Calibration},
  year = {2019},

Features, bugs, and contributions

Please feel free to submit feature requests, bug reports, or contribute PRs. Feel free to submit a brief description of a PR before you implement it to get feedback on it, or see how it can fit into our project.

Verified Uncertainty Calibration paper

This repository also contains code for the NeurIPS 2019 (Spotlight) paper Verified Uncertainty Calibration

In our paper, we show that:

  • The calibration error of methods like Platt scaling and temperature scaling are typically underestimated, and cannot be easily measured.
  • We propose an efficient recalibration method where the calibration error can be measured.
  • We show that we can estimate the calibration error with fewer samples (than the standard method) using an estimator from the meteorological literature.


The experiments folder contains experiments for the paper.

We have 4 sets of experiments:

  • Showing the Platt scaling is less calibrated than reported (Section 3)
  • Comparing the scaling binning calibrator with histogram binning on CIFAR-10 and ImageNet (Section 4)
  • Synthetic experiments to validate our theoretical bounds (Section 4)
  • Experiments showing the debiased estimator can estimate calibration error with fewer samples than standard estimator (Section 5) Running each experiment saves plots in the corresponding folder in saved_files

This section is still work in progress. More detailed instructions, and a link to a CodaLab worksheet, are coming soon.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Files for uncertainty-calibration, version 0.0.5
Filename, size File type Python version Upload date Hashes
Filename, size uncertainty-calibration-0.0.5.tar.gz (8.6 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page