Using artificial intelligence algorithms to predict the overall survival of hemodialysis patients during the COVID-19 pandemic: A prospective cohort study

Abstract

Background: Hemodialysis (HD) patients are a vulnerable population at high risk for severe complications from COVID-19. The impact of partial COVID-19 vaccination on the survival of HD patients remains uncertain. This prospective cohort study was designed to use artificial intelligence algorithms to predict the survival impact of partial COVID-19 vaccination in HD patients. Methods: A cohort of 433 HD patients was used to develop machine-learning models based on a subset of clinical features assessed between July 1, 2021, and April 29, 2022. The patient cohort was randomly split into training (80%) and testing (20%) sets for model development and evaluation. Machine-learning models, including categorical boosting (CatBoost), light gradient boosting machines (LightGBM), RandomForest, and extreme gradient boosting models (XGBoost), were applied to evaluate their discriminative performance using the patient cohorts. Results: Among these models, LightGBM achieved the highest F1 score of 0.95, followed by CatBoost, RandomForest, and XGBoost, with area under the receiver operating characteristic curve values of 0.94 on the testing dataset. The SHapley Additive explanation summary plot derived from the XGBoost model indicated that key features such as age, albumin, and vaccination details had a significant impact on survival. Furthermore, the fully vaccinated group exhibited higher levels of anti-spike (S) receptor-binding domain antibodies. Conclusion: This prospective cohort study involved using artificial intelligence algorithms to predict overall survival in HD patients during the COVID-19 pandemic. These predictive models assisted in identifying high-risk individuals and guiding vaccination strategies for HD patients, ultimately improving overall prognosis. Further research is warranted to validate and refine these predictive models in larger and more diverse populations of HD patients.