A Novel Approach for Classifying Brain Tumours Combining a SqueezeNet Model with SVM and Fine-Tuning

被引:21
作者
Rasool, Mohammed [1 ]
Ismail, Nor Azman [1 ]
Al-Dhaqm, Arafat [1 ]
Yafooz, Wael M. S. [2 ]
Alsaeedi, Abdullah [2 ]
机构
[1] Univ Teknol Malaysia, Fac Comp, Johor Baharu 81310, Malaysia
[2] Taibah Univ, Coll Comp Sci & Engn, Dept Comp Sci, Medina 42353, Saudi Arabia
关键词
brain tumour; MRI images; deep learning; CNN; SqueezeNet; SVM; fine-tuning; NEURAL-NETWORK; SEGMENTATION; IMAGES;
D O I
10.3390/electronics12010149
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Cancer of the brain is most common in the elderly and young and can be fatal in both. Brain tumours can heal better if they are diagnosed and treated quickly. When it comes to processing medical images, the deep learning method is essential in aiding humans in diagnosing various diseases. Classifying brain tumours is an essential step that relies heavily on the doctor's experience and training. A smart system for detecting and classifying these tumours is essential to aid in the non-invasive diagnosis of brain tumours using MRI (magnetic resonance imaging) images. This work presents a novel hybrid deep learning CNN-based structure to distinguish between three distinct types of human brain tumours through MRI scans. This paper proposes a method that employs a dual approach to classification using deep learning and CNN. The first approach combines the unsupervised classification of an SVM for pattern classification with a pre-trained CNN (i.e., SqueezeNet) for feature extraction. The second approach combines the supervised soft-max classifier with a finely tuned SqueezeNet. To evaluate the efficacy of the suggested method, MRI scans of the brain were used to analyse a total of 1937 images of glioma tumours, 926 images of meningioma tumours, 926 images of pituitary tumours, and 396 images of a normal brain. According to the experiment results, the finely tuned SqueezeNet model obtained an accuracy of 96.5%. However, when SqueezeNet was used as a feature extractor and an SVM classifier was applied, recognition accuracy increased to 98.7%.
引用
收藏
页数:18
相关论文
共 42 条
[1]  
Abhinav G., SQUEEZENET
[2]  
Alqudah A.M., 2020, arXiv, DOI [10.30534/ijatcse/2019/155862019, DOI 10.30534/IJATCSE/2019/155862019]
[3]   Developing of robust and high accurate ECG beat classification by combining Gaussian mixtures and wavelets features [J].
Alqudah, Ali Mohammad ;
Albadarneh, Alaa ;
Abu-Qasmieh, Isam ;
Alquran, Hiam .
AUSTRALASIAN PHYSICAL & ENGINEERING SCIENCES IN MEDICINE, 2019, 42 (01) :149-157
[4]   Classification of Brain Tumors from MRI Images Using a Convolutional Neural Network [J].
Badza, Milica M. ;
Barjaktarovic, Marko C. .
APPLIED SCIENCES-BASEL, 2020, 10 (06)
[5]   Squeeze U-Net: A Memory and Energy Efficient Image Segmentation Network [J].
Beheshti, Nazanin ;
Johnsson, Lennart .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, :1495-1504
[6]  
Bhavsar H., 2012, Int. J. Adv. Res. Comput. Eng. Technol, V1, P185
[7]  
Chappa RTNVS, 2020, 2020 10TH ANNUAL COMPUTING AND COMMUNICATION WORKSHOP AND CONFERENCE (CCWC), P691, DOI [10.1109/ccwc47524.2020.9031119, 10.1109/CCWC47524.2020.9031119]
[8]  
Cheng Jun., brain tumor dataset
[9]   A Fine-tuned deep convolutional neural network for chest radiography image classification on COVID-19 cases [J].
Dash, Amiya Kumar ;
Mohapatra, Puspanjali .
MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 81 (01) :1055-1075
[10]   A Deep Learning Approach for Brain Tumor Classification and Segmentation Using a Multiscale Convolutional Neural Network [J].
Diaz-Pernas, Francisco Javier ;
Martinez-Zarzuela, Mario ;
Anton-Rodriguez, Miriam ;
Gonzalez-Ortega, David .
HEALTHCARE, 2021, 9 (02)