- This repository provides the code for our accepted MICCAI'2022 paper "TBraTS: Trusted Brain Tumor Segmentation"
- Official implementation of TBraTS: Trusted Brain Tumor Segmentation
Despite recent improvements in the accuracy of brain tumor segmentation, the results still exhibit low levels of confidence and robustness. Uncertainty estimation is one effective way to change this situation, as it provides a measure of confidence in the segmentation results. In this paper, we propose a trusted brain tumor segmentation network which can generate robust segmentation results and reliable uncertainty estimations without excessive computational burden and modification of the backbone network. In our method, uncertainty is modeled explicitly using subjective logic theory, which treats the predictions of backbone neural network as subjective opinions by parameterizing the class probabilities of the segmentation as a Dirichlet distribution. Meanwhile, the trusted segmentation framework learns the function that gathers reliable evidence from the feature leading to the final segmentation results. Overall, our unified trusted segmentation framework endows the model with reliability and robustness to out-of-distribution samples. To evaluate the effectiveness of our model in robustness and reliability, qualitative and quantitative experiments are conducted on the BraTS 2019 dataset.
Some important required packages include:
Pytorch version >=0.4.1.
Visdom
Python == 3.7
Some basic python packages such as Numpy.
- The multimodal brain tumor datasets (BraTS 2019) could be acquired from here.
After downloading the dataset from here, data preprocessing is needed which is to convert the .nii files as .pkl files and realize date normalization.
Follow the python3 data/preprocessBraTS.py which is referenced from the TransBTS
Run the python3 trainTBraTS.py : your own backbone with our framework(U/V/AU/TransBTS)
Run the python3 train.py : the backbone without our framework
- [09/17] More experiments on trustworthy medical image segmentation please refer to UMIS.
- [09/17] We released all the codes.
- [06/05] We will release the code as soon as possible.
- [06/13] We have uploaded the main part of our code. We will upload all the code after camera-ready.
- [06/22] Our pre-printed version of the paper is available at TBraTS: Trusted Brain Tumor Segmentation
If you find our work is helpful for your research, please consider to cite:
@inproceedings{Coco2022TBraTS,
title={TBraTS: Trusted Brain Tumor Segmentation},
author={Zou, Ke and Yuan, Xuedong and Shen, Xiaojing and Wang, Meng and Fu, Huazhu},
booktitle={International Conference on Medical Image Computing and Computer Assisted Intervention (MICCAI)},
year={2022}
}
or
@article{zou2022tbrats,
title={TBraTS: Trusted Brain Tumor Segmentation},
author={Ke Zou and Xuedong Yuan and Xiaojing Shen and Meng Wang and Huazhu Fu},
year={2022},
journal={arXiv preprint arXiv:2206.09309},
}