Prostate MR Image Segmentation Using a Multi-Stage Network Approach

Prostate cancer (PCa) remains one of the most prevalent cancers among men, with over 1.4 million new cases and 375,304 deaths reported globally in 2020. Current diagnostic approaches, such as prostate specific antigen (PSA) testing and trans-rectal ultrasound (TRUS)-guided biopsies, are often limited by low specificity and accuracy. This study addresses these limitations by leveraging deep learning-based image segmentation techniques on a dataset comprising 61,119 T2-weighted MR images from 1,151 patients to enhance PCa detection and characterisation. A multi-stage segmentation approach, including One-Stage, Sequential Two-Stage, and End-to-End Two-Stage methods, was evaluated using various deep learning architectures. The MultiResUNet model, integrated into a multi-stage segmentation framework, demonstrated significant improvements in delineating prostate boundaries. The study utilised a dataset of over 61,000 T2-weighted magnetic resonance (MR) images from more than 1,100 patients, employing three distinct segmentation strategies: One-Stage, Sequential Two-Stage, and End-to-End Two-Stage methods. The end-to-end approach, leveraging shared feature representations, consistently outperformed other methods, underscoring its effectiveness in enhancing diagnostic accuracy. These findings highlight the potential of advanced deep learning architectures in streamlining prostate cancer detection and treatment planning. Future work will focus on further optimisation of the models and assessing their generalisability to diverse medical imaging contexts.