arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2401.08058cs.CV

迈向临床可信赖的深度学习:将保形预测应用于颅内出血检测

Toward Clinically Trustworthy Deep Learning: Applying Conformal Prediction to Intracranial Hemorrhage Detection

Mayo Clinic(梅奥诊所)

Cooper Gamble, Shahriar Faghani, Bradley J. Erickson

AI总结:

本研究将Mondrian保形预测应用于颅内出血检测,构建不确定性感知深度学习模型,在CQ500数据集上以99.7%准确率识别挑战性案例且不误标确定案例,验证了保形预测提升放射学深度学习可信赖性的有效性。

摘要:

随着深度学习(DL)在放射学任务中持续展现其能力,优化临床DL解决方案以纳入安全性至关重要。临床采用DL工具的主要关切之一是信任。本研究旨在应用保形预测,作为迈向放射学DL可信赖性的一步。这是一项对来自CQ500数据集的491例非增强头部CT的回顾性研究,其中三位资深放射科医生对包含颅内出血(ICH)的切片进行了标注。数据集被划分为确定子集和挑战性子集,其中挑战性图像被定义为读者之间存在分歧的图像。一个DL模型在来自确定数据(训练数据集)的146名患者(10,815个切片)上训练,以执行五类ICH的定位和分类。为了开发一个不确定性感知的DL模型,使用确定数据中的1,546个案例(校准数据集)进行Mondrian保形预测(MCP)。该不确定性感知DL模型在8,401个确定和挑战性案例上进行了测试,以评估其识别挑战性案例的能力。经过MCP程序后,该模型在测试数据集上的ICH分类F1分数达到0.920。此外,它正确地将总共6,856个挑战性案例中的6,837个识别为挑战性案例(准确率99.7%)。它没有将任何确定案例错误标记为挑战性案例。该不确定性感知ICH检测器的性能与最先进模型相当。MCP在检测挑战性案例方面的表现表明,它在自动化ICH检测中是有用的,并且对放射学DL的可信赖性具有前景。

英文摘要:

As deep learning (DL) continues to demonstrate its ability in radiological tasks, it is critical that we optimize clinical DL solutions to include safety. One of the principal concerns in the clinical adoption of DL tools is trust. This study aims to apply conformal prediction as a step toward trustworthiness for DL in radiology. This is a retrospective study of 491 non-contrast head CTs from the CQ500 dataset, in which three senior radiologists annotated slices containing intracranial hemorrhage (ICH). The dataset was split into definite and challenging subsets, where challenging images were defined to those in which there was disagreement among readers. A DL model was trained on 146 patients (10,815 slices) from the definite data (training dataset) to perform ICH localization and classification for five classes of ICH. To develop an uncertainty-aware DL model, 1,546 cases of the definite data (calibration dataset) was used for Mondrian conformal prediction (MCP). The uncertainty-aware DL model was tested on 8,401 definite and challenging cases to assess its ability to identify challenging cases. After the MCP procedure, the model achieved an F1 score of 0.920 for ICH classification on the test dataset. Additionally, it correctly identified 6,837 of the 6,856 total challenging cases as challenging (99.7% accuracy). It did not incorrectly label any definite cases as challenging. The uncertainty-aware ICH detector performs on par with state-of-the-art models. MCP's performance in detecting challenging cases demonstrates that it is useful in automated ICH detection and promising for trustworthiness in radiological DL.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑