Please login first
Multiclass Classification of Brain Tumors with Various Deep Learning Models
* 1 , 2
1  Department of Electrical and Electronics Engineering, Faculty of Engineering and Architecture, Kafkas University, Kars TR 36100, Turkey
2  Department of Electrical and Electronics Engineering, Faculty of Engineering, Gazi University, Ankara TR 06570, Turkey
Academic Editor: Francisco Falcone

Abstract:

Brain cancer is one of the most dangerous cancer types in the world, and thousands of people are suffering from malignant brain tumors. Depending on the level of cancer, early diagnosis can be a lifesaver. However, thousands of scans must be studied in order to classify tumor types with high accuracy. Deep learning models can handle that amount of data and they can present results with high accuracy. It’s already known that deep learning models can give different results depending on dataset. In this paper, the effectiveness of some of the deep learning models on 2 different publicly available MRI (Magnetic Resonance Imaging) brain tumor datasets is examined. The reason for choosing this topic is that we are trying to find the best solution to classify tumors in the datasets. Different deep learning models are used separately on preprocessed datasets with the CLAHE preprocessing variable to extract features from images and classify them. Datasets are shuffled randomly for 80% training, 10% validation, and 10% testing. For finetuning, models are modified so that the output channel of the classifier is equal to the number of classes in the datasets. Results show that pre-trained and fine-tuned ResNet, RegNet, and Vision Transformer (ViT) deep learning models can achieve accuracies higher than 90% and they can be used as classifiers when a diagnosis is required.

Keywords: brain tumors; classification; deep learning; transfer learning
Top