arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2411.02466eess.IVcs.AIcs.LG
收藏

带尺寸约束的弱监督深度学习模型用于多参数MRI前列腺癌检测及对未知领域的泛化

Weakly supervised deep learning model with size constraint for prostate cancer detection in multiparametric MRI and generalization to unseen domains

Univ. Lyon(里昂大学);CNRS(法国国家科学研究中心);Inserm(法国国家健康与医学研究院);INSA Lyon(里昂国立应用科学学院);UCBL(里昂第一大学);CREATIS(医学成像与治疗应用研究中心);Hospices Civils de Lyon(里昂民用医院集团);Edouard Herriot Hospital(爱德华·埃里奥医院)

Robin Trombetta, Olivier Rouvière, Carole Lartizien

AI总结:

针对多参数MRI前列腺癌检测,提出带尺寸约束的弱监督模型,仅用弱涂鸦标注即可达到全监督性能,并验证集成可提升跨域泛化。

摘要:

全监督深度模型在许多医学分割任务中已展现出良好的性能。然而,这些工具在临床中的部署受到耗时的手工专家标注数据收集的限制。此外,大多数最先进的模型是在中等同质的数据集上训练和验证的。众所周知,深度学习方法常常因领域或标签偏移而性能大幅下降,且尚未构建成对未见数据或标签分布具有鲁棒性的模型。在临床环境中,这一问题尤为相关,因为部署机构可能使用与训练模型所用数据采集不同的扫描仪或采集协议。在本工作中,我们提出解决这两个挑战,即从双参数MRI中检测临床显著性前列腺癌(csPCa)。我们评估了(Kervadec等人,2018)提出的方法,该方法引入了一种尺寸约束损失,以从弱圆形涂鸦标注中生成精细的语义癌症病变分割。模型性能基于两个公共数据库(PI-CAI和Prostate158)和一个私有数据库进行评估。首先,我们表明该模型在分布内验证数据和未见测试图像上均达到与强全监督基线模型相当的性能。其次,我们观察到当在未见数据领域上测试时,全监督和弱监督模型的性能均有所下降。这证实了如果深度学习模型旨在部署于临床环境,高效领域适应方法的迫切需要。最后,我们表明来自多次训练的集成预测提高了泛化性能。

英文摘要:

Fully supervised deep models have shown promising performance for many medical segmentation tasks. Still, the deployment of these tools in clinics is limited by the very timeconsuming collection of manually expert-annotated data. Moreover, most of the state-ofthe-art models have been trained and validated on moderately homogeneous datasets. It is known that deep learning methods are often greatly degraded by domain or label shifts and are yet to be built in such a way as to be robust to unseen data or label distributions. In the clinical setting, this problematic is particularly relevant as the deployment institutions may have different scanners or acquisition protocols than those from which the data has been collected to train the model. In this work, we propose to address these two challenges on the detection of clinically significant prostate cancer (csPCa) from bi-parametric MRI. We evaluate the method proposed by (Kervadec et al., 2018), which introduces a size constaint loss to produce fine semantic cancer lesions segmentations from weak circle scribbles annotations. Performance of the model is based on two public (PI-CAI and Prostate158) and one private databases. First, we show that the model achieves on-par performance with strong fully supervised baseline models, both on in-distribution validation data and unseen test images. Second, we observe a performance decrease for both fully supervised and weakly supervised models when tested on unseen data domains. This confirms the crucial need for efficient domain adaptation methods if deep learning models are aimed to be deployed in a clinical environment. Finally, we show that ensemble predictions from multiple trainings increase generalization performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑