2023/05/05 by Joshua Durso-Finley, Durso-Finley, Joshua, Jean-Pierre Falet +9 · 1 citation
Biochemistry, Genetics and Molecular Biology · Economics, Econometrics and Finance · Medicine · #Cell Image Analysis Techniques #FOS: Computer and information sciences #Health Systems, Economic Evaluations, Quality of Life #Machine Learning (cs.LG) #Radiomics and Machine Learning in Medical Imaging
paper · pdf · doi:10.48550/arxiv.2305.03829
openalex publication_date 2023/05/05 · openalex created_date 2023/05/18 · openalex updated_date 2026/07/28
Image-based precision medicine aims to personalize treatment decisions based on an individual's unique imaging features so as to improve their clinical outcome. Machine learning frameworks that integrate uncertainty estimation as part of their treatment recommendations would be safer and more reliable. However, little work has been done in adapting uncertainty estimation techniques and validation metrics for precision medicine. In this paper, we use Bayesian deep learning for estimating the posterior distribution over factual and counterfactual outcomes on several treatments. This allows for estimating the uncertainty for each treatment option and for the individual treatment effects (ITE) between any two treatments. We train and evaluate this model to predict future new and enlarging T2 lesion counts on a large, multi-center dataset of MR brain images of patients with multiple sclerosis, exposed to several treatments during randomized controlled trials. We evaluate the correlation of the uncertainty estimate with the factual error, and, given the lack of ground truth counterfactual outcomes, demonstrate how uncertainty for the ITE prediction relates to bounds on the ITE error. Lastly, we demonstrate how knowledge of uncertainty could modify clinical decision-making to improve individual patient and clinical trial outcomes.