Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty Quantification

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cortes-Gomez, Santiago, Patiño, Carlos, Byun, Yewon, Wu, Steven, Horvitz, Eric, Wilder, Bryan
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866916634645495808
author Cortes-Gomez, Santiago
Patiño, Carlos
Byun, Yewon
Wu, Steven
Horvitz, Eric
Wilder, Bryan
author_facet Cortes-Gomez, Santiago
Patiño, Carlos
Byun, Yewon
Wu, Steven
Horvitz, Eric
Wilder, Bryan
contents Interest has been growing in decision-focused machine learning methods which train models to account for how their predictions are used in downstream optimization problems. Doing so can often improve performance on subsequent decision problems. However, current methods for uncertainty quantification do not incorporate any information about downstream decisions. We develop a methodology based on conformal prediction to identify prediction sets that account for a downstream cost function, making them more appropriate to inform high-stakes decision-making. Our approach harnesses the strengths of conformal methods -- modularity, model-agnosticism, and statistical coverage guarantees -- while incorporating downstream decisions and user-specified utility functions. We prove that our methods retain standard coverage guarantees. Empirical evaluation across a range of datasets and utility metrics demonstrates that our methods achieve significantly lower costs than standard conformal methods. We present a real-world use case in healthcare diagnosis, where our method effectively incorporates the hierarchical structure of dermatological diseases. The method successfully generates sets with coherent diagnostic meaning, potentially aiding triage for dermatology diagnosis and illustrating how our method can ground high-stakes decision-making employing domain knowledge.
format Preprint
id arxiv_https___arxiv_org_abs_2410_01767
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty Quantification
Cortes-Gomez, Santiago
Patiño, Carlos
Byun, Yewon
Wu, Steven
Horvitz, Eric
Wilder, Bryan
Machine Learning
Interest has been growing in decision-focused machine learning methods which train models to account for how their predictions are used in downstream optimization problems. Doing so can often improve performance on subsequent decision problems. However, current methods for uncertainty quantification do not incorporate any information about downstream decisions. We develop a methodology based on conformal prediction to identify prediction sets that account for a downstream cost function, making them more appropriate to inform high-stakes decision-making. Our approach harnesses the strengths of conformal methods -- modularity, model-agnosticism, and statistical coverage guarantees -- while incorporating downstream decisions and user-specified utility functions. We prove that our methods retain standard coverage guarantees. Empirical evaluation across a range of datasets and utility metrics demonstrates that our methods achieve significantly lower costs than standard conformal methods. We present a real-world use case in healthcare diagnosis, where our method effectively incorporates the hierarchical structure of dermatological diseases. The method successfully generates sets with coherent diagnostic meaning, potentially aiding triage for dermatology diagnosis and illustrating how our method can ground high-stakes decision-making employing domain knowledge.
title Utility-Directed Conformal Prediction: A Decision-Aware Framework for Actionable Uncertainty Quantification
topic Machine Learning
url https://arxiv.org/abs/2410.01767