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.
Previous Article in event
Previous Article in session
Next Article in event
Next Article in session
Multiclass Classification of Brain Tumors with Various Deep Learning Models
Published:
01 November 2022
by MDPI
in 9th International Electronic Conference on Sensors and Applications
session Sensor Data Analytics
Abstract:
Keywords: brain tumors; classification; deep learning; transfer learning