arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2311.14462eess.IVcs.CV

CT-xCOV:一个基于CT扫描的可解释COVID-19诊断框架

CT-xCOV: a CT-scan based Explainable Framework for COVid-19 diagnosis

  • Hassan 2 University(哈桑二世大学)
  • Sidi Mohamed Ben Abdellah University(西迪·穆罕默德·本·阿卜杜拉大学)
  • Mohammed VI Polytechnic University(穆罕默德六世理工大学)

机构由 AI 辅助整理,请以论文原文为准。

Ismail Elbouknify, Afaf Bouhoute, Khalid Fardousse, Ismail Berrada, Abdelmajid Badri

更新

AI总结:

本研究提出CT-xCOV框架,通过U-Net分割和CNN分类实现COVID-19端到端检测,结合Grad-Cam等XAI技术提供可解释性,实验表明分类准确率达98.40%,且Grad-Cam解释效果最优。

AI中文摘要:

在本研究中,我们开发了CT-xCOV,这是一个使用深度学习(DL)在CT扫描上进行COVID-19诊断的可解释框架。CT-xCOV采用从肺部分割到COVID-19检测以及检测模型预测解释的端到端方法。对于肺部分割,我们使用了著名的U-Net模型。对于COVID-19检测,我们比较了三种不同的CNN架构:标准CNN、ResNet50和DenseNet121。检测之后,提供了视觉和文本解释。对于视觉解释,我们应用了三种不同的XAI技术,即Grad-Cam、Integrated Gradient(IG)和LIME。通过计算肺部感染百分比来添加文本解释。为了评估所用XAI技术的性能,我们提出了一种基于真实值的评估方法,测量可视化输出与真实感染之间的相似度。进行的实验表明,所应用的DL模型取得了良好的结果。U-Net分割模型实现了高Dice系数(98%)。我们提出的分类模型(标准CNN)的性能通过5折交叉验证进行了验证(准确率98.40%和f1分数98.23%)。最后,XAI技术的比较结果表明,与LIME和IG相比,Grad-Cam提供了最佳解释,在COVID-19阳性扫描上实现了55%的Dice系数,而IG和LIME分别为29%和24%。本文使用的代码和数据集可在GitHub仓库[1]中获取。

英文摘要:

In this work, CT-xCOV, an explainable framework for COVID-19 diagnosis using Deep Learning (DL) on CT-scans is developed. CT-xCOV adopts an end-to-end approach from lung segmentation to COVID-19 detection and explanations of the detection model's prediction. For lung segmentation, we used the well-known U-Net model. For COVID-19 detection, we compared three different CNN architectures: a standard CNN, ResNet50, and DenseNet121. After the detection, visual and textual explanations are provided. For visual explanations, we applied three different XAI techniques, namely, Grad-Cam, Integrated Gradient (IG), and LIME. Textual explanations are added by computing the percentage of infection by lungs. To assess the performance of the used XAI techniques, we propose a ground-truth-based evaluation method, measuring the similarity between the visualization outputs and the ground-truth infections. The performed experiments show that the applied DL models achieved good results. The U-Net segmentation model achieved a high Dice coefficient (98%). The performance of our proposed classification model (standard CNN) was validated using 5-fold cross-validation (acc of 98.40% and f1-score 98.23%). Lastly, the results of the comparison of XAI techniques show that Grad-Cam gives the best explanations compared to LIME and IG, by achieving a Dice coefficient of 55%, on COVID-19 positive scans, compared to 29% and 24% obtained by IG and LIME respectively. The code and the dataset used in this paper are available in the GitHub repository [1].

↑