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 条
[21]   Performance evaluation of feature extraction techniques in MR-Brain image classification system [J].
Khalil, Mohammed ;
Ayad, Habib ;
Adib, Abdellah .
PROCEEDINGS OF THE FIRST INTERNATIONAL CONFERENCE ON INTELLIGENT COMPUTING IN DATA SCIENCES (ICDS2017), 2018, 127 :218-225
[22]   Noninvasive Grading of Glioma Tumor Using Magnetic Resonance Imaging with Convolutional Neural Networks [J].
Khawaldeh, Saed ;
Pervaiz, Usama ;
Rafiq, Azhar ;
Alkhawaldeh, Rami S. .
APPLIED SCIENCES-BASEL, 2018, 8 (01)
[23]   Medical Imaging using Machine Learning and Deep Learning Algorithms: A Review [J].
Latif, Jahanzaib ;
Xiao, Chuangbai ;
Imran, Azhar ;
Tu, Shanshan .
2019 2ND INTERNATIONAL CONFERENCE ON COMPUTING, MATHEMATICS AND ENGINEERING TECHNOLOGIES (ICOMET), 2019,
[24]   An overview of deep learning in medical imaging focusing on MRI [J].
Lundervold, Alexander Selvikvag ;
Lundervold, Arvid .
ZEITSCHRIFT FUR MEDIZINISCHE PHYSIK, 2019, 29 (02) :102-127
[25]   Artificial intelligence with big data analytics-based brain intracranial hemorrhage e-diagnosis using CT images [J].
Mansour, Romany F. ;
Escorcia-Gutierrez, Jose ;
Gamarra, Margarita ;
Garcia Diaz, Vicente ;
Gupta, Deepak ;
Kumar, Sachin .
NEURAL COMPUTING & APPLICATIONS, 2023, 35 (22) :16037-16049
[26]  
Nagabandi A, 2018, IEEE INT CONF ROBOT, P7579
[27]   Brain Tumor Classification Using Dense Efficient-Net [J].
Nayak, Dillip Ranjan ;
Padhy, Neelamadhab ;
Mallick, Pradeep Kumar ;
Zymbler, Mikhail ;
Kumar, Sachin .
AXIOMS, 2022, 11 (01)
[28]   Brain Tumour Classification Using Noble Deep Learning Approach with Parametric Optimization through Metaheuristics Approaches [J].
Nayak, Dillip Ranjan ;
Padhy, Neelamadhab ;
Mallick, Pradeep Kumar ;
Bagal, Dilip Kumar ;
Kumar, Sachin .
COMPUTERS, 2022, 11 (01)
[29]  
Nguyen L., 2016, Applied and Computational Mathematics, V6, P1
[30]   On the Classification of MR Images Using "ELM-SSA" Coated Hybrid Model [J].
Pradhan, Ashwini ;
Mishra, Debahuti ;
Das, Kaberi ;
Panda, Ganapati ;
Kumar, Sachin ;
Zymbler, Mikhail .
MATHEMATICS, 2021, 9 (17)