arXiv:2411.02466eess.IVcs.AI2024-11被引 3

用弱监督+尺寸约束,让前列腺癌检测模型更省力且泛化更强。

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

  • 仅需圈画草图标注,通过尺寸约束损失提升分割精度
  • 在公开和私有数据集上表现接近全监督模型,跨域测试仍保持稳定
  • 多轮训练集成显著提升对未知数据域的适应能力,适合临床部署

全监督深度学习模型在医学图像分割中表现优异,但其临床应用受限于人工标注耗时。现有模型多在同质数据集上训练,对不同设备或采集协议带来的域偏移敏感。本文针对多参数MRI中临床显著性前列腺癌(csPCa)检测,采用弱监督方法结合尺寸约束损失,仅需医生圈画圆形草图即可训练。在PI-CAI、Prostate158及一个私有数据库上评估,模型在分布内验证与未见测试图像上表现接近全监督基线。跨域测试中,全监督与弱监督模型均性能下降,凸显临床部署中域适应的重要性。最终,通过多次训练集成预测,显著提升模型泛化能力。

原文摘要 · Abstract (English)

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.

前列腺癌弱监督域泛化MRI

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。