用可解释AI提升脑瘤和肺炎影像诊断准确率与可信度
Explainable Deep Learning in Medical Imaging: Brain Tumor and Pneumonia Detection
- 结合ResNet50和DenseNet121模型,采用Grad-CAM生成热力图增强可解释性
- DenseNet121在脑瘤和肺炎检测中分别达94.3%和89.1%准确率,优于ResNet50
- DenseNet121聚焦病灶核心区域,而ResNet50常分散至非病灶区,更可信
深度学习在医学影像诊断中潜力巨大,但多数模型缺乏可解释性,阻碍临床信任与应用。本文提出一种可解释深度学习框架,用于检测脑部MRI中的脑瘤和胸部X光中的肺炎,采用ResNet50与DenseNet121两种主流卷积神经网络,在包含7,023张脑部MRI和5,863张胸部X光的Kaggle公开数据集上训练。DenseNet121在脑瘤检测中表现更优,准确率达94.3%,高于ResNet50的92.5%;在肺炎检测中准确率为89.1%,优于ResNet50的84.4%。为提升可解释性,引入梯度加权类激活映射(Grad-CAM),生成叠加于图像上的热力图,揭示决策关键区域。结果显示,DenseNet121始终聚焦核心病灶区域,而ResNet50偶有注意力分散至边缘或非病灶区域。
原文摘要 · Abstract (English)
Deep Learning (DL) holds enormous potential for improving medical imaging diagnostics, yet the lack of interpretability in most models hampers clinical trust and adoption. This paper presents an explainable deep learning framework for detecting brain tumors in MRI scans and pneumonia in chest X-ray images using two leading Convolutional Neural Networks, ResNet50 and DenseNet121. These models were trained on publicly available Kaggle datasets comprising 7,023 brain MRI images and 5,863 chest X-ray images, achieving high classification performance. DenseNet121 consistently outperformed ResNet50 with 94.3 percent vs. 92.5 percent accuracy for brain tumors and 89.1 percent vs. 84.4 percent accuracy for pneumonia. For better explainability, Gradient-weighted Class Activation Mapping (Grad-CAM) was integrated to create heatmap visualizations superimposed on the test images, indicating the most influential image regions in the decision-making process. Interestingly, while both models produced accurate results, Grad-CAM showed that DenseNet121 consistently focused on core pathological regions, whereas ResNet50 sometimes scattered attention to peripheral or non-pathological areas. Combining deep learning and explainable AI offers a promising path toward reliable, interpretable, and clinically useful diagnostic tools.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。