Quantification of Uncertainty in Brain Tumor Segmentation using Generative Network and Bayesian Active Learning

Rasha Alshehhi, Anood Alshehhi

2021

Abstract

Convolutional neural networks have shown great potential in medical segmentation problems, such as brain-tumor segmentation. However, little consideration has been given to generative adversarial networks and uncertainty quantification over the output images. In this paper, we use the generative adversarial network to handle limited labeled images. We also quantify the modeling uncertainty by utilizing Bayesian active learning to reduce untoward outcomes. Bayesian active learning is dependent on selecting uncertain images using acquisition functions to increase accuracy. We introduce supervised acquisition functions based on distance functions between ground-truth and predicted images to quantify segmentation uncertainty. We evaluate the method by comparing it with the state-of-the-art methods based on Dice score, Hausdorff distance and sensitivity. We demonstrate that the proposed method achieves higher or comparable performance to state-of-the-art methods for brain tumor segmentation (on BraTS 2017, BraTS 2018 and BraTS 2019 datasets).

Paper Citation

in Harvard Style

Alshehhi R. and Alshehhi A. (2021). Quantification of Uncertainty in Brain Tumor Segmentation using Generative Network and Bayesian Active Learning. In Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 4: VISAPP; ISBN 978-989-758-488-6, SciTePress, pages 701-709. DOI: 10.5220/0010341007010709

in Bibtex Style

@conference{visapp21,
author={Rasha Alshehhi and Anood Alshehhi},
title={Quantification of Uncertainty in Brain Tumor Segmentation using Generative Network and Bayesian Active Learning},
booktitle={Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 4: VISAPP},
year={2021},
pages={701-709},
publisher={SciTePress},
organization={INSTICC},
doi={10.5220/0010341007010709},
isbn={978-989-758-488-6},
}

in EndNote Style

TY - CONF

JO - Proceedings of the 16th International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2021) - Volume 4: VISAPP
TI - Quantification of Uncertainty in Brain Tumor Segmentation using Generative Network and Bayesian Active Learning
SN - 978-989-758-488-6
AU - Alshehhi R.
AU - Alshehhi A.
PY - 2021
SP - 701
EP - 709
DO - 10.5220/0010341007010709
PB - SciTePress