Ultrasound prostate segmentation based on 3D V-Net with deep supervision
Xiaofeng Yang, Yang Lei, Sibo Tian, Tonghe Wang, Bo Wang, Ashesh B. Jani, Pretesh Patel, Hui Mao, Walter John Curran, Tian Liu, Xiuxiu He · 2019
We propose a method to automatically segment prostate from TRUS image based on multi-derivate deeply supervised network and multi-directional contour refinement. 3D multi-derivate V-Net is introduced to enable end-to-end segmentation. Deep supervision mechanism is integrated into the hidden layers to cope with the optimization difficulties when training such a network with limited training data. The probability map of new prostate contour is generated by the well-trained network and fused to reconstruct the prostate contour by multi-directional contour refinement. This proposed algorithm was evaluated using 30 patients’ data with TRUS image and manual contours. The mean Dice similarity coefficient (DSC) and mean surface distance (MSD) were 0.92 and 0.60 mm, which demonstrate the high accuracy of the proposed segmentation method. We have developed a novel deep learning-based method demonstrated that this method could significantly improve contour accuracy especially around the apex and base region. This segmentation technique could be a useful tool in ultrasound-guided interventions for prostate-cancer diagnosis and treatment.