Deep Feature Extraction with SVM Classifier for Brain Tumor Classification

Abstract

Recently, methods for classifying brain tumors have used Deep Learning (DL), in particular, Convolutional Neural Networks (CNNs), to obtain promising results. Unlike conventional Machine Learning (ML) algorithms, CNNs can automatically extract features from images; they do not, however, need the extraction regarding hand-crafted features. Many obstacles reduce the accuracy of brain tumor classification systems; to address such obstacles, we propose a DL-based feature extraction method for brain tumor classification. A total of two pretrained CNNs, VggNet-16 and ResNet-18, are applied as deep feature extractors in the scheme to accomplish deep feature extraction. Contrast Limited Adaptive Histogram Equalization (CLAHE) technique has been first used for enhancing the images. Second, each separately pretrained CNN, VggNet-16 and ResNet-18, is used for extracting the deep features from Magnetic Resonance Imaging (MRI). Lastly, a Multiclass Support Vector Machines (Multiclass -SVM) classifier is used to complete a classification task. Tests conducted on the aforementioned datasets: the effectiveness of the suggested approach is demonstrated by the CE-MRI Figshare, the REMBRANDT, and the Brain Tumor Classification, where the pre-trained CNN models and SVM classifier enhance performance. Specifically, the pretrained CNN with SVM approach produces accuracy ranging from 95.28% to 98.62% across all data-sets.