Dempster-Shafer Theory Empowered Deep Learning for Enhanced 3D PET-CT Medical Image Segmentation: A First Test on Automated Lesion Segmentation in Whole-body Dataset

Siyuan Gu,Zhicheng Dai,Licheng Wu,Wanyu Liu,Lunde Chen
DOI: https://doi.org/10.1109/nnice61279.2024.10498919
2024-01-01
Abstract:In recent years, the fusion of deep learning with Dempster-Shafer Theory has shown great promise in medical image analysis, particularly for its capacity to model uncertainty and enhance accuracy. This paper presents an ongoing study utilizing deep evidential networks for Lymphoma segmentation in three-dimensional Positron Emission Tomography (PET) and Computed Tomography (CT) images. Expanding on the Evidential Neural Network U-Net (ENN-UNet), which integrates UNet with Dempster-Shafer Theory, we explore various feature extractors (3D-UNet, V-Net, 3D ResNet, HighRes3DNet, 3D DenseNet, and DeepMedic) incorporated into a Dempster-Shafer layer to boost segmentation results. Our primary experiments, focusing on ENN-based segmentation methods (ENN-UNet and ENN-VNet), conducted on a mixed dataset of 501 samples, encompassing melanoma, lymphoma, and lung cancer, reveal that the inclusion of an evidential layer can enhance the model's performance in complex tasks.
What problem does this paper attempt to address?