School of Computer Science and Engineering, Vellore Institute of Technology, Vellore, 632014, Tamil Nadu, India.
Department of Computer Science and Engineering, Vel Tech Rangarajan Dr. Sagunthala R&D Institute of Science and Technology, Avadi, Chennai, 600062, India.
Sci Rep. 2023 Dec 27;13(1):23029. doi: 10.1038/s41598-023-50505-6.
Accurately classifying brain tumor types is critical for timely diagnosis and potentially saving lives. Magnetic Resonance Imaging (MRI) is a widely used non-invasive method for obtaining high-contrast grayscale brain images, primarily for tumor diagnosis. The application of Convolutional Neural Networks (CNNs) in deep learning has revolutionized diagnostic systems, leading to significant advancements in medical imaging interpretation. In this study, we employ a transfer learning-based fine-tuning approach using EfficientNets to classify brain tumors into three categories: glioma, meningioma, and pituitary tumors. We utilize the publicly accessible CE-MRI Figshare dataset to fine-tune five pre-trained models from the EfficientNets family, ranging from EfficientNetB0 to EfficientNetB4. Our approach involves a two-step process to refine the pre-trained EfficientNet model. First, we initialize the model with weights from the ImageNet dataset. Then, we add additional layers, including top layers and a fully connected layer, to enable tumor classification. We conduct various tests to assess the robustness of our fine-tuned EfficientNets in comparison to other pre-trained models. Additionally, we analyze the impact of data augmentation on the model's test accuracy. To gain insights into the model's decision-making, we employ Grad-CAM visualization to examine the attention maps generated by the most optimal model, effectively highlighting tumor locations within brain images. Our results reveal that using EfficientNetB2 as the underlying framework yields significant performance improvements. Specifically, the overall test accuracy, precision, recall, and F1-score were found to be 99.06%, 98.73%, 99.13%, and 98.79%, respectively.
准确地对脑肿瘤类型进行分类对于及时诊断和挽救生命至关重要。磁共振成像(MRI)是一种广泛使用的非侵入性方法,用于获取高对比度的灰度脑图像,主要用于肿瘤诊断。卷积神经网络(CNNs)在深度学习中的应用彻底改变了诊断系统,使医学影像解释取得了重大进展。在本研究中,我们采用基于迁移学习的微调方法,使用 EfficientNets 将脑肿瘤分为三类:胶质瘤、脑膜瘤和垂体瘤。我们利用可公开访问的 CE-MRI Figshare 数据集对来自 EfficientNets 系列的五个预训练模型进行微调,范围从 EfficientNetB0 到 EfficientNetB4。我们的方法涉及两步过程来优化预训练的 EfficientNet 模型。首先,我们使用 ImageNet 数据集的权重初始化模型。然后,我们添加额外的层,包括顶部层和全连接层,以实现肿瘤分类。我们进行了各种测试,以评估我们微调的 EfficientNets 与其他预训练模型相比的稳健性。此外,我们分析了数据增强对模型测试准确性的影响。为了深入了解模型的决策过程,我们采用 Grad-CAM 可视化技术来检查最优化模型生成的注意力图,有效地突出脑图像中肿瘤的位置。我们的结果表明,使用 EfficientNetB2 作为基础框架可以显著提高性能。具体而言,总体测试准确率、精确率、召回率和 F1 得分为 99.06%、98.73%、99.13%和 98.79%。