Predicting the Severity of COVID-19 Pneumonia from Chest X-Ray Images: A Convolutional Neural Network Approach
DOI:
https://doi.org/10.4108/eetinis.v12i1.6240Keywords:
Pneumonia Severity, COVID-19, Convolutional Neural Networks, ClassificationAbstract
This study addresses significant limitations of previous works based on the Brixia and COVIDGR datasets, which primarily provided qualitative lung injury scores and focused mainly on detecting mild and moderate cases. To bridge these critical gaps, we developed a unified and comprehensive analytical framework that accurately assesses COVID-19-induced lung injuries across four levels: Normal, Mild, Moderate, and Severe. This approach’s core is a meticulously curated, balanced dataset comprising 9,294 high-quality chest X-ray images. Notably, this dataset has been made widely available to the research community, fostering collaborative efforts and enhancing the precision of lung injury classification at all severity levels. To validate the framework’s effectiveness, we conducted an in-depth evaluation using advanced deep learning models, including VGG16, RegNet, DenseNet, MobileNet, EfficientNet, and Vision Transformer (ViT), on this dataset. The top-performing model was further enhanced by optimizing additional fully connected layers and adjusting weights, achieving an outstanding sensitivity of 94.38%. These results affirm the accuracy and reliability of the proposed solution and demonstrate its potential for broad application in clinical practice. Our study represents a significant step forward in developing AI-powered diagnostic tools, contributing to the timely and precise diagnosis of COVID-19 cases. Furthermore, our dataset and methodological framework hold the potential to serve as a foundation for future research, paving the way for advancements in the detection and classification of respiratory diseases with higher accuracy and efficiency.
Downloads
References
[1] World Health Organization. COVID-19 Situation Report 94 [Internet]. World Health Organization. 2020. Available from: https://www.who.int/docs/default-source/coronaviruse/situation-reports/20200423-sitrep94-covid-19.pdf
[2] Maulanza H, Abidin TF, Mubarak Z, Abdullah A. Model and simulation to reduce COVID-19 new infectious cases: A survey. 2020 International Conference on Electrical Engineering and Informatics (ICELTICs). 2020. doi:10.1109/iceltics50595.2020.9315493 DOI: https://doi.org/10.1109/ICELTICs50595.2020.9315493
[3] Loey M, Manogaran G, Taha MHN, Khalifa NEM. Fighting against COVID-19: A novel deep learning model based on YOLO-v2 with ResNet-50 for medical face mask detection. Sustainable Cities and Society. 2020;102600 DOI: https://doi.org/10.1016/j.scs.2020.102600
[4] World Health Organization. WHO COVID-19 Dashboard [Internet]. World Health Organisation. 2023. Available from: https://covid19.who.int
[5] Chung M, Bernheim A, Mei X, Zhang N, Huang M, Zeng X, et al. CT Imaging Features of 2019 Novel Coronavirus (2019-nCoV). Radiology. 2020;295(1):200230. DOI: https://doi.org/10.1148/radiol.2020200230
[6] Shanahan MC, Akudjedu TN. Australian radiographers’ and radiation therapists’ experiences during the COVID-19 pandemic. Journal of Medical Radiation Sciences. 2021;68(2). DOI: https://doi.org/10.1002/jmrs.462
[7] Wang S, Zha Y, Li W, Wu Q, Li X, Niu M, et al. A fully automatic deep learning system for COVID-19 diagnostic and prognostic analysis. European Respiratory Journal. 2020;56(2):2000775. DOI: https://doi.org/10.1183/13993003.00775-2020
[8] Duong LT, Le NH, Tran TB, Ngo VM, Nguyen PT. Detection of tuberculosis from chest X-ray images: Boosting the performance with vision transformer and transfer learning. Expert Systems with Applications. 2021;184:115519. DOI: https://doi.org/10.1016/j.eswa.2021.115519
[9] Thien B. Nguyen-Tat and Thien-Qua T. Nguyen and Hieu-Nghia Nguyen and Vuong M. Ngo. Enhancing brain tumor segmentation in MRI images: A hybrid approach using UNet, attention mechanisms, and transformers. Egyptian Informatics Journal. 2024; 27:100528. DOI: https://doi.org/10.1016/j.eij.2024.100528
[10] Allioui H, Mohammed MA, Benameur N, Al-Khateeb B, Abdulkareem KH, Garcia-Zapirain B, et al. A Multi-Agent Deep Reinforcement Learning Approach for Enhancement of COVID-19 CT Image Segmentation. Journal of Personalized Medicine. 2022;12(2):309. DOI: https://doi.org/10.3390/jpm12020309
[11] Khan M, Alhaisoni M, Tariq U, Hussain N, Majid A, Damaševičius R, et al. COVID-19 Case Recognition from Chest CT Images by Deep Learning, Entropy-Controlled Firefly Optimization, and Parallel Feature Fusion. Sensors. 2021;21(21):7286. DOI: https://doi.org/10.3390/s21217286
[12] Abayomi-Alli OO, Damaševičius R, Abbasi AA, Maskeli ¯unas R. Detection of COVID-19 from Deep Breathing Sounds Using Sound Spectrum with Image Augmentation and Deep Learning Techniques. Electronics. 2022;11(16):2520. DOI: https://doi.org/10.3390/electronics11162520
[13] Haghanifar A, Majdabadi MM, Choi Y, Deivalakshmi S, Ko S. COVID-CXNet: Detecting COVID-19 in frontal chest X-ray images using deep learning. Multimedia Tools and Applications. 2022; DOI: https://doi.org/10.1007/s11042-022-12156-z
[14] Bhattacharyya A, Bhaik D, Kumar S, Thakur P, Sharma R, Pachori RB. A deep learning based approach for automatic detection of COVID-19 cases using chest X-ray images. Biomedical Signal Processing and Control. 2022;71:103182. DOI: https://doi.org/10.1016/j.bspc.2021.103182
[15] Loey M, El-Sappagh S, Mirjalili S. Bayesian-based optimized deep learning model to detect COVID-19 patients using chest X-ray image data. Computers in Biology and Medicine. 2022;105213. DOI: https://doi.org/10.1016/j.compbiomed.2022.105213
[16] Zoabi Y, Deri-Rozov S, Shomron N. Machine learning-based prediction of COVID-19 diagnosis based on symptoms. npj Digital Medicine [Internet]. 2021;4(1):1–5. DOI: https://doi.org/10.1038/s41746-020-00372-6
[17] He K, Zhao W, Xie X, Ji W, Liu M, Tang Z, et al. Synergistic learning of lung lobe segmentation and hierarchical multi-instance classification for automated severity assessment of COVID-19 in CT images. Pattern Recognition. 2021;113:107828. DOI: https://doi.org/10.1016/j.patcog.2021.107828
[18] Signoroni A, Savardi M, Benini S, Adami N, Leonardi R, Gibellini P, et al. BS-Net: Learning COVID-19 pneumonia severity on a large chest X-ray dataset. Medical Image Analysis. 2021;71:102046. DOI: https://doi.org/10.1016/j.media.2021.102046
[19] Zhu J, Shen B, Abbasi A, Hoshmand-Kochi M, Li H, Duong TQ. Deep transfer learning artificial intelligence accurately stages COVID-19 lung disease severity on portable chest radiographs. Singh D, editor. PLOS ONE. 2020;15(7):e0236621. DOI: https://doi.org/10.1371/journal.pone.0236621
[20] Tang Z, Zhao W, Xie X, Zhong Z, Shi F, Liu J, et al. Severity Assessment of Coronavirus Disease 2019 (COVID-19) Using Quantitative Features from Chest CT Images [Internet]. arXiv.org. 2020.
[21] Carvalho AR, Ranieri A, Gabriel Madeira Werberich, Nery S, Joana Carneiro Pinto, Schmitt W, et al. COVID-19 Chest Computed Tomography to Stratify Severity and Disease Extension by Artificial Neural Network Computer-Aided Diagnosis. 2020;7. DOI: https://doi.org/10.3389/fmed.2020.577609
[22] Irmak E. COVID-19 disease severity assessment using CNN model. IET Image Processing. 2021; DOI: https://doi.org/10.1049/ipr2.12153
[23] Albataineh Z, Aldrweesh F, Alzubaidi MA. COVID-19 CT-images diagnosis and severity assessment using machine learning algorithm. Cluster Computing. 2023; DOI: https://doi.org/10.1007/s10586-023-03972-5
[24] https://brixia.github.io/ [Internet]. Brixia. Available from: https://brixia.github.io/
[25] Tabik S. COVIDGR dataset and COVID-SDNet method-ology for predicting COVID-19 based on Chest X-Ray images. IEEE Journal of Biomedical and Health Informat-ics. 2020;1–1. DOI: https://doi.org/10.1109/JBHI.2020.3037127
[26] A. Abo-Hedibah S, Tharwat N, H. Elmokadem A. Is chest X-ray severity scoring for COVID-19 pneumonia reliable?Polish Journal of Radiology. 2021;86(1):432–9. DOI: https://doi.org/10.5114/pjr.2021.108172
[27] Marques P, Fernandez-Presa L, Carretero A, Gómez-Cabrera MC, Viña J, Signes-Costa J, et al. The radio-graphic assessment of lung edema score of lung edema severity correlates with inflammatory parameters in patients with coronavirus disease 2019—Potential new admission biomarkers to predict coronavirus disease 2019 worsening. Frontiers in Medicine. 2022;9. DOI: https://doi.org/10.3389/fmed.2022.871714
[28] Maroldi R, Paolo Rondi, Giorgio Maria Agazzi, Ravanelli M, Borghesi A, Farina D. Which role for chest x-ray score in predicting the outcome in COVID-19 pneumonia?2021;31(6):4016–22. DOI: https://doi.org/10.1007/s00330-020-07504-2
[29] Borghesi A, Maroldi R. COVID-19 outbreak in Italy: experimental chest X-ray scoring system for quantifying and monitoring disease progression. La radiologia medica. 2020;125(5):509–13. DOI: https://doi.org/10.1007/s11547-020-01200-3
[30] Warren MA, Zhao Z, Koyama T, Bastarache JA, Shaver CM, Semler MW, et al. Severity scoring of lung oedema on the chest radiograph is associated with clinical outcomes in ARDS. Thorax. 2018;73(9):840–6. DOI: https://doi.org/10.1136/thoraxjnl-2017-211280
[31] Valk CMA, Zimatore C, Mazzinari G, Pierrakos C, Sivakorn C, Dechsanga J, et al. The Prognostic Capacity of the Radiographic Assessment for Lung Edema Score in Patients With COVID-19 Acute Respiratory Distress Syndrome—An International Multicenter Observational Study. Frontiers in Medicine. 2022;8. DOI: https://doi.org/10.3389/fmed.2021.772056
[32] Leevy JL, Khoshgoftaar TM, Bauder RA, Seliya N. A survey on addressing high-class imbalance in big data. Journal of Big Data. 2018;5(1). DOI: https://doi.org/10.1186/s40537-018-0151-6
[33] Krawczyk B. Learning from imbalanced data: open challenges and future directions. Progress in Artificial Intelligence. 2016;5(4):221–32. DOI: https://doi.org/10.1007/s13748-016-0094-0
[34] Simonyan K, Zisserman A. Very Deep Convolutional Networks for Large-Scale Image Recognition [Internet]. arXiv.org. 2015. DOI: https://doi.org/10.1109/ICCV.2015.314
[35] Xu J, Pan Y, Pan X, Hoi S, Yi Z, Xu Z. RegNet: Self-Regulated Network for Image Classification. IEEE Transactions on Neural Networks and Learning Systems [Internet]. 2022;1–6.
[36] MHe K, Zhang X, Ren S, Sun J. Deep Residual Learning for Image Recognition. 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). 2016;770–8. DOI: https://doi.org/10.1109/CVPR.2016.90
[37] Michele A, Colin V, Santika DD. MobileNet Convolu-tional Neural Networks and Support Vector Machines for Palmprint Recognition. Procedia Computer Science [Internet]. 2019;157:110–7. [32] Leevy JL, Khoshgoftaar TM, Bauder RA, Seliya N. A survey on addressing high-class imbalance in big data. Journal of Big Data. 2018;5(1). DOI: https://doi.org/10.1016/j.procs.2019.08.147
[38] Le TM, Nguyen-Tat BT, Ngo VM. Automated evaluation of Tuberculosis using Deep Neural Networks. EAI Endorsed Transactions on Industrial Networks and Intelligent Systems. 2022;8(30):e4. DOI: https://doi.org/10.4108/eetinis.v8i30.478
[39] Tan M, Le QV. EfficientNet: Rethinking Model Scaling for Convolutional Neural Networks [Internet]. arXiv.org. 2019.
[40] Mozaffari J, Amirkhani A, Shokouhi SB. A survey on deep learning models for detection of COVID-19. Neural Computing and Applications [Internet]. 2023;1–29. DOI: https://doi.org/10.1007/s00521-023-08683-x
[41] Dosovitskiy A, Beyer L, Kolesnikov A, Weissenborn D, Zhai X, Unterthiner T, et al. An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale. International Conference on Learning Representations. 2021.
[42] Ngo VM, Duong TVT, Nguyen TBT, Nguyen PT, Conlan O. An Efficient Classification Algorithm for Traditional Textile Patterns from Different Cultures Based on Structures. Journal on Computing and Cultural Heritage. 2021;14(4):1–22. DOI: https://doi.org/10.1145/3465381
[43] Danilov, V.V., Litmanovich, D., Proutski, A. et al. Automatic scoring of COVID-19 severity in X-ray imaging based on a novel deep learning workflow. Sci Rep 12, 12791 (2022). https://doi.org/10.1038/s41598-022-15013-z DOI: https://doi.org/10.1038/s41598-022-15013-z
Downloads
Published
How to Cite
Issue
Section
Categories
License
Copyright (c) 2024 EAI Endorsed Transactions on Industrial Networks and Intelligent Systems
This work is licensed under a Creative Commons Attribution 3.0 Unported License.
This is an open-access article distributed under the terms of the Creative Commons Attribution CC BY 3.0 license, which permits unlimited use, distribution, and reproduction in any medium so long as the original work is properly cited.