A Novel Deep Learning-Based Classification Framework for COVID-19 Assisted with Weighted Average Ensemble Modeling

被引:2
|
作者
Chakraborty, Gouri Shankar [1 ]
Batra, Salil [1 ]
Singh, Aman [2 ,3 ,4 ]
Muhammad, Ghulam [5 ]
Torres, Vanessa Yelamos [3 ,6 ,7 ]
Mahajan, Makul [1 ]
机构
[1] Lovely Profess Univ, Dept Comp Sci & Engn, Phagwara 144411, Punjab, India
[2] Univ Europea Atlant, Higher Polytech Sch, C Isabel Torres 21, Santander 39011, Spain
[3] Univ Int Iberoamer, Dept Engn, Arecibo, PR 00613 USA
[4] Uttaranchal Univ, Uttaranchal Inst Technol, Dehra Dun 248007, Uttaranchal, India
[5] King Saud Univ, Coll Comp & Informat Sci, Dept Comp Engn, Riyadh 11543, Saudi Arabia
[6] Univ Europea Atlantico, Engn Res & Innovat Grp, C Isabel Torres 21, Santander 39011, Spain
[7] Univ Int Iberoamericana, Dept Project Management, Campeche 24560, Mexico
关键词
deep learning; convolutional neural network; image classification; COVID-19; ensemble prediction;
D O I
10.3390/diagnostics13101806
中图分类号
R5 [内科学];
学科分类号
1002 ; 100201 ;
摘要
COVID-19 is an infectious disease caused by the deadly virus SARS-CoV-2 that affects the lung of the patient. Different symptoms, including fever, muscle pain and respiratory syndrome, can be identified in COVID-19-affected patients. The disease needs to be diagnosed in a timely manner, otherwise the lung infection can turn into a severe form and the patient's life may be in danger. In this work, an ensemble deep learning-based technique is proposed for COVID-19 detection that can classify the disease with high accuracy, efficiency, and reliability. A weighted average ensemble (WAE) prediction was performed by combining three CNN models, namely Xception, VGG19 and ResNet50V2, where 97.25% and 94.10% accuracy was achieved for binary and multiclass classification, respectively. To accurately detect the disease, different test methods have been proposed and developed, some of which are even being used in real-time situations. RT-PCR is one of the most successful COVID-19 detection methods, and is being used worldwide with high accuracy and sensitivity. However, complexity and time-consuming manual processes are limitations of this method. To make the detection process automated, researchers across the world have started to use deep learning to detect COVID-19 applied on medical imaging. Although most of the existing systems offer high accuracy, different limitations, including high variance, overfitting and generalization errors, can be found that can degrade the system performance. Some of the reasons behind those limitations are a lack of reliable data resources, missing preprocessing techniques, a lack of proper model selection, etc., which eventually create reliability issues. Reliability is an important factor for any healthcare system. Here, transfer learning with better preprocessing techniques applied on two benchmark datasets makes the work more reliable. The weighted average ensemble technique with hyperparameter tuning ensures better accuracy than using a randomly selected single CNN model.
引用
收藏
页数:27
相关论文
共 50 条
  • [21] A novel deep learning-based method for COVID-19 pneumonia detection from CT images
    Luo, Ju
    Sun, Yuhao
    Chi, Jingshu
    Liao, Xin
    Xu, Canxia
    BMC MEDICAL INFORMATICS AND DECISION MAKING, 2022, 22 (01)
  • [22] Deep Learning-Based Approaches to Improve Classification Parameters for Diagnosing COVID-19 from CT Images
    Yasar, Huseyin
    Ceylan, Murat
    COGNITIVE COMPUTATION, 2024, 16 (04) : 1806 - 1833
  • [23] Deep learning-based automated COVID-19 classification from computed tomography images
    Morani, Kenan
    Unay, D.
    COMPUTER METHODS IN BIOMECHANICS AND BIOMEDICAL ENGINEERING-IMAGING AND VISUALIZATION, 2023, 11 (06) : 2145 - 2160
  • [24] A deep learning-based framework for detecting COVID-19 patients using chest X-rays
    Asif, Sohaib
    Zhao, Ming
    Tang, Fengxiao
    Zhu, Yusen
    MULTIMEDIA SYSTEMS, 2022, 28 (04) : 1495 - 1513
  • [25] CSGBBNet: An Explainable Deep Learning Framework for COVID-19 Detection
    Yao, Xu-Jing
    Zhu, Zi-Quan
    Wang, Shui-Hua
    Zhang, Yu-Dong
    DIAGNOSTICS, 2021, 11 (09)
  • [26] CovFrameNet: An Enhanced Deep Learning Framework for COVID-19 Detection
    Oyelade, Olaide Nathaniel
    Ezugwu, Absalom El-Shamir
    Chiroma, Haruna
    IEEE ACCESS, 2021, 9 : 77905 - 77919
  • [27] DIAGNOSING COVID-19 FROM CT IMAGES BASED ON AN ENSEMBLE LEARNING FRAMEWORK
    Li, Bingyang
    Zhang, Qi
    Song, Yinan
    Zhao, Zhicheng
    Meng, Zhu
    Su, Fei
    2021 IEEE INTERNATIONAL CONFERENCE ON ACOUSTICS, SPEECH AND SIGNAL PROCESSING (ICASSP 2021), 2021, : 8563 - 8567
  • [28] Unsupervised Deep Learning based Variational Autoencoder Model for COVID-19 Diagnosis and Classification
    Mansour, Romany F.
    Escorcia-Gutierrez, Jose
    Gamarra, Margarita
    Gupta, Deepak
    Castillo, Oscar
    Kumar, Sachin
    PATTERN RECOGNITION LETTERS, 2021, 151 : 267 - 274
  • [29] An Efficient Deep Learning Framework of COVID-19 CT Scans Using Contrastive Learning and Ensemble Strategy
    Zhang, Shenghan
    Zou, Binyi
    Xu, Binquan
    Su, Jionglong
    Hu, Huafeng
    PROCEEDINGS OF THE 2021 IEEE INTERNATIONAL CONFERENCE ON PROGRESS IN INFORMATICS AND COMPUTING (PIC), 2021, : 388 - 396
  • [30] Deep Transfer Learning Based Classification Model for COVID-19 Disease
    Pathak, Y.
    Tiwari, A.
    Stalin, S.
    Singh, S.
    Shukla, P. K.
    IRBM, 2022, 43 (02) : 87 - 92