轻量双流模型精准定位脑瘤,兼具高效与可解释性。
MobileDenseAttn:A Dual-Stream Architecture for Accurate and Interpretable Brain Tumor Detection
- 双流结构融合MobileNetV2与DenseNet201,逐级增强特征表示
- 测试准确率达98.35%,F1值稳定在0.9835,较基线提升3.67%
- 通过GradCAM热力图清晰展示病灶位置,适合临床辅助诊断
MRI中脑瘤检测对及时诊疗至关重要,但人工分析耗时且易出错。现有方法普遍存在泛化能力差、计算效率低、不可解释等问题。为此,本文提出MobileDenseAttn,一种融合MobileNetV2与DenseNet201的双流架构,通过特征级融合逐步提升特征表示能力、计算效率与可视化解释性(基于GradCAM)。模型在包含6,020张MRI扫描的增强数据集上训练,涵盖胶质瘤、脑膜瘤、垂体瘤及正常样本。严格5折交叉验证下,训练准确率99.75%,测试准确率98.35%,F1分数稳定为0.9835(95%置信区间:0.9743–0.9920)。对比实验表明,其性能显著优于基线模型(VGG19、DenseNet201、MobileNetV2),准确率提升3.67%,训练时间减少39.3%。GradCAM热力图能清晰定位肿瘤区域,具备临床意义的可解释性。结果表明,MobileDenseAttn是高效、高精度、可解释的模型,具备临床实用潜力。
原文摘要 · Abstract (English)
The detection of brain tumor in MRI is an important aspect of ensuring timely diagnostics and treatment; however, manual analysis is commonly long and error-prone. Current approaches are not universal because they have limited generalization to heterogeneous tumors, are computationally inefficient, are not interpretable, and lack transparency, thus limiting trustworthiness. To overcome these issues, we introduce MobileDenseAttn, a fusion model of dual streams of MobileNetV2 and DenseNet201 that can help gradually improve the feature representation scale, computing efficiency, and visual explanations via GradCAM. Our model uses feature level fusion and is trained on an augmented dataset of 6,020 MRI scans representing glioma, meningioma, pituitary tumors, and normal samples. Measured under strict 5-fold cross-validation protocols, MobileDenseAttn provides a training accuracy of 99.75%, a testing accuracy of 98.35%, and a stable F1 score of 0.9835 (95% CI: 0.9743 to 0.9920). The extensive validation shows the stability of the model, and the comparative analysis proves that it is a great advancement over the baseline models (VGG19, DenseNet201, MobileNetV2) with a +3.67% accuracy increase and a 39.3% decrease in training time compared to VGG19. The GradCAM heatmaps clearly show tumor-affected areas, offering clinically significant localization and improving interpretability. These findings position MobileDenseAttn as an efficient, high performance, interpretable model with a high probability of becoming a clinically practical tool in identifying brain tumors in the real world.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。