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 条
  • [1] A deep learning-based social distance monitoring framework for COVID-19
    Ahmed, Imran
    Ahmad, Misbah
    Rodrigues, Joel J. P. C.
    Jeon, Gwanggil
    Din, Sadia
    SUSTAINABLE CITIES AND SOCIETY, 2021, 65
  • [2] Modeling a deep transfer learning framework for the classification of COVID-19 radiology dataset
    Fayemiwo M.A.
    Olowookere T.A.
    Arekete S.A.
    Ogunde A.O.
    Odim M.O.
    Oguntunde B.O.
    Olaniyan O.O.
    Ojewumi T.O.
    Oyetade I.S.
    Aremu A.A.
    Kayode A.A.
    PeerJ Computer Science, 2021, 7 : 1 - 34
  • [3] Modeling a deep transfer learning framework for the classification of COVID-19 radiology dataset
    Fayemiwo, Michael Adebisi
    Olowookere, Toluwase Ayobami
    Arekete, Samson Afolabi
    Ogunde, Adewale Opeoluwa
    Odim, Mba Obasi
    Oguntunde, Bosede Oyenike
    Olaniyan, Oluwabunmi Omobolanle
    Ojewumi, Theresa Omolayo
    Oyetade, Idowu Sunday
    Aremu, Ademola Adegoke
    Kayode, Aderonke Anthonia
    PEERJ COMPUTER SCIENCE, 2021, 7
  • [4] Classification of COVID-19 Chest CT Images Based on Ensemble Deep Learning
    Li, Xiaoshuo
    Tan, Wenjun
    Liu, Pan
    Zhou, Qinghua
    Yang, Jinzhu
    JOURNAL OF HEALTHCARE ENGINEERING, 2021, 2021 (2021)
  • [5] Deep learning-based improved snapshot ensemble technique for COVID-19 chest X-ray classification
    Babu, Samson Anosh P.
    Annavarapu, Chandra Sekhara Rao
    APPLIED INTELLIGENCE, 2021, 51 (05) : 3104 - 3120
  • [6] Deep learning-based improved snapshot ensemble technique for COVID-19 chest X-ray classification
    P, Samson Anosh Babu
    Annavarapu, Chandra Sekhara Rao
    Applied Intelligence, 2021, 51 (05): : 3104 - 3120
  • [7] Deep learning-based improved snapshot ensemble technique for COVID-19 chest X-ray classification
    Samson Anosh Babu P
    Chandra Sekhara Rao Annavarapu
    Applied Intelligence, 2021, 51 : 3104 - 3120
  • [8] Deep Stacked Ensemble Learning Model for COVID-19 Classification
    Madhu, G.
    Bharadwaj, B. Lalith
    Boddeda, Rohit
    Vardhan, Sai
    Kautish, K. Sandeep
    Alnowibet, Khalid
    Alrasheedi, Adel F.
    Mohamed, Ali Wagdy
    CMC-COMPUTERS MATERIALS & CONTINUA, 2022, 70 (03): : 5467 - 5486
  • [9] Deep stacked ensemble learning model for COVID-19 classification
    Madhu, G.
    Bharadwaj, B. Lalith
    Boddeda, Rohit
    Vardhan, Sai
    Kautish, K. Sandeep
    Alnowibet, Khalid
    Alrasheedi, Adel F.
    Mohamed, Ali Wagdy
    Computers, Materials and Continua, 2022, 70 (03): : 5467 - 5486
  • [10] Deep ensemble transfer learning-based framework for mammographic image classification
    Parita Oza
    Paawan Sharma
    Samir Patel
    The Journal of Supercomputing, 2023, 79 : 8048 - 8069