Sagar Dikshant, Dwivedi Tanima, Gupta Anubha, Aggarwal Priya, Bhatnagar Sushma, Mohan Anant, Kaur Punit, Gupta Ritu
Computer Science, Indraprastha Institute of Information Technology - Delhi, Delhi, IND.
Computer Science, Calfornia State University, Los Angeles, Los Angeles, USA.
Cureus. 2024 Mar 31;16(3):e57336. doi: 10.7759/cureus.57336. eCollection 2024 Mar.
The global spread of COVID-19 has led to significant mortality and morbidity worldwide. Early identification of COVID-19 patients who are at high risk of developing severe disease can help in improved patient management, care, and treatment, as well as in the effective allocation of hospital resources. The severity prediction at the time of hospitalization can be extremely helpful in deciding the treatment of COVID-19 patients. To this end, this study presents an interpretable artificial intelligence (AI) model, named COVID-19 severity predictor (CoSP) that predicts COVID-19 severity using the clinical features at the time of hospital admission. We utilized a dataset comprising 64 demographic and laboratory features of 7,416 confirmed COVID-19 patients that were collected at the time of hospital admission. The proposed hierarchical CoSP model performs four-class COVID severity risk prediction into asymptomatic, mild, moderate, and severe categories. CoSP yielded better performance with good interpretability, as observed via Shapley analysis on COVID severity prediction compared to the other popular ML methods, with an area under the received operating characteristic curve (AUC-ROC) of 0.95, an area under the precision-recall curve (AUPRC) of 0.91, and a weighted F1-score of 0.83. Out of 64 initial features, 19 features were inferred as predictive of the severity of COVID-19 disease by the CoSP model. Therefore, an AI model predicting COVID-19 severity may be helpful for early intervention, optimizing resource allocation, and guiding personalized treatments, potentially enabling healthcare professionals to save lives and allocate resources effectively in the fight against the pandemic.
新冠病毒病(COVID-19)在全球的传播已导致全球范围内大量的死亡和发病。早期识别有发展为重症疾病高风险的COVID-19患者有助于改善患者管理、护理和治疗,以及有效分配医院资源。住院时的严重程度预测对于决定COVID-19患者的治疗极为有帮助。为此,本研究提出了一种名为COVID-19严重程度预测器(CoSP)的可解释人工智能(AI)模型,该模型利用入院时的临床特征来预测COVID-19的严重程度。我们使用了一个数据集,该数据集包含7416例确诊COVID-19患者在入院时收集的64个人口统计学和实验室特征。所提出的分层CoSP模型对COVID严重程度进行四类风险预测,分为无症状、轻度、中度和重度类别。与其他流行的机器学习方法相比,通过对COVID严重程度预测的Shapley分析观察到,CoSP具有更好的性能和良好的可解释性,其接受操作特征曲线下面积(AUC-ROC)为0.95,精确召回率曲线下面积(AUPRC)为0.91,加权F1分数为0.83。在64个初始特征中,CoSP模型推断出19个特征可预测COVID-19疾病的严重程度。因此,一个预测COVID-19严重程度的AI模型可能有助于早期干预、优化资源分配和指导个性化治疗,有可能使医疗保健专业人员在抗击疫情中拯救生命并有效分配资源。