Convolutional Neural Network-Machine Learning Model: Hybrid Model for Meningioma Tumour and Healthy Brain Classification

被引:4
作者
Moldovanu, Simona [1 ,2 ]
Tabacaru, Gigi [3 ]
Barbu, Marian [3 ]
机构
[1] Dunarea de Jos Univ Galati, Fac Automat Comp Elect Engn & Elect, Dept Comp Sci & Informat Technol, Galati 800146, Romania
[2] Dunarea de Jos Univ Galati, Modelling & Simulat Lab, 47 Domneasca Str, Galati 800008, Romania
[3] Dunarea de Jos Univ Galati, Fac Automat Comp Elect Engn & Elect, Dept Automat Control & Elect Engn, Galati 800146, Romania
关键词
meningioma tumour; convolutional neural networks; machine learning; transfer learning;
D O I
10.3390/jimaging10090235
中图分类号
TB8 [摄影技术];
学科分类号
0804 ;
摘要
This paper presents a hybrid study of convolutional neural networks (CNNs), machine learning (ML), and transfer learning (TL) in the context of brain magnetic resonance imaging (MRI). The anatomy of the brain is very complex; inside the skull, a brain tumour can form in any part. With MRI technology, cross-sectional images are generated, and radiologists can detect the abnormalities. When the size of the tumour is very small, it is undetectable to the human visual system, necessitating alternative analysis using AI tools. As is widely known, CNNs explore the structure of an image and provide features on the SoftMax fully connected (SFC) layer, and the classification of the items that belong to the input classes is established. Two comparison studies for the classification of meningioma tumours and healthy brains are presented in this paper: (i) classifying MRI images using an original CNN and two pre-trained CNNs, DenseNet169 and EfficientNetV2B0; (ii) determining which CNN and ML combination yields the most accurate classification when SoftMax is replaced with three ML models; in this context, Random Forest (RF), K-Nearest Neighbors (KNN), and Support Vector Machine (SVM) were proposed. In a binary classification of tumours and healthy brains, the EfficientNetB0-SVM combination shows an accuracy of 99.5% on the test dataset. A generalisation of the results was performed, and overfitting was prevented by using the bagging ensemble method.
引用
收藏
页数:13
相关论文
共 50 条
[21]   Depth-restricted convolutional neural network—a model for Gujarati food image classification [J].
Bhoomi Shah ;
Hetal Bhavsar .
The Visual Computer, 2024, 40 :1931-1946
[22]   Melanoma identification and classification model based on fine-tuned convolutional neural network [J].
Almufareh, Maram F. ;
Tariq, Noshina ;
Humayun, Mamoona ;
Khan, Farrukh Aslam .
DIGITAL HEALTH, 2024, 10
[23]   A Novel Deep Convolutional Neural Network Model for Alzheimer’s Disease Classification Using Brain MRI [J].
Ouafae Chaimae Ouchicha ;
Mohammed Ammor .
Automatic Control and Computer Sciences, 2022, 56 :261-271
[24]   A Novel Deep Convolutional Neural Network Model for Alzheimer's Disease Classification Using Brain MRI [J].
Ouchicha, Chaimae ;
Ammor, Ouafae ;
Meknassi, Mohammed .
AUTOMATIC CONTROL AND COMPUTER SCIENCES, 2022, 56 (03) :261-271
[25]   Optimizing the Performance of Breast Cancer Classification by Employing the Same Domain Transfer Learning from Hybrid Deep Convolutional Neural Network Model [J].
Alzubaidi, Laith ;
Al-Shamma, Omran ;
Fadhel, Mohammed A. ;
Farhan, Laith ;
Zhang, Jinglan ;
Duan, Ye .
ELECTRONICS, 2020, 9 (03)
[26]   A lightweight convolutional neural network (CNN) model for diatom classification: DiatomNet [J].
Gunduz, Huseyin ;
Gunal, Serkan .
PEERJ COMPUTER SCIENCE, 2024, 10
[27]   IMPROVING THE CONVOLUTIONAL NEURAL NETWORK PERFORMANCE THROUGH TRANSFER LEARNING FOR BRAIN-MACHINE INTERFACE SYSTEMS [J].
Petoku, Eneo ;
Takahashi, Ryota ;
Capi, Genci .
INTERNATIONAL JOURNAL OF INNOVATIVE COMPUTING INFORMATION AND CONTROL, 2022, 18 (05) :1587-1600
[28]   A Novel Hybrid Model Based on Machine and Deep Learning Techniques for the Classification of Microalgae [J].
Kaya, Volkan ;
Akgul, Smail ;
Tanir, Ozge Zencir .
PHYTON-INTERNATIONAL JOURNAL OF EXPERIMENTAL BOTANY, 2023, 92 (09) :2519-2534
[29]   A hybrid deep neural network model for query intent classification [J].
Xu, Bo ;
Ma, Yunlong ;
Lin, Hongfei .
JOURNAL OF INTELLIGENT & FUZZY SYSTEMS, 2019, 36 (06) :6413-6423
[30]   A text classification network model combining machine learning and deep learning [J].
Chen, Hao ;
Zhang, Haifei ;
Yang, Yuwei ;
He, Long .
INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2024, 44 (03) :182-192