Paper Info
Title
Uncertainty-Aware Training for Cardiac Resynchronisation Therapy Response Prediction.
Abstract
Evaluation of predictive deep learning (DL) models beyond conventional performance metrics has become increasingly important for applications in sensitive environments like healthcare. Such models might have the capability to encode and analyse large sets of data but they often lack comprehensive interpretability methods, preventing clinical trust in predictive outcomes. Quantifying uncertainty of a prediction is one way to provide such interpretability and promote trust. However, relatively little attention has been paid to how to include such requirements into the training of the model. In this paper we: (i) quantify the data (aleatoric) and model (epistemic) uncertainty of a DL model for Cardiac Resynchronisation Therapy response prediction from cardiac magnetic resonance images, and (ii) propose and perform a preliminary investigation of an uncertainty-aware loss function that can be used to retrain an existing DL image-based classification model to encourage confidence in correct predictions and reduce confidence in incorrect predictions. Our initial results are promising, showing a significant increase in the (epistemic) confidence of true positive predictions, with some evidence of a reduction in false negative confidence.
Year
DOI
Venue
2021
10.1007/978-3-030-93722-5_21
STACOM@MICCAI
DocType
Citations 
PageRank 
Conference
0
0.34
References 
Authors
0
13
Name
Order
Citations
PageRank
Tareen Dawood100.34
Chen Chen223277.45
Robin Andlauer300.34
Baldeep S. Sidhu400.34
Bram Ruijsink502.70
Justin Gould600.34
Bradley Porter700.34
Mark Elliott811.71
Vishal Mehta911.30
Christopher Aldo Rinaldi10168.07
Esther Puyol-Antón1102.03
Reza Razavi1292994.25
Andrew P. King1348359.98