Skip to main page content
U.S. flag

An official website of the United States government

Dot gov

The .gov means it’s official.
Federal government websites often end in .gov or .mil. Before sharing sensitive information, make sure you’re on a federal government site.

Https

The site is secure.
The https:// ensures that you are connecting to the official website and that any information you provide is encrypted and transmitted securely.

Access keys NCBI Homepage MyNCBI Homepage Main Content Main Navigation
. 2021 Aug;16(8):1158-1168.
doi: 10.2215/CJN.17311120. Epub 2021 May 24.

Predictive Approaches for Acute Dialysis Requirement and Death in COVID-19

Collaborators, Affiliations

Predictive Approaches for Acute Dialysis Requirement and Death in COVID-19

Akhil Vaid et al. Clin J Am Soc Nephrol. 2021 Aug.

Abstract

Background and objectives: AKI treated with dialysis initiation is a common complication of coronavirus disease 2019 (COVID-19) among hospitalized patients. However, dialysis supplies and personnel are often limited.

Design, setting, participants, & measurements: Using data from adult patients hospitalized with COVID-19 from five hospitals from the Mount Sinai Health System who were admitted between March 10 and December 26, 2020, we developed and validated several models (logistic regression, Least Absolute Shrinkage and Selection Operator (LASSO), random forest, and eXtreme GradientBoosting [XGBoost; with and without imputation]) for predicting treatment with dialysis or death at various time horizons (1, 3, 5, and 7 days) after hospital admission. Patients admitted to the Mount Sinai Hospital were used for internal validation, whereas the other hospitals formed part of the external validation cohort. Features included demographics, comorbidities, and laboratory and vital signs within 12 hours of hospital admission.

Results: A total of 6093 patients (2442 in training and 3651 in external validation) were included in the final cohort. Of the different modeling approaches used, XGBoost without imputation had the highest area under the receiver operating characteristic (AUROC) curve on internal validation (range of 0.93-0.98) and area under the precision-recall curve (AUPRC; range of 0.78-0.82) for all time points. XGBoost without imputation also had the highest test parameters on external validation (AUROC range of 0.85-0.87, and AUPRC range of 0.27-0.54) across all time windows. XGBoost without imputation outperformed all models with higher precision and recall (mean difference in AUROC of 0.04; mean difference in AUPRC of 0.15). Features of creatinine, BUN, and red cell distribution width were major drivers of the model's prediction.

Conclusions: An XGBoost model without imputation for prediction of a composite outcome of either death or dialysis in patients positive for COVID-19 had the best performance, as compared with standard and other machine learning models.

Podcast: This article contains a podcast at https://www.asn-online.org/media/podcast/CJASN/2021_07_09_CJN17311120.mp3.

Keywords: AKI; COVID-19; dialysis; machine learning; prediction.

PubMed Disclaimer

Figures

None
Graphical abstract
Figure 1.
Figure 1.
Overall schema of study. (A) Mount Sinai Hospital (MSH) was used for training and internal validation, whereas the other hospitals were used for external validation. (B) Study flow diagram. (C) Model predictions were generated for hospital day 1, 3, 5, and 7. COVID-19, coronavirus disease 2019; MSB, Mount Sinai Brooklyn; MSM, Mount Sinai Morningside; MSQ, Mount Sinai Queens; MSW, Mount Sinai West.
Figure 2.
Figure 2.
XGBoost without imputation had high area under the receiver operating characteristic (AUROC) curves. AUROC curves for the eXtreme Gradient Boosting (XGBoost) model without imputation at hospital day 1, 3, 5, and 7. Orange line represents results in the training set, blue line represents external validation, and the dotted red line is chance. OH, other hospitals.
Figure 3.
Figure 3.
XGBoost without imputation had high area under the precision-recall curves (AUPRC). AUPRC for the XGBoost model without imputation at hospital day 1, 3, 5, and 7. Orange line represents results in the training set, blue line represents external validation, and the dotted red line is chance. OH, other hospitals.
Figure 4.
Figure 4.
Top features contributing to model prediction were similar across time frames. SHapley Additive exPlanations (SHAP) summary plots showing features driving model prediction toward death or dialysis at 1, 3, 5, and 7 days postadmission. Each dot represents a single patient. In each of the four plots, features are listed on the left in decreasing order of importance. Higher values of a feature are shown in red, and lower values are shown in blue. Positive SHAP values (plotted to the right of the zero on the x axis) push the model toward a prediction of death or dialysis, whereas negative SHAP values push the model away from a prediction of death or dialysis. For example, creatinine levels are the most important feature driving prediction. Because creatinine levels are in red on the right, higher values of creatinine drive model prediction toward the outcome. Similarly, higher oxygen saturation is plotted in red on the left and is associated with the model’s prediction being driven away from death or dialysis. AST, aspartate aminotransferase; RDW, red cell distribution width.

References

    1. Johns Hopkins Coronavirus Resource Center: COVID-19 map. Available at: https://coronavirus.jhu.edu/map.html. Accessed June 12, 2020
    1. Guan WJ, Ni ZY, Hu Y, Liang WH, Ou CQ, He JX, Liu L, Shan H, Lei CL, Hui DSC, Du B, Li LJ, Zeng G, Yuen KY, Chen RC, Tang CL, Wang T, Chen PY, Xiang J, Li SY, Wang JL, Liang ZJ, Peng YX, Wei L, Liu Y, Hu YH, Peng P, Wang JM, Liu JY, Chen Z, Li G, Zheng ZJ, Qiu SQ, Luo J, Ye CJ, Zhu SY, Zhong NS; China Medical Treatment Expert Group for Covid-19: Clinical characteristics of coronavirus disease 2019 in China. N Engl J Med 382: 1708–1720, 2020 - PMC - PubMed
    1. Chan L, Chaudhary K, Saha A, Chauhan K, Vaid A, Zhao S, Paranjpe I, Somani S, Richter F, Miotto R, Lala A, Kia A, Timsina P, Li L, Freeman R, Chen R, Narula J, Just AC, Horowitz C, Fayad Z, Cordon-Cardo C, Schadt E, Levin MA, Reich DL, Fuster V, Murphy B, He JC, Charney AW, Böttinger EP, Glicksberg BS, Coca SG, Nadkarni GN; Mount Sinai COVID Informatics Center (MSCIC): AKI in hospitalized patients with COVID-19. J Am Soc Nephrol 32: 151–160, 2021 - PMC - PubMed
    1. Russo E, Esposito P, Taramasso L, Magnasco L, Saio M, Briano F, Russo C, Dettori S, Vena A, Di Biagio A, Garibotti G, Bassetti M, Viazzi F; GECOVID Working Group: Kidney disease and all-cause mortality in patients with COVID-19 hospitalized in Genoa, Northern Italy. J Nephrol 34: 173–183, 2021 - PMC - PubMed
    1. Reddy YNV, Walensky RP, Mendu ML, Green N, Reddy KP: Estimating shortages in capacity to deliver continuous kidney replacement therapy during the COVID-19 pandemic in the United States. Am J Kidney Dis 76: 696–709.e1, 2020 - PMC - PubMed

Publication types