arXiv:2409.12883cs.CVcs.AI2024-09被引 5

让AI识别肾结石类型时能说清‘为什么这样判’,提升医生信任度。

Improving Prototypical Parts Abstraction for Case-Based Reasoning Explanations Designed for the Kidney Stone Type Recognition

  • 用原型局部特征捕捉结石视觉细节,训练时引入新损失函数优化
  • 在六类结石上达到90.37%准确率,解释性提升且精度优于已有最佳模型
  • 生成决策位置与内容的可读解释,适合临床医生理解与使用

在输尿管镜检查中实时识别肾结石类型是泌尿外科的重大进展,可缩短取石时间并降低感染风险,同时实现即时抗复发治疗。目前仅有少数经验丰富的泌尿科医生能通过内窥镜视频图像识别结石类型。为此,近年已有多种深度学习(DL)模型用于基于输尿管镜图像自动分类结石类型,但其黑箱特性限制了临床应用。本文提出一种基于案例推理的DL模型,利用原型部分(PPs)生成局部与全局描述符。PPs编码每类结石(共六类)的色相、饱和度、亮度及纹理等视觉特征,类似于生物学家使用的特征。通过新设计的损失函数优化生成过程。局部与全局描述符可清晰解释决策依据(“是什么”和“在哪里”),便于生物学家和泌尿科医生理解。模型在包含六类常见结石的数据库上测试,平均分类准确率达90.37%。相较文献中八种先进模型,本方法在解释性显著提升的同时,准确率反而高于最佳基线(88.2%)。这些可解释且性能优越的结果增强了临床对AI方案的信任。

原文摘要 · Abstract (English)

The in-vivo identification of the kidney stone types during an ureteroscopy would be a major medical advance in urology, as it could reduce the time of the tedious renal calculi extraction process, while diminishing infection risks. Furthermore, such an automated procedure would make possible to prescribe anti-recurrence treatments immediately. Nowadays, only few experienced urologists are able to recognize the kidney stone types in the images of the videos displayed on a screen during the endoscopy. Thus, several deep learning (DL) models have recently been proposed to automatically recognize the kidney stone types using ureteroscopic images. However, these DL models are of black box nature whicl limits their applicability in clinical settings. This contribution proposes a case-based reasoning DL model which uses prototypical parts (PPs) and generates local and global descriptors. The PPs encode for each class (i.e., kidney stone type) visual feature information (hue, saturation, intensity and textures) similar to that used by biologists. The PPs are optimally generated due a new loss function used during the model training. Moreover, the local and global descriptors of PPs allow to explain the decisions ("what" information, "where in the images") in an understandable way for biologists and urologists. The proposed DL model has been tested on a database including images of the six most widespread kidney stone types. The overall average classification accuracy was 90.37. When comparing this results with that of the eight other DL models of the kidney stone state-of-the-art, it can be seen that the valuable gain in explanability was not reached at the expense of accuracy which was even slightly increased with respect to that (88.2) of the best method of the literature. These promising and interpretable results also encourage urologists to put their trust in AI-based solutions.

肾结石识别可解释AI原型网络

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