Gasmi Karim, Ben Aoun Najib, Alsalem Khalaf, Ltaifa Ibtihel Ben, Alrashdi Ibrahim, Ammar Lassaad Ben, Mrabet Manel, Shehab Abdulaziz
Department of Computer Science, College of Computer and Information Sciences, Jouf University, Sakkaka, Saudi Arabia.
College of Computing and Information, Al-Baha University, Alaqiq, Saudi Arabia.
Front Neuroinform. 2024 Nov 26;18:1444650. doi: 10.3389/fninf.2024.1444650. eCollection 2024.
Brain tumor classification is a critical task in medical imaging, as accurate diagnosis directly influences treatment planning and patient outcomes. Traditional methods often fall short in achieving the required precision due to the complex and heterogeneous nature of brain tumors. In this study, we propose an innovative approach to brain tumor multi-classification by leveraging an ensemble learning method that combines advanced deep learning models with an optimal weighting strategy. Our methodology integrates Vision Transformers (ViT) and EfficientNet-V2 models, both renowned for their powerful feature extraction capabilities in medical imaging. This model enhances the feature extraction step by capturing both global and local features, thanks to the combination of different deep learning models with the ViT model. These models are then combined using a weighted ensemble approach, where each model's prediction is assigned a weight. To optimize these weights, we employ a genetic algorithm, which iteratively selects the best weight combinations to maximize classification accuracy. We trained and validated our ensemble model using a well-curated dataset comprising labeled brain MRI images. The model's performance was benchmarked against standalone ViT and EfficientNet-V2 models, as well as other traditional classifiers. The ensemble approach achieved a notable improvement in classification accuracy, precision, recall, and F1-score compared to individual models. Specifically, our model attained an accuracy rate of 95%, significantly outperforming existing methods. This study underscores the potential of combining advanced deep learning models with a genetic algorithm-optimized weighting strategy to tackle complex medical classification tasks. The enhanced diagnostic precision offered by our ensemble model can lead to better-informed clinical decisions, ultimately improving patient outcomes. Furthermore, our approach can be generalized to other medical imaging classification problems, paving the way for broader applications of AI in healthcare. This advancement in brain tumor classification contributes valuable insights to the field of medical AI, supporting the ongoing efforts to integrate advanced computational tools in clinical practice.
脑肿瘤分类是医学成像中的一项关键任务,因为准确的诊断直接影响治疗方案的制定和患者的治疗结果。由于脑肿瘤具有复杂和异质性的特点,传统方法往往难以达到所需的精度。在本研究中,我们提出了一种创新的脑肿瘤多分类方法,利用集成学习方法,将先进的深度学习模型与最优加权策略相结合。我们的方法集成了视觉Transformer(ViT)和EfficientNet-V2模型,这两种模型在医学成像中都以其强大的特征提取能力而闻名。由于不同的深度学习模型与ViT模型相结合,该模型通过捕捉全局和局部特征增强了特征提取步骤。然后使用加权集成方法将这些模型进行组合,为每个模型的预测分配一个权重。为了优化这些权重,我们采用了遗传算法,该算法迭代选择最佳权重组合以最大化分类准确率。我们使用一个精心策划的包含标记脑MRI图像的数据集对我们的集成模型进行训练和验证。该模型的性能与独立的ViT和EfficientNet-V2模型以及其他传统分类器进行了基准测试。与单个模型相比,集成方法在分类准确率、精度、召回率和F1分数方面取得了显著提高。具体而言,我们的模型达到了95%的准确率,明显优于现有方法。本研究强调了将先进的深度学习模型与遗传算法优化的加权策略相结合来解决复杂医学分类任务的潜力。我们的集成模型提供的更高诊断精度可以导致更明智的临床决策,最终改善患者的治疗结果。此外,我们的方法可以推广到其他医学成像分类问题,为人工智能在医疗保健领域的更广泛应用铺平道路。脑肿瘤分类的这一进展为医学人工智能领域提供了有价值的见解,支持了在临床实践中集成先进计算工具的持续努力。