TY - GEN
T1 - Deep Learning-Based Approaches for Accurate Brain Tumor Detection in MRI Images
AU - Noori, Mohammed
AU - Ahmed, Mohsin
AU - Ibrahim, Amer
AU - Saeed, Ali
AU - Khishe, Mohammad
AU - Mahfuri, Mahmoud
N1 - Publisher Copyright:
© 2024 IEEE.
PY - 2024
Y1 - 2024
N2 - 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.
AB - 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.
KW - brain tumor classification
KW - class imbalance
KW - convolutional neural network
KW - data augmentation
KW - deep learning ResNet50V2
KW - medical image analysis
KW - model optimization
KW - MRI
UR - https://www.scopus.com/pages/publications/85217205277
UR - https://www.scopus.com/pages/publications/85217205277#tab=citedBy
U2 - 10.1109/DASA63652.2024.10836556
DO - 10.1109/DASA63652.2024.10836556
M3 - Conference contribution
AN - SCOPUS:85217205277
T3 - 2024 International Conference on Decision Aid Sciences and Applications, DASA 2024
BT - 2024 International Conference on Decision Aid Sciences and Applications, DASA 2024
PB - Institute of Electrical and Electronics Engineers Inc.
T2 - 2024 International Conference on Decision Aid Sciences and Applications, DASA 2024
Y2 - 11 December 2024 through 12 December 2024
ER -