MRI brain tumor classification based on CNN features and machine learning classifiers

Yefan Liu,Zhendong Wang,Yunpeng Xue,Nuo Cheng,Bingjun Shen,Lijie Hou,Lihong Jin
DOI: https://doi.org/10.1007/s12652-024-04928-2
IF: 3.662
2024-12-15
Journal of Ambient Intelligence and Humanized Computing
Abstract:Early brain tumor type identification is crucial for diagnosis and treatment, therefore cerebral tumour categorization is a major area of medical research. One of the most important instruments for researching brain cancers is NMR technology, but it has misleading challenges, jumbled and unevenly distributed images, and is more challenging for specialists to identify. Deep learning and machine learning algorithms are capable of effectively addressing the problem of image classification, and they hold considerable promise to detect and classify brain tumour-related images. In this work, convolutional neural networks are what we utilize to extract deep features from magnetic resonance images of the brain and categorize the extracted features through multiple machine learning classifiers to find the best classifier and feature combination. The Figshare open dataset, which includes MRI image of three different types of brain tumors, was used to evaluate the model. By an analysis of the CNN and SVM models'differences. For CNN's FC3 fully connected layer feature classification, the RBF-based support vector machine has the highest classification accuracy, the average accuracy of five-fold cross-validation is 99.94%. Furthermore, the proposed model was validated using the remaining two datasets, resulting in an average accuracy of 99.83% and 99.61% for the two-classification (tumour or not) and four-classification (no tumours, glioma, meningioma, pituitary tumour) of brain tumour images, respectively. The model's classification and segmentation capabilities are of paramount importance for enhancing the precision of brain tumour detection, which can significantly facilitate clinical diagnosis and treatment planning.
computer science, information systems,telecommunications, artificial intelligence
What problem does this paper attempt to address?