arXiv:2412.11953cs.CV2024-12被引 6

用乳腺钼靶图像预测癌症分型,同时给出结果可信度。

Reliable Breast Cancer Molecular Subtype Prediction based on uncertainty-aware Bayesian Deep Learning by Mammography

  • 基于贝叶斯深度学习,自动评估预测不确定性
  • 三类分型准确率AUC达0.71~0.86,优于传统方法
  • 适合临床辅助诊断,尤其关注预测可信度的场景

乳腺癌是异质性显著的疾病,不同分子亚型具有不同的临床表现、治疗反应及生存结局。开发一种可靠、准确、便捷且低成本的基于医学影像的分子亚型预测方法,对乳腺癌的诊断与预后具有重要意义。近年来,深度学习在利用多种医学影像进行乳腺癌分类任务中表现良好。然而,传统深度学习无法提供预测不确定性,而该不确定性直接影响诊断可靠性。为此,本研究提出一种基于全幅乳腺钼靶图像的不确定性感知贝叶斯深度学习模型,并引入新型两阶段分级分类策略以提升多类别亚型分类性能。模型对HER2富集型、腔面型和三阴性亚型的单独AUC分别为0.71、0.75和0.86。该模型不仅在性能上可媲美现有研究,即使仅使用全幅钼靶图像,且因具备不确定性量化能力,整体更为可靠。

原文摘要 · Abstract (English)

Breast cancer is a heterogeneous disease with different molecular subtypes, clinical behavior, treatment responses as well as survival outcomes. The development of a reliable, accurate, available and inexpensive method to predict the molecular subtypes using medical images plays an important role in the diagnosis and prognosis of breast cancer. Recently, deep learning methods have shown good performance in the breast cancer classification tasks using various medical images. Despite all that success, classical deep learning cannot deliver the predictive uncertainty. The uncertainty represents the validity of the predictions. Therefore, the high predicted uncertainty might cause a negative effect in the accurate diagnosis of breast cancer molecular subtypes. To overcome this, uncertainty quantification methods are used to determine the predictive uncertainty. Accordingly, in this study, we proposed an uncertainty-aware Bayesian deep learning model using the full mammogram images. In addition, to increase the performance of the multi-class molecular subtype classification task, we proposed a novel hierarchical classification strategy, named the two-stage classification strategy. The separate AUC of the proposed model for each subtype was 0.71, 0.75 and 0.86 for HER2-enriched, luminal and triple-negative classes, respectively. The proposed model not only has a comparable performance to other studies in the field of breast cancer molecular subtypes prediction, even using full mammography images, but it is also more reliable, due to quantify the predictive uncertainty.

乳腺癌贝叶斯深度学习不确定性量化影像诊断

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