Long-short-term memory machine learning of longitudinal clinical data accurately predicts acute kidney injury onset in COVID-19: a two-center study

Abstract

ObjectivesThis study used the long-short-term memory (LSTM) artificial intelligence method to model multiple time points of clinical laboratory data, along with demographics and comorbidities, to predict hospital-acquired acute kidney injury (AKI) onset in patients with COVID-19.MethodsMontefiore Health System data consisted of 1982 AKI and 2857 non-AKI (NAKI) hospitalized patients with COVID-19, and Stony Brook Hospital validation data consisted of 308 AKI and 721 NAKI hospitalized patients with COVID-19. Demographic, comorbidities, and longitudinal (3 days before AKI onset) laboratory tests were analyzed. LSTM was used to predict AKI with fivefold cross-validation (80%/20% for training/validation).ResultsThe top predictors of AKI onset were glomerular filtration rate, lactate dehydrogenase, alanine aminotransferase, aspartate aminotransferase, and C-reactive protein. Longitudinal data yielded marked improvement in prediction accuracy over individual time points. The inclusion of comorbidities and demographics further improves prediction accuracy. The best model yielded an area under the curve, accuracy, sensitivity, and specificity to be 0.965 ± 0.003, 89.57 ± 1.64%, 0.95 ± 0.03, and 0.84 ± 0.05, respectively, for the Montefiore validation dataset, and 0.86 ± 0.01, 83.66 ± 2.53%, 0.66 ± 0.10, 0.89 ± 0.03, respectively, for the Stony Brook Hospital validation dataset.ConclusionLSTM model of longitudinal clinical data accurately predicted AKI onset in patients with COVID-19. This approach could help heighten awareness of AKI complications and identify patients for early interventions to prevent long-term renal complications.