Multi-branch sustainable convolutional neural network for disease classification

被引:2
作者
Naz, Maria [1 ]
Shah, Munam Ali [1 ]
Khattak, Hasan Ali [2 ,8 ]
Wahid, Abdul [2 ,3 ]
Asghar, Muhammad Nabeel [4 ]
Rauf, Hafiz Tayyab [5 ]
Khan, Muhammad Attique [6 ]
Ameer, Zoobia [7 ]
机构
[1] COMSATS Univ Islamabad, Dept Comp Sci, Islamabad, Pakistan
[2] Natl Univ Sci & Technol NUST, Sch Elect Engn & Comp Sci SEECS, Islamabad 44500, Pakistan
[3] Univ Birmingham, Sch Comp Sci, Dubai, U Arab Emirates
[4] Bahauddin Zakariya Univ, Dept Comp Sci, Multan, Pakistan
[5] Staffordshire Univ, Ctr Smart Syst AI & Cybersecur, Stoke On Trent ST4, England
[6] HITEC Univ Taxila, Taxila, Pakistan
[7] Shaheed Benazir Bhutto Women Univ Peshawar, Peshawar, Pakistan
[8] Natl Univ Sci & Technol NUST, Sch Elect Engn & Comp Sci SEECS, Islamabad, Pakistan
关键词
chest computerized tomography (CT) scan; disease classification; machine learning; corona virus disease-19 (COVID-19); DEEP; PANDEMICS;
D O I
10.1002/ima.22884
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Pandemic and natural disasters are growing more often, imposing even more pressure on life care services and users. There are knowledge gaps regarding how to prevent disasters and pandemics. In recent years, after heart disease, corona virus disease-19 (COVID-19), brain stroke, and cancer are at their peak. Different machine learning and deep learning-based techniques are presented to detect these diseases. Existing technique uses two branches that have been used for detection and prediction of disease accurately such as brain hemorrhage. However, existing techniques have been focused on the detection of specific diseases with double-branches convolutional neural networks (CNNs). There is a need to develop a model to detect multiple diseases at the same time using computerized tomography (CT) scan images. We proposed a model that consists of 12 branches of CNN to detect the different types of diseases with their subtypes using CT scan images and classify them more accurately. We proposed multi-branch sustainable CNN model with deep learning architecture trained on the brain CT hemorrhage, COVID-19 lung CT scans and chest CT scans with subtypes of lung cancers. Feature extracted automatically from preprocessed input data and passed to classifiers for classification in the form of concatenated feature vectors. Six classifiers support vector machine (SVM), decision tree (DT), K-nearest neighbor (K-NN), artificial neural network (ANN), naive Bayes (NB), linear regression (LR) classifiers, and three ensembles the random forest (RF), AdaBoost, gradient boosting ensembles were tested on our model for classification and prediction. Our model achieved the best results on RF on each dataset. Respectively, on brain CT hemorrhage achieved (99.79%) accuracy, on COVID-19 lung CT scans achieved (97.61%), and on chest CT scans dataset achieved (98.77%).
引用
收藏
页码:1621 / 1633
页数:13
相关论文
共 54 条
[1]  
Bandyopadhyay S.K., 2012, INT J ENG SCI ADV TE, V2, P34
[2]   Convolutional neural network approach for automatic tympanic membrane detection and classification [J].
Basaran, Erdal ;
Comert, Zafer ;
Celik, Yuksel .
BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2020, 56
[3]  
Bernheim A, 2020, RADIOLOGY
[4]   Deep-learning framework to detect lung abnormality - A study with chest X-Ray and lung CT scan images [J].
Bhandary, Abhir ;
Prabhu, G. Ananth ;
Rajinikanth, V ;
Thanaraj, K. Palani ;
Satapathy, Suresh Chandra ;
Robbins, David E. ;
Shasky, Charles ;
Zhang, Yu-Dong ;
Tavares, Joao Manuel R. S. ;
Raja, N. Sri Madhava .
PATTERN RECOGNITION LETTERS, 2020, 129 :271-278
[5]   Novel Coronavirus Infection (COVID-19) in Humans: A Scoping Review and Meta-Analysis [J].
Borges do Nascimento, Israel Junior ;
Cacic, Nensi ;
Abdulazeem, Hebatullah Mohamed ;
von Groote, Thilo Caspar ;
Jayarajah, Umesh ;
Weerasekara, Ishanka ;
Esfahani, Meisam Abdar ;
Civile, Vinicius Tassoni ;
Marusic, Ana ;
Jeroncic, Ana ;
Carvas Junior, Nelson ;
Pericic, Tina Poklepovic ;
Zakarija-Grkovic, Irena ;
Meirelles Guimaraes, Silvana Mangeon ;
Bragazzi, Nicola Luigi ;
Bjorklund, Maria ;
Sofi-Mahmudi, Ahmad ;
Altujjar, Mohammad ;
Tian, Maoyi ;
Cespedes Arcani, Diana Maria ;
O'Mathuna, Donal P. ;
Marcolino, Milena Soriano .
JOURNAL OF CLINICAL MEDICINE, 2020, 9 (04)
[6]   The role of CT in case ascertainment and management of COVID-19 pneumonia in the UK: insights from high-incidence regions [J].
Chua, Felix ;
Armstrong-James, Darius ;
Desai, Sujal R. ;
Barnett, Joseph ;
Kouranos, Vasileios ;
Kon, Onn Min ;
Jose, Ricardo ;
Vancheeswaran, Rama ;
Loebinger, Michael R. ;
Wong, Joyce ;
Cutino-Moguel, Maria Teresa ;
Morgan, Cliff ;
Ledot, Stephane ;
Lams, Boris ;
Yip, Wing Ho ;
Li, Leski ;
Lee, Ying Cheong ;
Draper, Adrian ;
Kho, Sze Shyang ;
Renzoni, Elisabetta ;
Ward, Katie ;
Periselneris, Jimstan ;
Grubnic, Sisa ;
Lipman, Marc ;
Wells, Athol U. ;
Devaraj, Anand .
LANCET RESPIRATORY MEDICINE, 2020, 8 (05) :438-440
[7]  
Chung MS, 2020, EUR RADIOL, V30, P2182, DOI [10.1007/s00330-019-06574-1, 10.1148/radiol.2020200230]
[8]   Deep learning based liver cancer detection using watershed transform and Gaussian mixture model techniques [J].
Das, Amita ;
Acharya, U. Rajendra ;
Panda, Soumya S. ;
Sabut, Sukanta .
COGNITIVE SYSTEMS RESEARCH, 2019, 54 :165-175
[9]  
Deng J, 2009, PROC CVPR IEEE, P248, DOI 10.1109/CVPRW.2009.5206848
[10]   Diffuse optical measurement of blood flow, blood oxygenation, and metabolism in a human brain during sensorimotor cortex activation [J].
Durduran, T ;
Yu, GQ ;
Burnett, MG ;
Detre, JA ;
Greenberg, JH ;
Wang, JJ ;
Zhou, C ;
Yodh, AG .
OPTICS LETTERS, 2004, 29 (15) :1766-1768