Attention-guided multi-scale learning network for automatic prostate and tumor segmentation on MRI

Abstract

Background and objective: Image-guided clinical diagnosis can be achieved by automatically and accurately segmenting prostate and prostatic cancer in male pelvic magnetic resonance imaging (MRI) images. For accurate tumor removal, the location, number, and size of prostate cancer are crucial, especially in surgical patients. The morphological differences between the prostate and tumor regions are small, the size of the tumor is uncertain, the boundary between the tumor and surrounding tissue is blurred, and the classification that separates the normal region from the tumor is uneven. Therefore, segmenting prostate and tumor on MRI images is challenging.

Methods: This study offers a new prostate and prostatic cancer segmentation network based on double branch attention driven multi-scale learning for MRI. To begin, the dual branch structure provides two input images with different scales for feature coding, as well as a multi-scale attention module that collects details from different scales. The features of the double branch structure are then entered into the built feature fusion module to get more complete context information. Finally, to give a more precise learning representation, each stage is built using a deep supervision mechanism.

Results: The results of our proposed network's prostate and tumor segmentation on a variety of male pelvic MRI data sets show that it outperforms existing techniques. For prostate and prostatic cancer MRI segmentation, the dice similarity coefficient (DSC) values were 91.65% and 84.39%, respectively.

Conclusions: Our method maintains high correlation and consistency between automatic segmentation results and expert manual segmentation results. Accurate automatic segmentation of prostate and prostate cancer has important clinical significance.

Keywords: Deep supervision; Diffusion weighted imaging; Feature fusion; Multiscale attention; Prostatic cancer.