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 条
[11]  
Elsharif A., 2019, INT J ACAD INFORM SY, V3, P12
[12]   SqueezeNext: Hardware-Aware Neural Network Design [J].
Gholami, Amir ;
Kwon, Kiseok ;
Wu, Bichen ;
Tai, Zizheng ;
Yue, Xiangyu ;
Jin, Peter ;
Zhao, Sicheng ;
Keutzer, Kurt .
PROCEEDINGS 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW), 2018, :1719-1728
[13]  
Giraddi S, 2017, 2017 INTERNATIONAL CONFERENCE ON CURRENT TRENDS IN COMPUTER, ELECTRICAL, ELECTRONICS AND COMMUNICATION (CTCEEC), P640, DOI 10.1109/CTCEEC.2017.8454968
[14]   PET Image Denoising Using a Deep Neural Network Through Fine Tuning [J].
Gong, Kuang ;
Guan, Jiahui ;
Liu, Chih-Chieh ;
Qi, Jinyi .
IEEE TRANSACTIONS ON RADIATION AND PLASMA MEDICAL SCIENCES, 2019, 3 (02) :153-161
[15]  
Gosavi D., 2019, P 2 INT C ADV SCI TE
[16]   Detection of brain tumors from MRI images base on deep learning using hybrid model CNN and NADE [J].
Hashemzehi, Raheleh ;
Mahdavi, Seyyed Javad Seyyed ;
Kheirabadi, Maryam ;
Kamel, Seyed Reza .
BIOCYBERNETICS AND BIOMEDICAL ENGINEERING, 2020, 40 (03) :1225-1232
[17]   Review of MRI-based brain tumor image segmentation using deep learning methods [J].
Isin, Ali ;
Direkoglu, Cem ;
Sah, Melike .
12TH INTERNATIONAL CONFERENCE ON APPLICATION OF FUZZY SYSTEMS AND SOFT COMPUTING, ICAFS 2016, 2016, 102 :317-324
[18]   Medical Image Segmentation Algorithms using Deformable Models: A Review [J].
Jayadevappa, D. ;
Kumar, S. Srinivas ;
Murty, D. S. .
IETE TECHNICAL REVIEW, 2011, 28 (03) :248-255
[19]  
Kadam A., BRAIN TUMOR CLASSIFI
[20]   Automated segmentation of MR images of brain tumors [J].
Kaus, MR ;
Warfield, SK ;
Nabavi, A ;
Black, PM ;
Jolesz, FA ;
Kikinis, R .
RADIOLOGY, 2001, 218 (02) :586-591