优化EfficientNet架构的可解释性脑肿瘤分类模型研究
DOI:
CSTR:
作者:
作者单位:

兰州交通大学数理学院 兰州 730070

作者简介:

通讯作者:

中图分类号:

TP391;TN911

基金项目:

国家自然科学基金(12462002)项目资助


Research on optimizing interpretability of EfficientNet architecture brain tumor classification model
Author:
Affiliation:

School of Mathematics and Physics, Lanzhou Jiaotong University,Lanzhou 730070, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    脑肿瘤是一类具有高度侵袭性的神经系统疾病,其早期准确诊断对于制定个性化治疗方案至关重要。基于深度学习的计算机辅助诊断(CAD)技术在医学成像分析中取得了显著进展,但在分类准确性、计算效率及可解释性方面仍存在不足。为此,研究提出了一种基于迁移学习和微调策略优化的EfficientNet模型,通过改进部分卷积层和全连接层,并在网络顶部添加全局平均池化层与Dropout层,以增强模型的特征提取能力与分类性能。同时,引入Grad-CAM技术实现模型决策过程可视化,有效突出脑肿瘤的关键判别特征区域,从而增强模型的可解释性与临床应用可信度。在Figshare数据集上的实验结果表明,所提模型在显著降低参数量与计算复杂度的同时,测试集上的准确率达到99.35%,主要性能指标均优于VGG16、ResNet152V2及Vision Transformer等模型。此外,跨数据集验证实验中模型的准确率达到92.51%,进一步验证了其良好的稳定性与泛化能力。

    Abstract:

    Brain tumors are highly invasive neurological diseases, and accurate early diagnosis is crucial for developing personalized treatment plans. Computer-aided diagnosis (CAD) based on deep learning techniques has achieved significant progress in medical image analysis, but limitations remain in terms of classification accuracy, computational efficiency, and interpretability. To address these issues, this study proposes an optimized EfficientNet model based on transfer learning and fine-tuning strategies. The model improves certain convolutional and fully connected layers and adds a global average pooling layer and a Dropout layer at the top of the network to enhance feature extraction capability and classification performance. Additionally, gradient-weighted class activation mapping (Grad-CAM) is introduced to visualize the model′s decision-making process, effectively highlighting key discriminative regions of brain tumors, thereby improving interpretability and clinical reliability. Experimental results on the Figshare dataset demonstrate that the proposed model achieves an accuracy of 99.35% on the test set while significantly reducing parameter count and computational complexity, outperforming baseline models including VGG16, ResNet152V2, and Vision Transformer across all major metrics. Furthermore, cross-dataset validation shows that the model attains an accuracy of 92.51%, further demonstrating its robust stability and generalization capability.

    参考文献
    相似文献
    引证文献
引用本文

郭心茹,吕卫东,王蕊,赵迪妮.优化EfficientNet架构的可解释性脑肿瘤分类模型研究[J].电子测量技术,2026,49(6):220-228

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-05-13
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告