Combining Human Predictions with Model Probabilities via Confusion Matrices and CalibrationDownload PDF

21 May 2021, 20:51 (edited 24 Jan 2022)NeurIPS 2021 PosterReaders: Everyone
  • Keywords: human-machine, human-AI, human-in-the-loop, calibration, classification
  • TL;DR: We combine the class-level output of a human with the probabilistic output of a classifier in order to achieve low misclassification rates.
  • Abstract: An increasingly common use case for machine learning models is augmenting the abilities of human decision makers. For classification tasks where neither the human nor model are perfectly accurate, a key step in obtaining high performance is combining their individual predictions in a manner that leverages their relative strengths. In this work, we develop a set of algorithms that combine the probabilistic output of a model with the class-level output of a human. We show theoretically that the accuracy of our combination model is driven not only by the individual human and model accuracies, but also by the model's confidence. Empirical results on image classification with CIFAR-10 and a subset of ImageNet demonstrate that such human-model combinations consistently have higher accuracies than the model or human alone, and that the parameters of the combination method can be estimated effectively with as few as ten labeled datapoints.
  • Supplementary Material: pdf
  • Code Of Conduct: I certify that all co-authors of this work have read and commit to adhering to the NeurIPS Statement on Ethics, Fairness, Inclusivity, and Code of Conduct.
  • Code: https://github.com/GavinKerrigan/conf_matrix_and_calibration
12 Replies

Loading