Estimation of uncertainty in Brain Tumor segmentation using modified multistage 3D-UNet on multimodal MRI images

Authors

  • Bhavesh Parmar Gujarat Technological University
  • Mehul Parikh L. D. College of Engineering, Ahmedabad

DOI:

https://doi.org/10.62110/sciencein.jist.2024.v12.802

Keywords:

Brain Tumor Segmentation, uncertainty, MRI Images, 3D UNet, Deep Learning

Abstract

Automated brain tumor segmentation is challenging due to the tumor tissues' shape, size, and appearance. Various methods used multi-mode MRI scans to segment sub-regions of brain tumors.  3D CNN methods improved performance in recent years, but most methods do not use uncertainty information in segmentation. For reliability and understanding, model prediction is vital for clinical decisions. This work studies three models namely 3D-UNet, Modified 3D-UNet, and Modified Multistage-3D-UNet for brain tumor segmentation. MRI volume bias correction and normalization were carried out using z-score normalization. Two patch generation strategies reduce memory use and class imbalance. Voxel-wise uncertainty evaluation was made for aleatoric and epistemic uncertainties using test time augmentation and dropout, respectively. Variance and entropy are used to measure the uncertainty of the modified multistage-3D-Unet segmentation model from ground truth. Variance creates separate uncertainty maps for each tumor sub-regions, whereas, entropy provides only global information. Uncertainty is used to filter miss-segmented predictions and improve accuracy. Uncertainty awareness increases model accuracy with dice scores of 0.93, 0.91, and 0.83 for tumor sub-regions WT, TC, and ET respectively.

URN:NBN:sciencein.jist.2024.v12.802

Downloads

Download data is not yet available.

Author Biography

  • Bhavesh Parmar, Gujarat Technological University

    L. D. College of Engineering, Ahmedabad

Brain Tumor segmentation using U-Net

Downloads

Published

2024-02-02

Issue

Section

Biomedical and Pharmaceutical Sciences

URN

How to Cite

Estimation of uncertainty in Brain Tumor segmentation using modified multistage 3D-UNet on multimodal MRI images. (2024). Journal of Integrated Science and Technology, 12(5), 802. https://doi.org/10.62110/sciencein.jist.2024.v12.802

Similar Articles

1-10 of 66

You may also start an advanced similarity search for this article.