Evidential segmentation of 3D PET/CT images

Positron Emission Tomography (PET) and Computed Tomography (CT) are two modalities widely used in medical image analysis. Accurately detecting and segmenting lymphomas from these two imaging modalities are critical tasks for cancer staging and radiotherapy planning. However, this task is still challenging due to the complexity of PET/CT images, and the computation cost to process 3D data. In this paper, a segmentation method based on belief functions is proposed to segment lymphomas in 3D PET/CT images. The architecture is composed of a feature extraction module and an evidential segmentation (ES) module. The ES module outputs not only segmentation results (binary maps indicating the presence or absence of lymphoma in each voxel) but also uncertainty maps quantifying the classification uncertainty. The whole model is optimized by minimizing Dice and uncertainty loss functions to increase segmentation accuracy. The method was evaluated on a database of 173 patients with diffuse large b-cell lymphoma. Quantitative and qualitative results show that our method outperforms the state-of-the-art methods.

[1]  S. Jégou,et al.  Fully automatic segmentation of diffuse large B cell lymphoma lesions on 3D FDG-PET/CT for total metabolic tumour volume prediction using a convolutional neural network. , 2020, European Journal of Nuclear Medicine and Molecular Imaging.

[2]  Didier Dubois,et al.  Representations of Uncertainty in AI: Beyond Probability and Possibility , 2020, A Guided Tour of Artificial Intelligence Research.

[3]  Raghav Mehta,et al.  Propagating Uncertainty Across Cascaded Medical Imaging Tasks for Improved Deep Learning Inference , 2019, IEEE Transactions on Medical Imaging.

[4]  Thierry Denoeux,et al.  Evidential fully convolutional network for semantic segmentation , 2021, Applied Intelligence.

[5]  Thierry Denoeux,et al.  A neural network classifier based on Dempster-Shafer theory , 2000, IEEE Trans. Syst. Man Cybern. Part A.

[6]  Seyed-Ahmad Ahmadi,et al.  V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation , 2016, 2016 Fourth International Conference on 3D Vision (3DV).

[7]  Glenn Shafer,et al.  A Mathematical Theory of Evidence , 2020, A Mathematical Theory of Evidence.

[8]  Siqi Li,et al.  DenseX-Net: An End-to-End Model for Lymphoma Segmentation in Whole-Body PET/CT Images , 2020, IEEE Access.

[9]  Henrik Møller,et al.  Defining the optimal method for measuring baseline metabolic tumour volume in diffuse large B cell lymphoma , 2018, European Journal of Nuclear Medicine and Molecular Imaging.

[10]  Didier Dubois,et al.  Representations of Uncertainty in Artificial Intelligence: Probability and Possibility , 2020, A Guided Tour of Artificial Intelligence Research.

[11]  Su Ruan,et al.  Segmentation of heterogeneous or small FDG PET positive tissue based on a 3D-locally adaptive random walk algorithm , 2014, Comput. Medical Imaging Graph..

[12]  Klaus H. Maier-Hein,et al.  Abstract: nnU-Net: Self-adapting Framework for U-Net-Based Medical Image Segmentation , 2019, Bildverarbeitung für die Medizin.

[13]  Pierre Vera,et al.  Detection and segmentation of lymphomas in 3D PET images via clustering with entropy-based optimization strategy , 2019, International Journal of Computer Assisted Radiology and Surgery.

[14]  Thomas Brox,et al.  U-Net: Convolutional Networks for Biomedical Image Segmentation , 2015, MICCAI.

[15]  Andriy Myronenko,et al.  3D MRI brain tumor segmentation using autoencoder regularization , 2018, BrainLes@MICCAI.