Graph Cut Energy Minimization in a Probabilistic Learning Framework for 3D Prostate Segmentation in MRI

Soumya Ghose (1) , Jhimli Mitra (1) , Arnau Oliver (2) , Robert Marti (2) , Xavier Llado (2) , Jordi Freixenet (2) , Joan C. Vilanova (2) , Désiré Sidibé (1) , Fabrice Mériaudeau (1)

1 Le2i - Laboratoire Electronique, Informatique et Image [UMR6306]
2 VICOROB - Visio per computador i robotica

Variations in inter-patient prostate shape, and size and imaging artifacts in magnetic resonance images (MRI) hinders automatic accurate prostate segmentation. In this paper we propose a graph cut based energy minimization of the posterior probabilities obtained in a supervised learning schema for automatic 3D segmentation of the prostate in MRI. A probabilistic classification of the prostate voxels is achieved with a probabilistic atlas and a random forest based learning framework. The posterior probabilities are combined to obtain the likelihood of a voxel being prostate. Finally, 3D graph cut based energy minimization in the stochastic space provides segmentation of the prostate. The proposed method achieves a mean Dice similarity coefficient (DSC) value of 0.91±0.04 and 95% mean Hausdorff distance (HD) of 4.69±2.62 voxels when validated with 15 prostate volumes of a public dataset in a leave-one-patient-out validation framework. The model achieves statistically significant t-test p-value<0.0001 in mean DSC and mean HD values compared to some of the works in literature.

Domaines: Imagerie médicale

Soumya Ghose, Jhimli Mitra, Arnau Oliver, Robert Marti, Xavier Llado, et al.. Graph Cut Energy Minimization in a Probabilistic Learning Framework for 3D Prostate Segmentation in MRI. IAPR International Conference on Pattern Recognition, Nov 2012, Tsukba, Japan. ⟨hal-00710964⟩