Deep Learning-Based Approaches for Accurate Brain Tumor Detection in MRI Images

  • Mohammed Noori
  • , Mohsin Ahmed
  • , Amer Ibrahim
  • , Ali Saeed
  • , Mohammad Khishe
  • , Mahmoud Mahfuri

Research output: Chapter in Book/Report/Conference proceedingConference contribution

Abstract

Brain tumor classification is considered one of the major tasks in medical image analysis, where correct and timely diagnosis could be achieved to serve as the key to effective treatment. This research paper proposes a deep learning-based approach for automatic classification of brain tumors from MRI images by fine tuning the ResNet50V2 CNN model. The dataset is made up of four classes, namely: Glioma Tumor, Meningioma Tumor, No Tumor, and Pituitary Tumor, totaling 2,844 images. We have done some strategy data augmentation and class weighting in order to prevent class imbalance, which ensures good performance in all tumor classes. Pre-processing and augmentation of the input images are done via rotation, shifting, zooming, and flipping. It yielded 95.29% on the validation set, with the major highlights of performance metrics for the minority class being class precision equal to 0.97 and class recall equal to 0.92, hence proving the efficiency of performed class balancing. The range of F1-scores, from 0.93 to 0.97, means fantastic predictive capability across all tumor types. Other techniques were also used to make this model optimal, such as early stopping, learning rate reduction, and checkpointing. The high performance of this model shows great promise in helping clinicians toward the right diagnosis of brain tumors in an effective way. Future efforts will involve incorporating larger datasets, exploring advanced augmentation techniques to enhance further model generalizability. Model explainability tools such as Grad-CAM will be used to extract insight into the model's decision-making process. This enhances the clinical interpretability of the results.

Original languageEnglish
Title of host publication2024 International Conference on Decision Aid Sciences and Applications, DASA 2024
PublisherInstitute of Electrical and Electronics Engineers Inc.
ISBN (Electronic)9798350369106
DOIs
Publication statusPublished - 2024
Externally publishedYes
Event2024 International Conference on Decision Aid Sciences and Applications, DASA 2024 - Manama, Bahrain
Duration: Dec 11 2024Dec 12 2024

Publication series

Name2024 International Conference on Decision Aid Sciences and Applications, DASA 2024

Conference

Conference2024 International Conference on Decision Aid Sciences and Applications, DASA 2024
Country/TerritoryBahrain
CityManama
Period12/11/2412/12/24

Keywords

  • brain tumor classification
  • class imbalance
  • convolutional neural network
  • data augmentation
  • deep learning ResNet50V2
  • medical image analysis
  • model optimization
  • MRI

ASJC Scopus subject areas

  • Artificial Intelligence
  • Computer Networks and Communications
  • Computer Science Applications
  • Hardware and Architecture
  • Decision Sciences (miscellaneous)
  • Information Systems and Management
  • Control and Optimization

Fingerprint

Dive into the research topics of 'Deep Learning-Based Approaches for Accurate Brain Tumor Detection in MRI Images'. Together they form a unique fingerprint.

Cite this