Conformal Prediction Sets Improve Human Decision Making

Jesse C. Cresswell, Yi Sui, Bhargava Kumar, Noël Vouitsis
1/24/2024
cs.LGcs.HCstat.ML

No official code found. Best implementation: layer6ai-labs/hitl-conformal-prediction (6 stars, MIT, last commit 2024-05-12).

Abstract

In response to everyday queries, humans explicitly signal uncertainty and offer alternative answers when they are unsure. Machine learning models that output calibrated prediction sets through conformal prediction mimic this human behaviour; larger sets signal greater uncertainty while providing alternatives. In this work, we study the usefulness of conformal prediction sets as an aid for human decision making by conducting a pre-registered randomized controlled trial with conformal prediction sets provided to human subjects. With statistical significance, we find that when humans are given conformal prediction sets their accuracy on tasks improves compared to fixed-size prediction sets with the same coverage guarantee. The results show that quantifying model uncertainty with conformal prediction is helpful for human-in-the-loop decision making and human-AI teams.

View on arXivView PDF

Code Implementations(1)

MIT

Human in the Loop Conformal Prediction

62PythonJan 22, 20242 years agoMIT

Similar Papers

Cite this paper

@article{cresswell2024conformal,
  title  = {Conformal Prediction Sets Improve Human Decision Making},
  author = {Jesse C. Cresswell and Yi Sui and Bhargava Kumar and Noël Vouitsis},
  year   = {2024},
  eprint = {2401.13744},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2401.13744v3},
  journal = {ICML 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/2401.13744.svg)](https://codeofpaper.com/paper/2401.13744)

Discussion