A lightweight deep learning model for automated segmentation of Gynecologic organs and cervical Tumors on T2-weighted magnetic resonance imaging.
Authors
Affiliations (4)
Affiliations (4)
- Department of Imaging Physics, The University of Texas MD Anderson Cancer Center, Houston, TX, USA.
- Department of Biostatistics, The University of Texas MD Anderson Cancer Center, Houston, TX, USA.
- Department of Radiation Oncology, The University of Texas MD Anderson Cancer Center, Houston, TX, USA.
- Department of Abdominal Imaging, The University of Texas MD Anderson Cancer Center, Houston, TX, USA.
Abstract
<b>Background and Purpose:</b> Manual segmentation of gynecologic organs and cervical tumors for radiotherapy planning is time-consuming and variable. Automated segmentation on routine T2-weighted magnetic resonance imaging (MRI) remains limited. The aim of this study was to evaluate a lightweight deep learning model for automated segmentation of gynecologic organs and cervical tumors on T2-weighted MRI. <b>Materials and Methods:</b> This work applied a two-stage lightweight deep learning model (PocketNet) to segment the cervix, vagina, uterus, and tumor(s) on T2-weighted MRI in 102 patients with cervical cancer undergoing definitive radiotherapy. Model performance was assessed using the Dice-Sorensen coefficient (DSC) and 95th percentile Hausdorff distance (Haus95) on internal data and validated on an external dataset. A full nnU-Net model trained on the same internal dataset served as a benchmark for segmentation accuracy and computational efficiency. <b>Results:</b> On the institutional dataset, PocketNet achieved mean DSC values exceeding 70% for tumor segmentation and 80% for organ segmentation. External validation on The Cancer Imaging Archive (TCIA) Cervical Cancer Tumor Heterogeneity (CCTH) collection demonstrated the model's robustness, achieving DSC scores of 67.3% for tumor segmentation and 80.8% for organ segmentation. Compared to the PocketNet architecture, nnUNet achieved similar accuracy but required approximately twice the training time, more than 35 times as many parameters, and 40 times more memory for model storage. <b>Conclusion:</b> The PocketNet architecture provides reliable automated segmentation of gynecologic organs and cervical tumors on T2-weighted MRI, with performance comparable to a full-sized nnUNet while requiring substantially less memory and training time, supporting its potential integration into time-sensitive radiotherapy workflows.