Classification of Brain Tumors on MRI Images Using Convolutional Neural Network Model EfficientNet

Abstract

A brain tumor is a lump caused by an imperfect cell turnover cycle in the brain and can affect all ages. Brain tumors have 4 grades, namely grades 1 to 2 are benign tumor grades, and grades 3 to 4 are malignant tumor grades. Therefore, early identification of brain tumor disease is very important in providing appropriate treatment and treatment. This study uses a dataset obtained through the Kaggle website titled Brain Tumor Classification (MRI). The number of data is 3264 images with details of Glioma tumors (926 images), Meningioma tumors (937 images), pituitary tumors (901 images), and without tumors (500 images). In this study, there are 4 scenarios with different testers. This study proposes the classification of brain tumors using Hyperparameter Tuning and EfficientNet models on MRI images. The EfficientNet model used is the EfficientNetB0 and EfficientNetB7 models with the architecture used are the input layer, GlobalAveragePooling2D layer, dropout layer, and dense layer as well as adding augmentation data to the dataset to manipulate the data in order to improve the results of the proposed model. After building the model, the results of accuracy, precision, recall, and f1-score will be obtained in each scenario. Accuracy results in Scenario 1 are 91%, scenario 2 is 95% accurate, scenario 3 is 95%, and scenario 4 is 98%.

Keywords

• brain tumor
• efficientnet
• convolutional neural network
• mri