A novel registration-based algorithm for prostate segmentation via the combination of SSM and CNN

Chunxia Qin (1, 2) , Puxun Tu (1) , Xiaojun Chen (1) , Jocelyne Troccaz (3)

1 School of Mechanical Engineering Shanghai Jiao Tong University
2 School of Biomedical Engineering, Shanghai Jiao Tong University
3 TIMC-GMCAO - Gestes Medico-chirurgicaux Assistés par Ordinateur

Precise determination of target is an essential procedure in prostate interventions, such as prostate biopsy, lesion detection, and targeted therapy. However, the prostate delineation may be tough in some cases due to tissue ambiguity or lack of partial anatomical boundary. In this study, we proposed a novel supervised registrationbased algorithm for precise prostate segmentation, which combine the convolutional neural network (CNN) with a statistical shape model (SSM). Methods: The proposed network mainly consists of two branches. One called SSM-Net branch was exploited to predict the shape transform matrix, shape control parameters, and shape fine-tuning vector, for the generation of the prostate boundary. Furtherly, according to the inferred boundary, a normalized distance map was calculated as the output of SSM-Net. Another branch named ResU-Net was employed to predict a probability label map from the input images at the same time. Integrating the output of these two branches, the optimal weighted sum of the distance map and the probability map was regarded as the prostate segmentation. Results: Two public datasets PROMISE12 and NCI-ISBI 2013 were utilized to evaluate the performance of the proposed algorithm. The results demonstrate that the segmentation algorithm achieved the best performance with an SSM of 9500 nodes, which obtained a dice of 0.907 and an average surface distance of 1.85 mm. Compared with other methods, our algorithm delineates the prostate region more accurately and efficiently. In addition, we verified the impact of model elasticity augmentation and the fine-tuning item on the network segmentation capability. As a result, both factors have improved the delineation accuracy, with dice increased by 10% and 7% respectively. Conclusions: Our segmentation method has the potential to be an effective and robust approach for prostate segmentation.

Mots clés: registration-based segmentation; statistical shape mode; probability map; boundary distance map

Domaines: Traitement des images [eess.IV]; Intelligence artificielle [cs.AI]; Ingénierie biomédicale

Chunxia Qin, Puxun Tu, Xiaojun Chen, Jocelyne Troccaz. A novel registration-based algorithm for prostate segmentation via the combination of SSM and CNN. Medical Physics, 2022, 49 (8), pp.5268-5282. ⟨10.1002/mp.15698⟩. ⟨hal-03654900⟩