Cinepro提升前列腺超声影像中癌症检测的鲁棒性,有效应对标注噪声。
Cinepro: Robust Training of Foundation Models for Cancer Detection in Prostate Ultrasound Cineloops
- 引入病理报告中癌组织比例作为损失函数,缓解标注噪声问题。
- 利用多帧时序信息进行鲁棒增强,学习稳定癌症特征。
- 在多中心数据集上达到77.1% AUROC和83.8%平衡准确率。
基于深度学习的前列腺癌(PCa)检测在实时穿刺引导中展现出潜力,但超声图像缺乏像素级癌症标注,导致标签噪声。现有方法多聚焦于有限兴趣区域,忽视解剖上下文。基础模型可分析全图以捕捉全局空间关系,但仍受超声数据粗略病理标注带来的弱标签影响。我们提出Cinepro框架,通过将活检核心中癌组织比例纳入损失函数,实现更精细的监督,增强模型对癌症的定位能力;同时利用多帧时序数据施加鲁棒增强,提升模型学习稳定癌症特征的能力。在多中心前列腺超声数据集上,Cinepro取得77.1% AUROC与83.8%平衡准确率,优于现有基准,验证了其在弱标签超声数据中推进基础模型的有效性。
原文摘要 · Abstract (English)
Prostate cancer (PCa) detection using deep learning (DL) models has shown potential for enhancing real-time guidance during biopsies. However, prostate ultrasound images lack pixel-level cancer annotations, introducing label noise. Current approaches often focus on limited regions of interest (ROIs), disregarding anatomical context necessary for accurate diagnosis. Foundation models can overcome this limitation by analyzing entire images to capture global spatial relationships; however, they still encounter challenges stemming from the weak labels associated with coarse pathology annotations in ultrasound data. We introduce Cinepro, a novel framework that strengthens foundation models' ability to localize PCa in ultrasound cineloops. Cinepro adapts robust training by integrating the proportion of cancer tissue reported by pathology in a biopsy core into its loss function to address label noise, providing a more nuanced supervision. Additionally, it leverages temporal data across multiple frames to apply robust augmentations, enhancing the model's ability to learn stable cancer-related features. Cinepro demonstrates superior performance on a multi-center prostate ultrasound dataset, achieving an AUROC of 77.1% and a balanced accuracy of 83.8%, surpassing current benchmarks. These findings underscore Cinepro's promise in advancing foundation models for weakly labeled ultrasound data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。