Covid-19 mortality risk prediction using small dataset of chest x-ray images

Olowolayemo, Akeem and Shams, Wafaa Khazaal and Omer, Abubakar Yagoub Ibrahim and Mohammed, Yasin and Batha, Mohammed Raashid Salih (2023) Covid-19 mortality risk prediction using small dataset of chest x-ray images. Artificial Intelligence and Applications (AIA). ISSN 2811-0854 (In Press)

PDF (Article - In Press) - Accepted Version Restricted to Repository staff only Download (2MB) | Request a copy

Abstract

COVID-19 outbreak ravaged the whole world starting from the early part of 2020. The rapid spread of the pandemic accounts for the major reason the world was thrown into panic mode and pervasive confusion. However, COVID-19’s greatest strength is its virility but its severity on an individual is mostly ambiguous, which is dependent on the particular individual. This, combined with the increasingly limited capacity of the global healthcare infrastructure warrants some mechanism that can predict the prognosis of an individual to better determine if the patient would require hospital resources or be better treated as an outpatient. The lack of such a mechanism leads to suboptimal utilization of valuable hospital resources leading to unnecessary loss of life. However, often at the onset of a pandemic such as it was experienced during the outbreak of COVID-19, ample and appropriately labelled dataset to build accurate deep learning models to assist in this respect was limited. In this vein, frantic efforts were made to acquire dataset to train deep learning models for the stated objectives, unfortunately only a small dataset from a single source was available at the time of the study. Consequently, deep learning models based on the ResNet-18 architecture were trained on a small dataset of chest X-rays of patients infected with COVID-19 to predict mortality risk. The models exhibit considerable accuracy with high sensitivity. The appropriateness of the techniques proposed in this study for predictive modelling maybe particularly suited when only small datasets are available especially at the onset of similar pandemics. From existing literature, models with low complexity such as ResNet perform better with small dataset. Hence, this study utilised ResNet-18 as the baseline to evaluate the performance of other popular models on small datasets. The performance of the baseline models based on ResNet-18 with an accuracy of 0.89 compared favourably with those of the several other models including AlexNet, MobileNetV3, EfficientNetV2, SwinTransformer, and ConvNeXt using the same datasets and similar parameters.

Item Type:	Article (Journal)
Uncontrolled Keywords:	small dataset, deep learning, convolutional neural networks(CNNs), X-Rays image classification, COVID-19 mortality
Subjects:	Q Science > QA Mathematics > QA75 Electronic computers. Computer science
Kulliyyahs/Centres/Divisions/Institutes (Can select more than one option. Press CONTROL button):	Kulliyyah of Information and Communication Technology Kulliyyah of Information and Communication Technology Kulliyyah of Information and Communication Technology > Department of Computer Science Kulliyyah of Information and Communication Technology > Department of Computer Science
Depositing User:	Dr Akeem Olowolayemo
Date Deposited:	27 Sep 2023 09:25
Last Modified:	27 Sep 2023 09:26
URI:	http://irep.iium.edu.my/id/eprint/95715

Actions (login required)

View Item