Afzal Shiraz, Rauf Muhammad, Ashraf Shahzad, Bin Md Ayob Shahrin, Ahmad Arfeen Zeeshan
Department of Electronic Engineering, Dawood University of Engineering and Technology, Karachi 74800, Pakistan.
Department of Computer Science, DHA Suffa University, Karachi 75500, Pakistan.
Diagnostics (Basel). 2025 Feb 5;15(3):378. doi: 10.3390/diagnostics15030378.
Deep transfer learning, leveraging convolutional neural networks (CNNs), has become a pivotal tool for brain tumor detection. However, key challenges include optimizing hyperparameter selection and enhancing the generalization capabilities of models. This study introduces a novel CART-ANOVA (Cartesian-ANOVA) hyperparameter tuning framework, which differs from traditional optimization methods by systematically integrating statistical significance testing (ANOVA) with the Cartesian product of hyperparameter values. This approach ensures robust and precise parameter tuning by evaluating the interaction effects between hyperparameters, such as batch size and learning rate, rather than relying solely on grid or random search. Additionally, it implements seven distinct classification schemes for brain tumors, aimed at improving diagnostic accuracy and robustness. The proposed framework employs a ResNet18-based knowledge transfer learning (KTL) model trained on a primary dataset, with 20% allocated for testing. Hyperparameters were optimized using CART-ANOVA analysis, and statistical validation ensured robust parameter selection. The model's generalization and robustness were evaluated on an independent second dataset. Performance metrics, including precision, accuracy, sensitivity, and F1 score, were compared against other pre-trained CNN models. The framework achieved exceptional testing accuracy of 99.65% for four-class classification and 98.05% for seven-class classification on the source 1 dataset. It also maintained high generalization capabilities, achieving accuracies of 98.77% and 96.77% on the source 2 datasets for the same tasks. The incorporation of seven distinct classification schemes further enhanced variability and diagnostic capability, surpassing the performance of other pre-trained models. The CART-ANOVA hyperparameter tuning framework, combined with a ResNet18-based KTL approach, significantly improves brain tumor classification accuracy, robustness, and generalization. These advancements demonstrate strong potential for enhancing diagnostic precision and informing effective treatment strategies, contributing to advancements in medical imaging and AI-driven healthcare solutions.
利用卷积神经网络(CNN)的深度迁移学习已成为脑肿瘤检测的关键工具。然而,关键挑战包括优化超参数选择和增强模型的泛化能力。本研究引入了一种新颖的CART-ANOVA(笛卡尔积-方差分析)超参数调整框架,该框架与传统优化方法不同,它通过将统计显著性检验(方差分析)与超参数值的笛卡尔积系统地集成在一起。这种方法通过评估超参数之间的交互作用(如批量大小和学习率)来确保稳健而精确的参数调整,而不是仅仅依赖于网格搜索或随机搜索。此外,它还针对脑肿瘤实施了七种不同的分类方案,旨在提高诊断准确性和稳健性。所提出的框架采用了基于ResNet18的知识迁移学习(KTL)模型,该模型在一个主要数据集上进行训练,其中20%用于测试。使用CART-ANOVA分析对超参数进行了优化,并通过统计验证确保了稳健的参数选择。在一个独立的第二个数据集上评估了该模型的泛化能力和稳健性。将性能指标(包括精确率、准确率、灵敏度和F1分数)与其他预训练的CNN模型进行了比较。该框架在源1数据集上实现了四分类99.65%和七分类98.05%的卓越测试准确率。在相同任务的源2数据集上,它还保持了较高的泛化能力,准确率分别达到98.77%和96.77%。七种不同分类方案的纳入进一步增强了变异性和诊断能力,超过了其他预训练模型的性能。CART-ANOVA超参数调整框架与基于ResNet18的KTL方法相结合,显著提高了脑肿瘤分类的准确性、稳健性和泛化能力。这些进展显示出在提高诊断精度和为有效治疗策略提供信息方面的强大潜力,有助于医学成像和人工智能驱动的医疗保健解决方案的进步。