Heliyon (Sep 2024)

An efficient brain tumor detection and classification using pre-trained convolutional neural network models

  • K. Nishanth Rao,
  • Osamah Ibrahim Khalaf,
  • V. Krishnasree,
  • Aruru Sai Kumar,
  • Deema Mohammed Alsekait,
  • S. Siva Priyanka,
  • Ahmed Saleh Alattas,
  • Diaa Salama AbdElminaam

Journal volume & issue
Vol. 10, no. 17
p. e36773

Abstract

Read online

In cases of brain tumors, some brain cells experience abnormal and rapid growth, leading to the development of tumors. Brain tumors represent a significant source of illness affecting the brain. Magnetic Resonance Imaging (MRI) stands as a well-established and coherent diagnostic method for brain cancer detection. However, the resulting MRI scans produce a vast number of images, which require thorough examination by radiologists. Manual assessment of these images consumes considerable time and may result in inaccuracies in cancer detection. Recently, deep learning has emerged as a reliable tool for decision-making tasks across various domains, including finance, medicine, cybersecurity, agriculture, and forensics. In the context of brain cancer diagnosis, Deep Learning and Machine Learning algorithms applied to MRI data enable rapid prognosis. However, achieving higher accuracy is crucial for providing appropriate treatment to patients and facilitating prompt decision-making by radiologists. To address this, we propose the use of Convolutional Neural Networks (CNN) for brain tumor detection. Our approach utilizes a dataset consisting of two classes: three representing different tumor types and one representing non-tumor samples. We present a model that leverages pre-trained CNNs to categorize brain cancer cases. Additionally, data augmentation techniques are employed to augment the dataset size. The effectiveness of our proposed CNN model is evaluated through various metrics, including validation loss, confusion matrix, and overall loss. The proposed approach employing ResNet50 and EfficientNet demonstrated higher levels of accuracy, precision, and recall in detecting brain tumors.

Keywords