用可解释性指导CNN精简,提升脑肿瘤分类的可信度与准确性。
From Explanations to Architecture: Explainability-Driven CNN Refinement for Brain Tumor Classification in MRI
- 基于Grad-CAM移除低贡献层,减少模型深度和参数。
- 在多类MRI数据集上达98.21%准确率,跨数据集泛化能力强。
- 适合临床医疗场景,提升AI诊断可信度与可解释性。
近期脑肿瘤分类方法虽报告高准确率,但依赖过参数化的深层架构,可解释性差,难以判断预测是否基于肿瘤相关证据而非背景伪影或正常组织等干扰因素。本文提出一种可解释卷积神经网络(CNN)框架,在不牺牲分类准确率的前提下增强模型透明性,支持更可信的医疗AI应用,并助力实现可持续发展目标3:良好健康与福祉。不同于仅用于事后可视化的可解释AI,本方法利用Grad-CAM量化各层重要性,引导移除低贡献层,降低模型冗余深度与参数量,同时促进对判别性肿瘤区域的关注。进一步通过结合Grad-CAM、SHAP与LIME三种方法验证决策合理性。在多类脑MRI数据集上的实验表明,该模型在主数据集上达到98.21%准确率,在未见数据集上达95.74%,表现出强跨数据集泛化能力。整体方法在简洁性、透明性与准确性之间取得平衡,支持更可信且临床可用的脑肿瘤分类,助力改善健康结果与无创疾病检测。
原文摘要 · Abstract (English)
Recent brain tumor classification methods often report high accuracy but rely on deep, over-parameterized architectures with limited interpretability, making it difficult to determine whether predictions are driven by tumor-relevant evidence or by spurious cues such as background artifacts or normal tissue. We propose an explainable convolutional neural network (CNN) framework that enhances model transparency without sacrificing classification accuracy. This approach supports more trustworthy AI in healthcare and contributes to SDG 3: Good Health and Well-being by enabling more dependable MRI-based brain tumor diagnosis and earlier detection. Rather than using explainable AI solely for post hoc visualization, we employ Grad-CAM to quantify layer-wise relevance and guide the removal of low-contribution layers, reducing unnecessary depth and parameters while encouraging attention to discriminative tumor regions. We further validate the model's decision rationale using complementary explainability methods, combining Grad-CAM for spatial localization with SHAP and LIME for attribution-based verification. Experiments on multi-class brain MRI datasets show that the proposed model achieves 98.21% accuracy on the primary dataset and 95.74% accuracy on an unseen dataset, indicating strong cross-dataset generalization. Overall, the proposed approach balances simplicity, transparency, and accuracy, supporting more trustworthy and clinically applicable brain tumor classification for improved health outcomes and non-invasive disease detection.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。