整合多种技术提升超声影像前列腺癌检测的准确与可信度
TRUSWorthy: Toward Clinically Applicable Deep Learning for Confident Detection of Prostate Cancer in Micro-Ultrasound
- 融合自监督学习与注意力机制,缓解标注数据少和标签弱的问题
- 在多中心数据集上实现79.9%的AUROC和71.5%的平衡准确率
- 高置信度预测中准确率达91%,适合临床辅助诊断场景
尽管深度学习在经直肠超声(TRUS)图像中识别可疑病灶方面展现出巨大潜力,但其在临床应用中仍面临多重挑战:组织形态异质性高、良性样本占主导导致类别不平衡,且高质量标注数据稀缺。若任一问题未解决,可能导致不可接受的临床后果。本文提出TRUSWorthy,一个经过精心设计、调优与集成的可靠前列腺癌检测系统。该系统整合自监督学习、基于Transformer的多实例学习聚合、随机欠采样提升与模型集成,分别应对标注稀缺、弱标签、类别不平衡和模型过自信问题。我们在大规模多中心微超声数据集上训练并严格评估该方法,结果表明其在准确率与不确定性校准方面优于现有最先进方法,达到79.9%的AUROC和71.5%的平衡准确率;在置信度最高的前20%预测中,平衡准确率可达91%。TRUSWorthy的成功验证了集成深度学习方案在严苛临床环境中的可行性,是迈向可信计算机辅助前列腺癌诊断的重要一步。
原文摘要 · Abstract (English)
While deep learning methods have shown great promise in improving the effectiveness of prostate cancer (PCa) diagnosis by detecting suspicious lesions from trans-rectal ultrasound (TRUS), they must overcome multiple simultaneous challenges. There is high heterogeneity in tissue appearance, significant class imbalance in favor of benign examples, and scarcity in the number and quality of ground truth annotations available to train models. Failure to address even a single one of these problems can result in unacceptable clinical outcomes.We propose TRUSWorthy, a carefully designed, tuned, and integrated system for reliable PCa detection. Our pipeline integrates self-supervised learning, multiple-instance learning aggregation using transformers, random-undersampled boosting and ensembling: these address label scarcity, weak labels, class imbalance, and overconfidence, respectively. We train and rigorously evaluate our method using a large, multi-center dataset of micro-ultrasound data. Our method outperforms previous state-of-the-art deep learning methods in terms of accuracy and uncertainty calibration, with AUROC and balanced accuracy scores of 79.9% and 71.5%, respectively. On the top 20% of predictions with the highest confidence, we can achieve a balanced accuracy of up to 91%. The success of TRUSWorthy demonstrates the potential of integrated deep learning solutions to meet clinical needs in a highly challenging deployment setting, and is a significant step towards creating a trustworthy system for computer-assisted PCa diagnosis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。