AuthorsF. Jacobsen, S. Hicks, P. Halvorsen and M. Riegler
TitleEstimating Predictive Uncertainty in Gastrointestinal Polyp Segmentation
AfilliationMachine Learning
Project(s)Department of Holistic Systems
StatusPublished
Publication TypeProceedings, refereed
Year of Publication2022
Conference Name2022 IEEE 35th International Symposium on Computer-Based Medical Systems (CBMS)
Pagination44-49
PublisherIEEE
Place PublishedShenzen, China
Abstract

Deep neural networks have achieved state-of-the-art performance on numerous applications in the medical field, with use-cases ranging from automation of mundane tasks to diagnosis of life-threatening diseases. Despite these achievements, deep neural networks are considered “black boxes” due to their complex structure and general lack of transparency in their decision-making process. These attributes make it challenging to incorporate deep learning into existing clinical workflows as decisions often need more support than blind faith in a statistical model. This paper presents an investigation of uncertainty estimation for the detection of colon polyps using deep convolutional neural networks (CNNs). We experiment with two different approaches to measure uncertainty, Monte Carlo (MC) dropout and deep ensembles, and discuss the advantages and disadvantages of both methods in terms of computational efficiency and performance gain. Furthermore, we apply the two uncertainty methods to two different state-of-the-art CNN-based polyp segmentation architectures. The uncertainty is visualized as heatmaps on the input images and can be used to make more informed decisions on whether or not to trust a model's predictions. The results show that the predictive uncertainties provide a comparison between different models' predictions which can be interpreted as contrastive explanations where the values are largely influenced by the degree of independence between the models in the ensemble. We also reveal that MC dropout is shown to lack at providing contrastive uncertainty values due to the high correlation between the models' in the ensemble.

URLhttps://ieeexplore.ieee.org/document/9867012/http://xplorestaging.ieee.org/ielx7/9866947/9866899/09867012.pdf?arnumber=9867012
DOI10.1109/CBMS55023.2022.00015
Citation Key42725