Multi-branch sustainable convolutional neural network for disease classification

Maria Naz, Munam Ali Shah, Hasan Ali Khattak, Abdul Wahid, Muhammad Nabeel Asghar, Hafiz Tayyab Rauf, Muhammad Attique Khan, Zoobia Ameer

Research output: Contribution to journalArticlepeer-review

Abstract

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), naïve 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%).
Original languageEnglish
Pages (from-to)1621-1633
Number of pages13
JournalInternational Journal of Imaging Systems and Technology
Volume33
Issue number5
DOIs
Publication statusPublished - Sept 2023
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2023 Wiley Periodicals LLC.

Keywords

  • chest computerized tomography (CT) scan
  • corona virus disease-19 (COVID-19)
  • disease classification
  • machine learning

Fingerprint

Dive into the research topics of 'Multi-branch sustainable convolutional neural network for disease classification'. Together they form a unique fingerprint.

Cite this