An Efficient Ensemble Approach for Brain Tumors Classification Using Magnetic Resonance Imaging

Zubair Saeed*, Tarraf Torfeh, Souha Aouadi, Xiuquan Ji, Othmane Bouhali

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

Abstract

Tumors in the brain can be life-threatening, making early and precise detection crucial for effective treatment and improved patient outcomes. Deep learning (DL) techniques have shown significant potential in automating the early diagnosis of brain tumors by analyzing magnetic resonance imaging (MRI), offering a more efficient and accurate approach to classification. Deep convolutional neural networks (DCNNs), which are a sub-field of DL, have the potential to analyze rapidly and accurately MRI data and, as such, assist human radiologists, facilitating quicker diagnoses and earlier treatment initiation. This study presents an ensemble of three high-performing DCNN models, i.e., DenseNet169, EfficientNetB0, and ResNet50, for accurate classification of brain tumors and non-tumor MRI samples. Our proposed ensemble model demonstrates significant improvements over various evaluation parameters compared to individual state-of-the-art (SOTA) DCNN models. We implemented ten SOTA DCNN models, i.e., EfficientNetB0, ResNet50, DenseNet169, DenseNet121, SqueezeNet, ResNet34, ResNet18, VGG16, VGG19, and LeNet5, and provided a detailed performance comparison. We evaluated these models using two learning rates (LRs) of 0.001 and 0.0001 and two batch sizes (BSs) of 64 and 128 and identified the optimal hyperparameters for each model. Our findings indicate that the ensemble approach outperforms individual models, having 92% accuracy, 90% precision, 92% recall, and an F1 score of 91% at a 64 BS and 0.0001 LR. This study not only highlights the superior performance of the ensemble technique but also offers a comprehensive comparison with the latest research.

Original languageEnglish
Article number641
Number of pages19
JournalInformation (Switzerland)
Volume15
Issue number10
DOIs
Publication statusPublished - Oct 2024

Keywords

  • Batch sizes
  • Deep convolutional neural networks
  • Deep learning
  • Ensemble model
  • Learning rates
  • Magnetic resonance imaging
  • State-of-the-art

Fingerprint

Dive into the research topics of 'An Efficient Ensemble Approach for Brain Tumors Classification Using Magnetic Resonance Imaging'. Together they form a unique fingerprint.

Cite this