论文标题
不确定性的深度学习模型可以对数字组织病理学高信任预测
Uncertainty-Informed Deep Learning Models Enable High-Confidence Predictions for Digital Histopathology
论文作者
论文摘要
模型表达自己的预测不确定性的能力是维持临床用户信心的重要属性,因为计算生物标志物被部署到现实世界中的医疗环境中。在癌症数字组织病理学的领域,我们描述了一种新型的,面向临床的临床导向方法,用于全面斜线图像的不确定性定量(UQ),使用辍学和计算训练数据上的阈值来估算不确定性,以确定低信心预测的临界值。我们训练模型以鉴定肺腺癌与鳞状细胞癌,并表明在两个跨越多个机构的大型外部数据集中,在交叉验证和测试中,高信心预测都优于没有UQ的预测预测。我们的测试策略密切接近现实世界的应用,并使用预定的阈值对无监督的,无调的幻灯片产生了预测。此外,我们表明,在域移位的环境中,UQ阈值仍然可靠,并具有准确的高度自信预测腺癌与鳞状细胞癌,用于过度分布,非肺癌同类。
A model's ability to express its own predictive uncertainty is an essential attribute for maintaining clinical user confidence as computational biomarkers are deployed into real-world medical settings. In the domain of cancer digital histopathology, we describe a novel, clinically-oriented approach to uncertainty quantification (UQ) for whole-slide images, estimating uncertainty using dropout and calculating thresholds on training data to establish cutoffs for low- and high-confidence predictions. We train models to identify lung adenocarcinoma vs. squamous cell carcinoma and show that high-confidence predictions outperform predictions without UQ, in both cross-validation and testing on two large external datasets spanning multiple institutions. Our testing strategy closely approximates real-world application, with predictions generated on unsupervised, unannotated slides using predetermined thresholds. Furthermore, we show that UQ thresholding remains reliable in the setting of domain shift, with accurate high-confidence predictions of adenocarcinoma vs. squamous cell carcinoma for out-of-distribution, non-lung cancer cohorts.