Uncertainty quantification in automated tumor segmentation using deep learning

• Main Author: Abbas, Syed Ahmar
• Format: Dissertation
• Language: English
• Subjects: Monte carlo dropouts; Tumor segmentation; Uncertainty

Introduction Head and neck cancer is one of the leading causes of cancer-related deaths globally and arguably has a long-standing history of impacting human life both medically and economically. Common treatment options which are considered most effective require early and precise delineation of tumors. But this is not an easy task as it requires hours of discussion and iterations for every patient and clinical expertise, and is also prone to human error. Due to advancements in medical imaging and deep learning, particularly with con- volutional neural networks (CNN), automatic segmentation of tumors has become a hot topic for researchers, and results from different studies have shown prom- ising outcomes. However, these auto delineation algorithms are still far from per- fect, and given the nature of their use, their efficacy in the clinical environment is a hot debate. One of the reasons for the low acceptance of these CNN-based mod- els is their indecipherable black-box nature and inability to quantify and visualize confidence in their delineations. In this thesis, we have proposed monte carlo dropouts based technique to visu- alize uncertainty in the predictions of convolutional neural networks using V-net architecture to increase the interpretability of the model. This is done through visualizing uncertainties in the input feature selection of the model and also its predictions. Moreover, we have tabled a novel approach to quantify confidence in predictions with a single comparable value as a percentage for easy interpretabil- ity of the auto-segmentation to the clinicians. Methodology Monte carlo methods are used to obtain the probabilistic distribution of the out- comes for a numerical problem by repeated sampling. In this thesis, we are using the same approach to find uncertainties in the prediction and input feature se- lection of CNN-based automatic tumor segmentation model. The dataset being used for this study consists of 197 patients diagnosed with head and neck tumors. For repeated sampling, we have used dropout layers with two different rates in the model to randomly disable neurons while making predictions, this setup ensures to have slightly different predictions for the same image in each iteration. The vari- ance in these predictions of each voxel is then visualized as an uncertainty map of the prediction, higher variance defines an uncertain region. For input feature importance, we have used guided backpropagation which high- lights onl