Brain Tumor Classification using a Convolutional Neural Network and Different Optimizers

Mulla M., DİREKOĞLU C.

2023 Innovations in Intelligent Systems and Applications Conference, ASYU 2023, Sivas, Turkey, 11 - 13 October 2023, (Full Text) identifier

  • Publication Type: Conference Paper / Full Text
  • Doi Number: 10.1109/asyu58738.2023.10296659
  • City: Sivas
  • Country: Turkey
  • Keywords: brain tumor classification, Convolutional Neural Network, Deep learning, Figshare brain tumor dataset, VGG-16
  • Middle East Technical University Affiliated: Yes

Abstract

In this paper, we designed a brain tumor classifier using a convolutional neural network (CNN) which is a subfield of deep learning. The proposed framework aims to classify three different brain tumor classes which are meningioma, glioma, and pituitary from the publicly available Figshare brain tumor dataset. In this regard, VGG-16 CNN architecture is trained for three brain tumor classes and the results are very promising. This architecture is experimented with six different optimizers namely Adaptive Moment Estimation (Adam), Stochastic Gradient Descent (SGD), Root Mean Square Propagation (RMSprop), Adaptive Gradient (Adagrad), Adadelta, and Adamax. Simulation results demonstrate that the proposed VGG-16 architecture achieved 97.15% on the test dataset with respect to 70% training, 15% validation, and 15% testing data.