The semantic segmentation of pelvic organs via MRI has important clinical significance. Recently, deep learning-enabled semantic segmentation has facilitated the three-dimensional geometric reconstruction of pelvic floor organs, providing clinicians with accurate and intuitive diagnostic results. However, the task of labeling pelvic floor MRI segmentation, typically performed by clinicians, is labor-intensive and costly, leading to a scarcity of labels. Insufficient segmentation labels limit the precise segmentation and reconstruction of pelvic floor organs. To address these issues, we propose a semi-supervised framework for pelvic organ segmentation. The implementation of this framework comprises two stages. In the first stage, it performs self-supervised pre-training using image restoration tasks. Subsequently, fine-tuning of the self-supervised model is performed, using labeled data to train the segmentation model. In the second stage, the self-supervised segmentation model is used to generate pseudo labels for unlabeled data. Ultimately, both labeled and unlabeled data are utilized in semi-supervised training. Upon evaluation, our method significantly enhances the performance in the semantic segmentation and geometric reconstruction of pelvic organs, Dice coefficient can increase by 2.65% averagely. Especially for organs that are difficult to segment, such as the uterus, the accuracy of semantic segmentation can be improved by up to 3.70%.
翻译:盆底器官的MRI语义分割具有重要的临床意义。近年来,基于深度学习的语义分割技术促进了盆底器官的三维几何重建,为临床医生提供了准确且直观的诊断结果。然而,由临床医生执行的盆底MRI分割标注任务耗时且成本高昂,导致标注数据稀缺。分割标注不足限制了盆底器官的精确分割与重建。为解决这些问题,我们提出了一种半监督盆底器官分割框架。该框架的实现包含两个阶段:第一阶段通过图像修复任务进行自监督预训练,随后利用标注数据对自监督模型进行微调以训练分割模型;第二阶段使用自监督分割模型为未标注数据生成伪标签,最终结合标注与未标注数据进行半监督训练。评估结果显示,我们的方法显著提升了盆底器官语义分割与几何重建的性能,Dice系数平均提升2.65%。尤其对于难以分割的器官(如子宫),语义分割准确率最高可提升3.70%。