Metric-Guided Conformal Bounds for Probabilistic Image Reconstruction

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Cheung, Matt Y, Netherton, Tucker J, Court, Laurence E, Veeraraghavan, Ashok, Balakrishnan, Guha
Format: Preprint
Veröffentlicht: 2024
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866911176890253312
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 Modern deep learning reconstruction algorithms generate impressively realistic scans from sparse inputs, but can often produce significant inaccuracies. This makes it difficult to provide statistically guaranteed claims about the true state of a subject from scans reconstructed by these algorithms. In this study, we propose a framework for computing provably valid prediction bounds on claims derived from probabilistic black-box image reconstruction algorithms. The key insights behind our framework are to represent reconstructed scans with a derived clinical metric of interest, and to calibrate bounds on the ground truth metric with conformal prediction (CP) using a prior calibration dataset. These bounds convey interpretable feedback about the subject's state, and can also be used to retrieve nearest-neighbor reconstructed scans for visual inspection. We demonstrate the utility of this framework on sparse-view computed tomography (CT) for fat mass quantification and radiotherapy planning tasks. Results show that our framework produces bounds with better semantical interpretation than conventional pixel-based bounding approaches. Furthermore, we can flag dangerous outlier reconstructions that look plausible but have statistically unlikely metric values.
format Preprint
id arxiv_https___arxiv_org_abs_2404_15274
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Metric-Guided Conformal Bounds for Probabilistic Image Reconstruction
Cheung, Matt Y
Netherton, Tucker J
Court, Laurence E
Veeraraghavan, Ashok
Balakrishnan, Guha
Machine Learning
Computer Vision and Pattern Recognition
Image and Video Processing
Medical Physics
Modern deep learning reconstruction algorithms generate impressively realistic scans from sparse inputs, but can often produce significant inaccuracies. This makes it difficult to provide statistically guaranteed claims about the true state of a subject from scans reconstructed by these algorithms. In this study, we propose a framework for computing provably valid prediction bounds on claims derived from probabilistic black-box image reconstruction algorithms. The key insights behind our framework are to represent reconstructed scans with a derived clinical metric of interest, and to calibrate bounds on the ground truth metric with conformal prediction (CP) using a prior calibration dataset. These bounds convey interpretable feedback about the subject's state, and can also be used to retrieve nearest-neighbor reconstructed scans for visual inspection. We demonstrate the utility of this framework on sparse-view computed tomography (CT) for fat mass quantification and radiotherapy planning tasks. Results show that our framework produces bounds with better semantical interpretation than conventional pixel-based bounding approaches. Furthermore, we can flag dangerous outlier reconstructions that look plausible but have statistically unlikely metric values.
title Metric-Guided Conformal Bounds for Probabilistic Image Reconstruction
topic Machine Learning
Computer Vision and Pattern Recognition
Image and Video Processing
Medical Physics
url https://arxiv.org/abs/2404.15274