Bias-Aware Conformal Prediction for Metric-Based Imaging Pipelines

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Cheung, Matt Y., Netherton, Tucker J., Court, Laurence E., Veeraraghavan, Ashok, Balakrishnan, Guha
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866908786944376832
author Cheung, Matt Y.
Netherton, Tucker J.
Court, Laurence E.
Veeraraghavan, Ashok
Balakrishnan, Guha
author_facet Cheung, Matt Y.
Netherton, Tucker J.
Court, Laurence E.
Veeraraghavan, Ashok
Balakrishnan, Guha
contents Reliable confidence measures of metrics derived from medical imaging reconstruction pipelines would improve the standard of decision-making in many clinical workflows. Conformal Prediction (CP) provides a robust framework for producing calibrated prediction intervals, but standard CP formulations face a critical challenge in the imaging pipeline: common mismatches between image reconstruction objectives and downstream metrics can introduce systematic prediction deviations from ground truth values, known as bias. These biases in turn compromise the efficiency of prediction intervals, which is a problem that has been unexplored in the CP literature. In this study, we formalize the behavior of symmetric (where bounds expand equally in both directions) and asymmetric (where bounds expand unequally) formulations for common non-conformity scores in CP in the presence of bias, and argue that this measurable bias must inform the choice of CP formulation. We theoretically and empirically demonstrate that symmetric intervals are inflated by a factor of two times the magnitude of bias while asymmetric intervals remain unaffected by bias, and provide conditions under which each formulation produces tighter intervals. We empirically validated our theoretical analyses on sparse-view CT reconstruction for downstream radiotherapy planning. Our work enables users of medical imaging pipelines to proactively select optimal CP formulations, thereby improving interval length efficiency for critical downstream metrics.
format Preprint
id arxiv_https___arxiv_org_abs_2410_05263
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Bias-Aware Conformal Prediction for Metric-Based Imaging Pipelines
Cheung, Matt Y.
Netherton, Tucker J.
Court, Laurence E.
Veeraraghavan, Ashok
Balakrishnan, Guha
Machine Learning
Artificial Intelligence
Statistics Theory
Methodology
Reliable confidence measures of metrics derived from medical imaging reconstruction pipelines would improve the standard of decision-making in many clinical workflows. Conformal Prediction (CP) provides a robust framework for producing calibrated prediction intervals, but standard CP formulations face a critical challenge in the imaging pipeline: common mismatches between image reconstruction objectives and downstream metrics can introduce systematic prediction deviations from ground truth values, known as bias. These biases in turn compromise the efficiency of prediction intervals, which is a problem that has been unexplored in the CP literature. In this study, we formalize the behavior of symmetric (where bounds expand equally in both directions) and asymmetric (where bounds expand unequally) formulations for common non-conformity scores in CP in the presence of bias, and argue that this measurable bias must inform the choice of CP formulation. We theoretically and empirically demonstrate that symmetric intervals are inflated by a factor of two times the magnitude of bias while asymmetric intervals remain unaffected by bias, and provide conditions under which each formulation produces tighter intervals. We empirically validated our theoretical analyses on sparse-view CT reconstruction for downstream radiotherapy planning. Our work enables users of medical imaging pipelines to proactively select optimal CP formulations, thereby improving interval length efficiency for critical downstream metrics.
title Bias-Aware Conformal Prediction for Metric-Based Imaging Pipelines
topic Machine Learning
Artificial Intelligence
Statistics Theory
Methodology
url https://arxiv.org/abs/2410.05263