Z-SSMNet:用于双参数MRI前列腺癌检测与诊断的区域感知自监督网格网络
Z-SSMNet: Zonal-aware Self-supervised Mesh Network for Prostate Cancer Detection and Diagnosis with Bi-parametric MRI
AI总结:
针对前列腺癌检测中各向异性MRI空间信息学习受限及标注数据依赖问题,提出Z-SSMNet,通过自适应多维卷积、自监督预训练和区域解剖约束,在PI-CAI数据集上取得了优异的病灶级与患者级诊断性能。
摘要:
双参数磁共振成像(bpMRI)已成为检测和诊断临床显著性前列腺癌(csPCa)的关键模态。开发基于人工智能的系统利用bpMRI识别csPCa,可通过提高效率和成本效益来变革PCa管理。然而,当前使用卷积神经网络(CNNs)的最先进方法在学习各向异性图像的平面内和三维空间信息方面存在局限。其性能也依赖于大型、多样化且标注良好的bpMRI数据集的可用性。我们提出了一种区域感知自监督网格网络(Z-SSMNet),该网络自适应地整合多维(2D/2.5D/3D)卷积,以平衡的方式学习各向异性bpMRI的密集切片内信息和稀疏切片间信息。提出了一种自监督学习(SSL)技术,利用大规模无标签数据预训练我们的网络,以学习bpMRI的外观、纹理和结构语义。其旨在在预训练阶段捕获切片内和切片间信息。此外,我们约束网络聚焦于区域解剖结构,以进一步提升csPCa的检测与诊断能力。我们在包含10000+多中心和多扫描仪数据的PI-CAI数据集上进行了广泛实验。我们的Z-SSMNet在病灶级检测(AP分数0.633)和患者级诊断(AUROC分数0.881)上均表现优异,在PI-CAI挑战赛的开放开发阶段稳居榜首,并在封闭测试阶段保持强劲表现,取得AP分数0.690和AUROC分数0.909,荣获亚军。
英文摘要:
Bi-parametric magnetic resonance imaging (bpMRI) has become a pivotal modality in the detection and diagnosis of clinically significant prostate cancer (csPCa). Developing AI-based systems to identify csPCa using bpMRI can transform PCa management by improving efficiency and cost-effectiveness. However, current state-of-the-art methods using convolutional neural networks (CNNs) are limited in learning in-plane and three-dimensional spatial information from anisotropic images. Their performances also depend on the availability of large, diverse, and well-annotated bpMRI datasets. We propose a Zonal-aware Self-supervised Mesh Network (Z-SSMNet) that adaptively integrates multi-dimensional (2D/2.5D/3D) convolutions to learn dense intra-slice information and sparse inter-slice information of the anisotropic bpMRI in a balanced manner. A self-supervised learning (SSL) technique is proposed to pre-train our network using large-scale unlabeled data to learn the appearance, texture, and structure semantics of bpMRI. It aims to capture both intra-slice and inter-slice information during the pre-training stage. Furthermore, we constrained our network to focus on the zonal anatomical regions to further improve the detection and diagnosis capability of csPCa. We conducted extensive experiments on the PI-CAI dataset comprising 10000+ multi-center and multi-scanner data. Our Z-SSMNet excelled in both lesion-level detection (AP score of 0.633) and patient-level diagnosis (AUROC score of 0.881), securing the top position in the Open Development Phase of the PI-CAI challenge and maintained strong performance, achieving an AP score of 0.690 and an AUROC score of 0.909, and securing the second-place ranking in the Closed Testing Phase.
机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。