Towards Human-AI Complementarity with Prediction Sets

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: De Toni, Giovanni, Okati, Nastaran, Thejaswi, Suhas, Straitouri, Eleni, Gomez-Rodriguez, Manuel
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909385067855872
author De Toni, Giovanni
Okati, Nastaran
Thejaswi, Suhas
Straitouri, Eleni
Gomez-Rodriguez, Manuel
author_facet De Toni, Giovanni
Okati, Nastaran
Thejaswi, Suhas
Straitouri, Eleni
Gomez-Rodriguez, Manuel
contents Decision support systems based on prediction sets have proven to be effective at helping human experts solve classification tasks. Rather than providing single-label predictions, these systems provide sets of label predictions constructed using conformal prediction, namely prediction sets, and ask human experts to predict label values from these sets. In this paper, we first show that the prediction sets constructed using conformal prediction are, in general, suboptimal in terms of average accuracy. Then, we show that the problem of finding the optimal prediction sets under which the human experts achieve the highest average accuracy is NP-hard. More strongly, unless P = NP, we show that the problem is hard to approximate to any factor less than the size of the label set. However, we introduce a simple and efficient greedy algorithm that, for a large class of expert models and non-conformity scores, is guaranteed to find prediction sets that provably offer equal or greater performance than those constructed using conformal prediction. Further, using a simulation study with both synthetic and real expert predictions, we demonstrate that, in practice, our greedy algorithm finds near-optimal prediction sets offering greater performance than conformal prediction.
format Preprint
id arxiv_https___arxiv_org_abs_2405_17544
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Human-AI Complementarity with Prediction Sets
De Toni, Giovanni
Okati, Nastaran
Thejaswi, Suhas
Straitouri, Eleni
Gomez-Rodriguez, Manuel
Machine Learning
Computers and Society
Human-Computer Interaction
Decision support systems based on prediction sets have proven to be effective at helping human experts solve classification tasks. Rather than providing single-label predictions, these systems provide sets of label predictions constructed using conformal prediction, namely prediction sets, and ask human experts to predict label values from these sets. In this paper, we first show that the prediction sets constructed using conformal prediction are, in general, suboptimal in terms of average accuracy. Then, we show that the problem of finding the optimal prediction sets under which the human experts achieve the highest average accuracy is NP-hard. More strongly, unless P = NP, we show that the problem is hard to approximate to any factor less than the size of the label set. However, we introduce a simple and efficient greedy algorithm that, for a large class of expert models and non-conformity scores, is guaranteed to find prediction sets that provably offer equal or greater performance than those constructed using conformal prediction. Further, using a simulation study with both synthetic and real expert predictions, we demonstrate that, in practice, our greedy algorithm finds near-optimal prediction sets offering greater performance than conformal prediction.
title Towards Human-AI Complementarity with Prediction Sets
topic Machine Learning
Computers and Society
Human-Computer Interaction
url https://arxiv.org/abs/2405.17544