Open Access iconOpen Access

ARTICLE

RBEBT: A ResNet-Based BA-ELM for Brain Tumor Classification

Ziquan Zhu1, Muhammad Attique Khan2, Shui-Hua Wang1, Yu-Dong Zhang1,*

1 University of Leicester, Leicester, LE1 7RH, UK
2 HITEC University Taxila, Taxila, 47080, Pakistan

* Corresponding Author: Yu-Dong Zhang. Email: email

Computers, Materials & Continua 2023, 74(1), 101-111. https://doi.org/10.32604/cmc.2023.030790

Abstract

Brain tumor refers to the formation of abnormal cells in the brain. It can be divided into benign and malignant. The main diagnostic methods for brain tumors are plain X-ray film, Magnetic resonance imaging (MRI), and so on. However, these artificial diagnosis methods are easily affected by external factors. Scholars have made such impressive progress in brain tumors classification by using convolutional neural network (CNN). However, there are still some problems: (i) There are many parameters in CNN, which require much calculation. (ii) The brain tumor data sets are relatively small, which may lead to the overfitting problem in CNN. In this paper, our team proposes a novel model (RBEBT) for the automatic classification of brain tumors. We use fine-tuned ResNet18 to extract the features of brain tumor images. The RBEBT is different from the traditional CNN models in that the randomized neural network (RNN) is selected as the classifier. Meanwhile, our team selects the bat algorithm (BA) to optimize the parameters of RNN. We use five-fold cross-validation to verify the superiority of the RBEBT. The accuracy (ACC), specificity (SPE), precision (PRE), sensitivity (SEN), and F1-score (F1) are 99.00%, 95.00%, 99.00%, 100.00%, and 100.00%. The classification performance of the RBEBT is greater than 95%, which can prove that the RBEBT is an effective model to classify brain tumors.

Keywords


Cite This Article

APA Style
Zhu, Z., Khan, M.A., Wang, S., Zhang, Y. (2023). RBEBT: A resnet-based BA-ELM for brain tumor classification. Computers, Materials & Continua, 74(1), 101-111. https://doi.org/10.32604/cmc.2023.030790
Vancouver Style
Zhu Z, Khan MA, Wang S, Zhang Y. RBEBT: A resnet-based BA-ELM for brain tumor classification. Comput Mater Contin. 2023;74(1):101-111 https://doi.org/10.32604/cmc.2023.030790
IEEE Style
Z. Zhu, M.A. Khan, S. Wang, and Y. Zhang "RBEBT: A ResNet-Based BA-ELM for Brain Tumor Classification," Comput. Mater. Contin., vol. 74, no. 1, pp. 101-111. 2023. https://doi.org/10.32604/cmc.2023.030790



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 2223

    View

  • 727

    Download

  • 0

    Like

Share Link