Auto-segmentation of pelvic organs at risk on 0.35T MRI using 2D and 3D Generative Adversarial Network models

Marica Vagni,Huong Elena Tran,Angela Romano,Giuditta Chiloiro,Luca Boldrini,Konstantinos Zormpas-Petridis,Maria Kawula,Guillaume Landry,Christopher Kurz,Stefanie Corradini,Claus Belka,Luca Indovina,Maria Antonietta Gambacorta,Lorenzo Placidi,Davide Cusumano
DOI: https://doi.org/10.1016/j.ejmp.2024.103297
IF: 3.119
2024-02-05
Physica Medica
Abstract:Highlights • GANs segment organs-at-risk in pelvic 0.35T MRIs with high accuracy. • Segmented organs: rectum, bladder and femoral heads (left and right) • Our networks generalised well on an external independent dataset. • 3D GAN outperforms its 2D equivalent in both accuracy and generation time. • This is the first attempt of organ auto-segmentation in 0.35T MRI with GANs. Purpose Manual recontouring of targets and Organs At Risk (OARs) is a time-consuming and operator-dependent task. We explored the potential of Generative Adversarial Networks (GAN) to auto-segment the rectum, bladder and femoral heads on 0.35T MRIs to accelerate the online MRI-guided-Radiotherapy (MRIgRT) workflow. Methods 3D planning MRIs from 60 prostate cancer patients treated with 0.35T MR-Linac were collected. A 3D GAN architecture and its equivalent 2D version were trained, validated and tested on 40, 10 and 10 patients respectively. The volumetric Dice Similarity Coefficient (DSC) and 95th percentile Hausdorff Distance (HD95 th ) were computed against expert drawn ground-truth delineations. The networks were also validated on an independent external dataset of 16 patients. Results In the internal test set, the 3D and 2D GANs showed DSC/HD95 th of 0.83/9.72 mm and 0.81/10.65 mm for the rectum, 0.92/5.91 mm and 0.85/15.72 mm for the bladder, and 0.94/3.62 mm and 0.90/9.49 mm for the femoral heads. In the external test set, the performance was 0.74/31.13 mm and 0.72/25.07 mm for the rectum, 0.92/9.46 mm and 0.88/11.28 mm for the bladder, and 0.89/7.00 mm and 0.88/10.06 mm for the femoral heads. The 3D and 2D GANs required on average 1.44 s and 6.59 s respectively to generate the OARs' volumetric segmentation for a single patient. Conclusions The proposed 3D GAN auto-segments pelvic OARs with high accuracy on 0.35T, in both the internal and the external test sets, outperforming its 2D equivalent in both segmentation robustness and volume generation time.
radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?