» Articles » PMID: 39659489

Enhanced Brain Tumor Diagnosis Using Combined Deep Learning Models and Weight Selection Technique

Overview
Specialty Neurology
Date 2024 Dec 11
PMID 39659489
Authors
Affiliations
Soon will be listed here.
Abstract

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.

Citing Articles

Optimized Hybrid Deep Learning Framework for Early Detection of Alzheimer's Disease Using Adaptive Weight Selection.

Gasmi K, Alyami A, Hamid O, Altaieb M, Shahin O, Ben Ammar L Diagnostics (Basel). 2025; 14(24.

PMID: 39767140 PMC: 11674777. DOI: 10.3390/diagnostics14242779.

References
1.
Aziz M, Zade A, Farnia P, Alimohamadi M, Makkiabadi B, Ahmadian A . Accurate Automatic Glioma Segmentation in Brain MRI images Based on CapsNet. Annu Int Conf IEEE Eng Med Biol Soc. 2021; 2021:3882-3885. DOI: 10.1109/EMBC46164.2021.9630324. View

2.
Kang J, Ullah Z, Gwak J . MRI-Based Brain Tumor Classification Using Ensemble of Deep Features and Machine Learning Classifiers. Sensors (Basel). 2021; 21(6). PMC: 8004778. DOI: 10.3390/s21062222. View

3.
Zahid U, Ashraf I, Khan M, Alhaisoni M, Yahya K, Hussein H . BrainNet: Optimal Deep Learning Feature Fusion for Brain Tumor Classification. Comput Intell Neurosci. 2022; 2022:1465173. PMC: 9371837. DOI: 10.1155/2022/1465173. View

4.
Zacharaki E, Wang S, Chawla S, Yoo D, Wolf R, Melhem E . Classification of brain tumor type and grade using MRI texture and shape in a machine learning scheme. Magn Reson Med. 2009; 62(6):1609-18. PMC: 2863141. DOI: 10.1002/mrm.22147. View

5.
Pereira S, Pinto A, Alves V, Silva C . Brain Tumor Segmentation Using Convolutional Neural Networks in MRI Images. IEEE Trans Med Imaging. 2016; 35(5):1240-1251. DOI: 10.1109/TMI.2016.2538465. View