Abstract
The COVID-19 pandemic has been adversely affecting the patient management systems in hospitals around the world. Radiological imaging, especially chest x-ray and lung Computed Tomography (CT) scans, plays a vital role in the severity analysis of hospitalized COVID-19 patients. However, with an increasing number of patients and a lack of skilled radiologists, automated assessment of COVID-19 severity using medical image analysis has become increasingly important. Chest x-ray (CXR) imaging plays a significant role in assessing the severity of pneumonia, especially in low-resource hospitals, and is the most frequently used diagnostic imaging in the world. Previous methods that automatically predict the severity of COVID-19 pneumonia mainly focus on feature pooling from pre-trained CXR models without explicitly considering the underlying human anatomical attributes. This paper proposes an anatomy-aware (AA) deep learning model that learns the generic features from x-ray images considering the underlying anatomical information. Utilizing a pre-trained model and lung segmentation masks, the model generates a feature vector including disease-level features and lung involvement scores. We have used four different open-source datasets, along with an in-house annotated test set for training and evaluation of the proposed method. The proposed method improves the geographical extent score by 11% in terms of mean squared error (MSE) while preserving the benchmark result in lung opacity score. The results demonstrate the effectiveness of the proposed AA model in COVID-19 severity prediction from chest X-ray images. The algorithm can be used in low-resource setting hospitals for COVID-19 severity prediction, especially where there is a lack of skilled radiologists.
Similar content being viewed by others
References
W.H.O. (2023) WHO Coronavirus (COVID-19) Dashboard With Vaccination Data, https://covid19.who.int/, (Accessed on 03/16/2021)
Irkham I, A. U. Ibrahim, C. W. Nwekwo, F. Al-Turjman, Y. W. Hartati (2022) Current technologies for detection of COVID-19: Biosensors, artificial intelligence and internet of medical things (IOMT). Sensors 23(1):426.
M. Francone, F. Iafrate, G. M. Masci, S. Coco, F. Cilia, L. Manganaro, V. Panebianco, C. Andreoli, M. C. Colaiacomo, M. A. Zingaropoli, et al., Chest ct score in COVID-19 patients: correlation with disease severity and short-term prognosis, European Radiology 30 (12) (2020) 6808–6817.
S. Punitha, F. Al-Turjman, T. Stephan, Genetically optimized computer-aided diagnosis for detection and classification of COVID-19, AI-powered IoT for COVID-19 (2020) 105–122.
J. T. Wu, K. C. Wong, Y. Gur, N. Ansari, A. Karargyris, A. Sharma, M. Morris, B. Saboury, H. Ahmad, O. Boyko, et al., Comparison of chest radiograph interpretations by artificial intelligence algorithm vs radiology residents, JAMA network open 3 (10) (2020) e2022779–e2022779.
E. Hussain, M. Hasan, M. A. Rahman, I. Lee, T. Tamanna, M. Z. Parvez, Corodet: A deep learning based classification for COVID-19 detection using chest x-ray images, Chaos, Solitons & Fractals 142 (2021) 110495.
R. A. Al-Falluji, Z. D. Katheeth, B. Alathari, Automatic detection of COVID-19 using chest x-ray images and modified resnet18-based convolution neural networks, Comput. Mater. Contin. (2021) 1301–1313.
A. Abbas, M. M. Abdelsamea, M. M. Gaber, Classification of COVID-19 in chest x-ray images using detrac deep convolutional neural network, Appl. Intell. 51 (2) (2021) 854–864.
A. Afifi, N. E. Hafsa, M. A. Ali, A. Alhumam, S. Alsalman, An ensemble of global and local-attention based convolutional neural networks for COVID-19 diagnosis on chest x-ray images, Symmetry 13 (1) (2021) 113.
M. Umer, I. Ashraf, S. Ullah, A. Mehmood, G. S. Choi, Covinet: a convolutional neural network approach for predicting COVID-19 from chest x-ray images, J. Ambient Intell. Humaniz. Comput. (2021) 1–13.
G. Wang, X. Liu, J. Shen, C. Wang, Z. Li, L. Ye, X. Wu, T. Chen, K. Wang, X. Zhang, et al., A deep-learning pipeline for the diagnosis and discrimination of viral, non-viral and COVID-19 pneumonia from chest x-ray images, Nature Biomedical Engineering (2021) 1–13.
X. Wang, Y. Peng, L. Lu, Z. Lu, M. Bagheri, R. M. Summers, Chestx-ray8: Hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases, in: Proc. IEEE CVPR, 2017, pp. 2097–2106.
J. P. Cohen, L. Dao, K. Roth, P. Morrison, Y. Bengio, A. F. Abbasi, B. Shen, H. K. Mahsa, M. Ghassemi, H. Li, et al., Predicting COVID-19 pneumonia severity on chest x-ray with deep learning, Cureus 12 (7) (2020).
H. J. Koo, S. Lim, J. Choe, S.-H. Choi, H. Sung, K.-H. Do, Radiographic and ct features of viral pneumonia, Radiographics 38 (3) (2018) 719–739.
A. Narin, C. Kaya, Z. Pamuk, Automatic detection of coronavirus disease (COVID-19) using x-ray images and deep convolutional neural networks, arXiv preprint arXiv:2003.10849 (2020).
A. M. Ismael, A. Şengür, Deep learning approaches for COVID-19 detection based on chest x-ray images, Expert Systems with Applications 164 (2021) 114054.
A. U. Ibrahim, M. Ozsoz, S. Serte, F. Al-Turjman, P. S. Yakoi, Pneumonia classification using deep learning from chest x-ray images during COVID-19, Cognitive Computation (2021) 1–13.
A. S. Al-Waisy, S. Al-Fahdawi, M. A. Mohammed, K. H. Abdulkareem, S. A. Mostafa, M. S. Maashi, M. Arif, B. Garcia-Zapirain, COVID-chexnet: hybrid deep learning framework for identifying COVID-19 virus in chest x-rays images, Soft computing 27 (5) (2023) 2657–2672.
A. Hamza, M. A. Khan, M. Alhaisoni, A. Al Hejaili, K. A. Shaban, S. Alsubai, A. Alasiry, M. Marzougui, D2bof-covidnet: A framework of deep bayesian optimization and fusion-assisted optimal deep features for COVID-19 classification using chest x-ray and MRI scans, Diagnostics 13 (1) (2023) 101.
Y. Kaya, E. Gürsoy, A mobilenet-based CNN model with a novel fine-tuning mechanism for COVID-19 infection detection, Soft Computing (2023) 1–15.
K. Jyoti, S. Sushma, S. Yadav, P. Kumar, R. B. Pachori, S. Mukherjee, Automatic diagnosis of COVID-19 with mca-inspired tqwt-based classification of chest x-ray images, Computers in Biology and Medicine 152 (2023) 106331.
M. M. Islam, M. Z. Islam, A. Asraf, M. S. Al-Rakhami, W. Ding, A. H. Sodhro, Diagnosis of COVID-19 from x-rays using combined CNN-RNN architecture with transfer learning, BenchCouncil Transactions on Benchmarks, Standards and Evaluations (2023) 100088.
S. Motamed, P. Rogalla, F. Khalvati, Data augmentation using generative adversarial networks (gans) for gan-based detection of pneumonia and COVID-19 in chest x-ray images, arXiv preprint arXiv:2006.03622 (2021).
T. Zebin, S. Rezvy, Covid-19 detection and disease progression visualization: Deep learning on chest x-rays for classification and coarse localization, Appl. Intell. 51 (2) (2021) 1010–1021.
Z. Ullah, M. Usman, S. Latif, J. Gwak, Densely attention mechanism based network for COVID-19 detection in chest x-rays, Scientific Reports 13 (1) (2023) 261.
T. B. Chandra, K. Verma, B. K. Singh, D. Jain, S. S. Netam, Coronavirus disease (COVID-19) detection in chest x-ray images using majority voting based classifier ensemble, Expert Syst. Appl. 165 (2021) 113909.
K. K. Singh, A. Singh, Diagnosis of COVID-19 from chest x-ray images using wavelets-based depthwise convolution network, Big Data Mining and Analytics 4 (2) (2021) 84–93.
H. Malik, A. Naeem, R. A. Naqvi, W.-K. Loh, DMFL_net: A federated learning-based framework for the classification of COVID-19 from multiple chest diseases using x-rays, Sensors 23 (2) (2023) 743.
A. Degerli, M. Ahishali, S. Kiranyaz, M. E. Chowdhury, M. Gabbouj, Reliable COVID-19 detection using chest x-ray images, in: 2021 IEEE International Conference on Image Processing (ICIP), IEEE, 2021, pp. 185–189.
A. Haghanifar, M. M. Majdabadi, Y. Choi, S. Deivalakshmi, S. Ko, COVID-CXNET: Detecting COVID-19 in frontal chest x-ray images using deep learning, Multimedia Tools and Applications 81 (21) (2022) 30615–30645.
T. Ozturk, M. Talo, E. A. Yildirim, U. B. Baloglu, O. Yildirim, U. R. Acharya, Automated detection of COVID-19 cases using deep neural networks with x-ray images, Computers in biology and medicine 121 (2020) 103792.
F. Khozeimeh, D. Sharifrazi, N. H. Izadi, J. H. Joloudari, A. Shoeibi, R. Alizadehsani, J. M. Gorriz, S. Hussain, Z. A. Sani, H. Moosaei, et al., Combining a convolutional neural network with autoencoders to predict the survival chance of COVID-19 patients, Scientific Reports 11 (1) (2021) 15343.
A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. Gomez, Ł. Kaiser, I. Polosukhin, Attention is all you need. in advances in neural information processing systems, in: Proc. NIPS, 2007, pp. 5998–6008.
K. Li, Z. Wu, K.-C. Peng, J. Ernst, Y. Fu, Tell me where to look: Guided attention inference network, in: Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, pp. 9215–9223.
J. Hu, L. Shen, G. Sun, Squeeze-and-excitation networks, in: Proceedings of the IEEE conference on computer vision and pattern recognition, 2018, pp. 7132–7141.
N. Sarafianos, X. Xu, I. A. Kakadiaris, Deep imbalanced attribute classification using visual attention aggregation, in: Proceedings of the European Conference on Computer Vision (ECCV), 2018, pp. 680–697.
M. Jaderberg, K. Simonyan, A. Zisserman, et al., Spatial transformer networks, Advances in neural information processing systems 28 (2015).
S. Woo, J. Park, J.-Y. Lee, I. S. Kweon, Cbam: Convolutional block attention module, in: Proceedings of the European conference on computer vision (ECCV), 2018, pp. 3–19.
J. Fu, J. Liu, H. Tian, Y. Li, Y. Bao, Z. Fang, H. Lu, Dual attention network for scene segmentation, in: Proceedings of the IEEE/CVF conference on computer vision and pattern recognition, 2019, pp. 3146–3154.
S. Albahli, N. Ayub, M. Shiraz, Coronavirus disease (COVID-19) detection using x-ray images and enhanced densenet, Applied Soft Computing 110 (2021) 107645.
G. K. Yenurkar, S. Mal, Effective detection of COVID-19 outbreak in chest x-rays using fusionnet model, The Imaging Science Journal (2023) 1–21.
L. T. Duong, P. T. Nguyen, L. Iovino, M. Flammini, Automatic detection of COVID-19 from chest x-ray and lung computed tomography images using deep neural networks and transfer learning, Applied Soft Computing 132 (2023) 109851.
H. Mukherjee, S. Ghosh, A. Dhar, S. M. Obaidullah, K. Santosh, K. Roy, Deep neural network to detect COVID-19: one architecture for both CT scans and chest x-rays, Applied Intelligence 51 (2021) 2777–2789.
S. Roy, A. K. Das, Deep-cov: An integrated deep learning model to detect COVID-19 using chest x-ray and CT images, Computational Intelligence (2023).
M. Karim, T. Döhmen, D. Rebholz-Schuhmann, S. Decker, M. Cochez, O. Beyan, et al., DeepCOVIDexplainer: Explainable COVID-19 predictions based on chest x-ray images, arXiv preprint arXiv:2004.04582 (2020).
H. Liu, L. Wang, Y. Nan, F. Jin, Q. Wang, J. Pu, Sdfn: Segmentation-based deep fusion network for thoracic disease classification in chest x-ray images, Comput. Med. Imaging Graph. 75 (2019) 66–73.
U. Kamal, M. Zunaed, N. B. Nizam, T. Hasan, Anatomy-xnet: An anatomy aware convolutional neural network for thoracic disease classification in chest x-rays, IEEE Journal of Biomedical and Health Informatics 26 (11) (2022) 5518–5528.
T. B. Chandra, K. Verma, D. Jain, S. S. Netam, Localization of the suspected abnormal region in chest radiograph images, in: Proc. IEEE ICPC2T, IEEE, 2020, pp. 204–209.
Ü. Budak, M. Çıbuk, Z. Cömert, A. Şengür, Efficient COVID-19 segmentation from CT slices exploiting semantic segmentation with integrated attention mechanism, Journal of Digital Imaging 34 (2021) 263–272.
J. Zhu, B. Shen, A. Abbasi, M. Hoshmand-Kochi, H. Li, T. Q. Duong, Deep transfer learning artificial intelligence accurately stages COVID-19 lung disease severity on portable chest radiographs, PloS one 15 (7) (2020) e0236621.
Z. Jiao, J. W. Choi, K. Halsey, T. M. L. Tran, B. Hsieh, D. Wang, F. Eweje, R. Wang, K. Chang, J. Wu, et al., Prognostication of patients with COVID-19 using artificial intelligence based on chest x-rays and clinical data: a retrospective study, The Lancet Digital Health 3 (5) (2021) e286–e294.
S. Park, G. Kim, Y. Oh, J. B. Seo, S. M. Lee, J. H. Kim, S. Moon, J.-K. Lim, J. C. Ye, Vision transformer using low-level chest x-ray feature corpus for COVID-19 diagnosis and severity quantification, arXiv preprint arXiv:2104.07235 (2021).
A. Signoroni, M. Savardi, S. Benini, N. Adami, R. Leonardi, P. Gibellini, F. Vaccher, M. Ravanelli, A. Borghesi, R. Maroldi, et al., BS-Net: Learning COVID-19 pneumonia severity on a large chest X-ray dataset, Med. Image Anal. 71 (2021) 102046.
H. Li, P. Xiong, J. An, L. Wang, Pyramid attention network for semantic segmentation. arXiv preprint arXiv:1805.10180 (2018).
W. Ye, J. Yao, H. Xue, Y. Li, Weakly supervised lesion localization with probabilistic-cam pooling, arXiv preprint arXiv:2005.14480 (2020).
A. K. Mondal, A. Agarwal, J. Dolz, C. Desrosiers, Revisiting cyclegan for semi-supervised segmentation, arXiv preprint arXiv:1908.11569 (2019).
J.-Y. Zhu, T. Park, P. Isola, A. A. Efros, Unpaired image-to-image translation using cycle-consistent adversarial networks, in: Proc. IEEE CVPR, 2017, pp. 2223–2232.
O. Ronneberger, P. Fischer, T. Brox, U-net: Convolutional networks for biomedical image segmentation, in: Proc. MICCAI, Springer, 2015, pp. 234–241.
G. Huang, Z. Liu, L. Van Der Maaten, K. Q. Weinberger, Densely connected convolutional networks, in: Proc. IEEE CVPR, 2017, pp. 4700–4708.
B. Zhou, A. Khosla, A. Lapedriza, A. Oliva, A. Torralba, Learning deep features for discriminative localization, in: Proc. IEEE CVPR, 2016, pp. 2921–2929.
J. Shiraishi, S. Katsuragawa, J. Ikezoe, T. Matsumoto, T. Kobayashi, K.-I. Komatsu, M. Matsui, H. Fujita, Y. Kodera, K. Doi, Development of a digital image database for chest radiographs with and without a lung nodule: receiver operating characteristic analysis of radiologists’ detection of pulmonary nodules, Am. J. Roentgenol. 174 (1) (2000) 71–74.
B. Van Ginneken, M. B. Stegmann, M. Loog, Segmentation of anatomical structures in chest radiographs using supervised methods: a comparative study on a public database, Med. Image Anal. 10 (1) (2006) 19–40.
J. Irvin, P. Rajpurkar, M. Ko, Y. Yu, S. Ciurea-Ilcus, C. Chute, H. Marklund, B. Haghgoo, R. Ball, K. Shpanskaya, et al., Chexpert: A large chest radiograph dataset with uncertainty labels and expert comparison, in: Proc. AAAI Conf. on Artificial Intelligence, Vol. 33, 2019, pp. 590–597.
J. Deng, W. Dong, R. Socher, L.-J. Li, K. Li, L. Fei-Fei, Imagenet: A large-scale hierarchical image database, in: Proc. IEEE CVPR, Ieee, 2009, pp. 248–255.
Funding
This work was supported by a Special Grant from the Information and Communication Technology (ICT) Division, Bangladesh.
Author information
Authors and Affiliations
Contributions
Nusrat Binta Nizam: literature review, software/coding, experiments, writing manuscript, editing manuscript, preparing response to review document. Sadi Mohammad Siddiquee: conceptualization, methodology development, software/coding, experiments, writing first draft. Mahbuba Shirin: data collection and annotation. Mohammed Imamul Hassan Bhuiyan: conceptualization, supervision, writing — reviewing and editing. Taufiq Hasan: conceptualization, supervision, writing — reviewing and editing.
Corresponding author
Ethics declarations
Ethics Approval
No human subjects were involved in this research. Existing open-access medical image datasets were used for this work.
Consent to Participate
Not applicable, since no human subjects were involved in this research.
Consent for Publication
Not applicable, since no human subjects were involved in this research.
Competing Interests
The authors declare no competing interests.
Additional information
Publisher's Note
Springer Nature remains neutral with regard to jurisdictional claims in published maps and institutional affiliations.
Rights and permissions
Springer Nature or its licensor (e.g. a society or other partner) holds exclusive rights to this article under a publishing agreement with the author(s) or other rightsholder(s); author self-archiving of the accepted manuscript version of this article is solely governed by the terms of such publishing agreement and applicable law.
About this article
Cite this article
Nizam, N.B., Siddiquee, S.M., Shirin, M. et al. COVID-19 Severity Prediction from Chest X-ray Images Using an Anatomy-Aware Deep Learning Model. J Digit Imaging 36, 2100–2112 (2023). https://doi.org/10.1007/s10278-023-00861-6
Received:
Revised:
Accepted:
Published:
Issue Date:
DOI: https://doi.org/10.1007/s10278-023-00861-6