Improving Prostate Cancer Segmentation on T2-Weighted MRI Using Prostate Detection and Cascaded Networks

Prostate cancer is one of the most lethal cancers in the male population, and accurate localization of intraprostatic lesions on MRI remains challenging. In this work, we investigate methods for improving prostate cancer segmentation on T2weighted pelvic MRI using cascaded neural networks. We use anonymized dataset of 400 multiparametric MRI studies from two centers, in which experienced radiologists delineated the prostate and clinically significant cancer on the T2 series. Our baseline approach applies 2D and 3D segmentation networks (UNETR, UNET++, SwinUNETR, SegResNetDS, SegResNetVAE) directly to full MRI volumes. We then introduce additional stages that filter slices using DenseNet201 classifiers (cancer/nocancer and prostate/noprostate) and localize the prostate with a YOLObased detector to crop a 3D region of interest before segmentation. Using SwinUNETR as the backbone, the prostate segmentation Dice score increased from 71.37% for direct 3D segmentation to 76.09% when using prostate detection and cropped 3D inputs. For cancer segmentation, the final cascaded pipeline – prostate detection, 3D prostate segmentation, and 3D cancer segmentation within the prostate – improved Dice score from 55.03% for direct 3D segmentation to 67.11%, with a ROC AUC of 0.89 on the test set. These results suggest that cascaded detection and segmentationbased preprocessing of the prostate region can substantially improve automatic prostate cancer segmentation on MRI while remaining compatible with standard segmentation architectures.