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 条
[41]   Convolutional Neural Networks for Drone Model Classification [J].
Dale, H. ;
Antoniou, M. ;
Baker, C. J. ;
Jahangir, M. ;
Catherall, A. .
2021 18TH EUROPEAN RADAR CONFERENCE (EURAD), 2021, :361-364
[42]   Identifying urethral strictures using machine learning: a proof-of-concept evaluation of convolutional neural network model [J].
Jin Kyu Kim ;
Kurt McCammon ;
Catherine Robey ;
Marvin Castillo ;
Odina Gomez ;
Patricia Jarmin L. Pua ;
Francis Pile ;
Manuel See ;
Mandy Rickard ;
Armando J. Lorenzo ;
Michael E. Chua .
World Journal of Urology, 2022, 40 :3107-3111
[43]   Identifying urethral strictures using machine learning: a proof-of-concept evaluation of convolutional neural network model [J].
Kim, Jin Kyu ;
McCammon, Kurt ;
Robey, Catherine ;
Castillo, Marvin ;
Gomez, Odina ;
Pua, Patricia Jarmin L. ;
Pile, Francis ;
See, Manuel ;
Rickard, Mandy ;
Lorenzo, Armando J. ;
Chua, Michael E. .
WORLD JOURNAL OF UROLOGY, 2022, 40 (12) :3107-3111
[44]   A hybrid convolutional neural network for intelligent wear particle classification [J].
Peng, Yeping ;
Cai, Junhao ;
Wu, Tonghai ;
Cao, Guangzhong ;
Kwok, Ngaiming ;
Zhou, Shengxi ;
Peng, Zhongxiao .
TRIBOLOGY INTERNATIONAL, 2019, 138 :166-173
[45]   Classification model for reducing absenteeism of nurses at hospitals using machine learning and artificial neural network techniques [J].
Alzu'bi, Dalia ;
El-Heis, Mwaffaq ;
Alsoud, Anas Ratib ;
Almahmoud, Mothanna ;
Abualigah, Laith .
INTERNATIONAL JOURNAL OF SYSTEM ASSURANCE ENGINEERING AND MANAGEMENT, 2024, 15 (07) :3266-3278
[46]   The Application of a Hybrid Transfer Algorithm Based on a Convolutional Neural Network Model and an Improved Convolution Restricted Boltzmann Machine Model in Facial Expression Recognition [J].
Wang, Yingying ;
Li, Yibin ;
Song, Yong ;
Rong, Xuewen .
IEEE ACCESS, 2019, 7 :184599-184610
[47]   Metabolite selection for machine learning in childhood brain tumour classification [J].
Zhao, Dadi ;
Grist, James T. ;
Rose, Heather E. L. ;
Davies, Nigel P. ;
Wilson, Martin ;
MacPherson, Lesley ;
Abernethy, Laurence J. ;
Avula, Shivaram ;
Pizer, Barry ;
Gutierrez, Daniel R. ;
Jaspan, Tim ;
Morgan, Paul S. ;
Mitra, Dipayan ;
Bailey, Simon ;
Sawlani, Vijay ;
Arvanitis, Theodoros N. ;
Sun, Yu ;
Peet, Andrew C. .
NMR IN BIOMEDICINE, 2022, 35 (06)
[48]   Metric Learning Based Convolutional Neural Network for Left-Right Brain Dominance Classification [J].
Lim, Zheng You ;
Sim, Kok Swee ;
Tan, Shing Chiang .
IEEE ACCESS, 2021, 9 :120551-120566
[49]   A Comparative Study on Machine Learning Algorithms and A Hybrid Model of Genetic Algorithm and Neural Network for Mesothelioma Diagnosis [J].
Leong, Mae .
2020 IEEE CONFERENCE ON COMPUTATIONAL INTELLIGENCE IN BIOINFORMATICS AND COMPUTATIONAL BIOLOGY (CIBCB), 2020, :146-153
[50]   Transfer learning with deep convolutional neural network for constitution classification with face image [J].
Huan, Er-Yang ;
Wen, Gui-Hua .
MULTIMEDIA TOOLS AND APPLICATIONS, 2020, 79 (17-18) :11905-11919