Region-Based Evidential Deep Learning to Quantify Uncertainty and Improve Robustness of Brain Tumor Segmentation

08/11/2022
by   Hao Li, et al.
17

Despite recent advances in the accuracy of brain tumor segmentation, the results still suffer from low reliability and robustness. Uncertainty estimation is an efficient solution to this problem, as it provides a measure of confidence in the segmentation results. The current uncertainty estimation methods based on quantile regression, Bayesian neural network, ensemble, and Monte Carlo dropout are limited by their high computational cost and inconsistency. In order to overcome these challenges, Evidential Deep Learning (EDL) was developed in recent work but primarily for natural image classification. In this paper, we proposed a region-based EDL segmentation framework that can generate reliable uncertainty maps and robust segmentation results. We used the Theory of Evidence to interpret the output of a neural network as evidence values gathered from input features. Following Subjective Logic, evidence was parameterized as a Dirichlet distribution, and predicted probabilities were treated as subjective opinions. To evaluate the performance of our model on segmentation and uncertainty estimation, we conducted quantitative and qualitative experiments on the BraTS 2020 dataset. The results demonstrated the top performance of the proposed method in quantifying segmentation uncertainty and robustly segmenting tumors. Furthermore, our proposed new framework maintained the advantages of low computational cost and easy implementation and showed the potential for clinical application.

READ FULL TEXT

page 7

page 9

page 10

research
06/19/2022

TBraTS: Trusted Brain Tumor Segmentation

Despite recent improvements in the accuracy of brain tumor segmentation,...
research
06/08/2018

Uncertainty-driven Sanity Check: Application to Postoperative Brain Tumor Cavity Segmentation

Uncertainty estimates of modern neuronal networks provide additional inf...
research
04/04/2023

Uncertainty estimation in Deep Learning for Panoptic segmentation

As deep learning-based computer vision algorithms continue to improve an...
research
05/09/2023

Trustworthy Multi-phase Liver Tumor Segmentation via Evidence-based Uncertainty

Multi-phase liver contrast-enhanced computed tomography (CECT) images co...
research
06/14/2021

Embracing Uncertainty in "Small Data" Problems: Estimating Earthquakes from Historical Anecdotes

We apply the Bayesian inversion process to make principled estimates of ...
research
06/05/2018

Evidential Deep Learning to Quantify Classification Uncertainty

Deterministic neural nets have been shown to learn effective predictors ...
research
01/15/2018

An Efficient Method for Uncertainty Propagation in Robust Software Performance Estimation

Software engineers often have to estimate the performance of a software ...

Please sign up or login with your details

Forgot password? Click here to reset