PZS-Net: Incorporating of Frame Sequence and Multi-Scale Priors for Prostate Zonal Segmentation in Transrectal Ultrasound

Abstract

Transrectal ultrasound (TRUS) videos offer valuable histopathologic information about the prostate. Accurate prostate zonal segmentation in TRUS videos is vital for diagnosing prostate cancer and guiding surgery. However, TRUS videos are manually recorded by urologists, resulting in no standardized coordinate system, which limits direct prostate zonal segmentation in these videos. To overcome the limitation, a novel Prostate Zonal Segmentation Network (PZS-Net), based on U-Net, which learns critical cross-frame information and multi-scale features from sequential frames, is proposed. First, a sequential frame cross-attention (SFCA) module is designed to capture remote information from sequential frames to enhance the feature representation of the current frame. The SFCA module is embedded at each skip connection layer to extract crucial cross-frame information. Then, a multi-scale fusion (MSF) module that utilizes three parallel branches with different atrous convolutions is designed. The MSF module is placed at the bottleneck layer to dynamically fuse multi-scale context information from high-level features. Extensive experiments on TRUS image datasets show that the PZS-Net achieves higher accuracy in both the transitional zone (dice coefficient [Dice]: 68.90% ± 1.73%, mean intersection over union [mIoU]: 59.19% ± 2.09%, 95% Hausdorff distance [HD95]: 5.02 ± 0.83 mm) and the peripheral zone (Dice: 63.99% ± 3.16%, mIoU: 54.60% ± 3.35%, HD95: 5.28 ± 1.12 mm) and demonstrates the effectiveness and competitiveness of its key components via comprehensive ablation studies.