Development of a federated learning approach to predict acute kidney injury in adult hospitalized patients with COVID-19 in New York City

Federated learning is a technique for training predictive models without sharing patient-level data, thus maintaining data security while allowing inter-institutional collaboration. We used federated learning to predict acute kidney injury within three and seven days of admission, using demographics, comorbidities, vital signs, and laboratory values, in 4029 adults hospitalized with COVID-19 at five sociodemographically diverse New York City hospitals, between March-October 2020. Prediction performance of federated models was generally higher than single-hospital models and was comparable to pooled-data models. In the first use-case in kidney disease, federated learning improved prediction of a common complication of COVID-19, while preserving data privacy.

Temporal Trends in COVID-19 associated AKI from March to December 2020 in New York City

Acute Kidney Injury (AKI) is among the most common complications of Coronavirus Disease 2019 (COVID-19). Throughout 2020 pandemic, the clinical approach to COVID-19 has progressively improved, but it is unknown how these changes have affected AKI incidence and severity. In this retrospective analysis, we report the trend over time of COVID-19 associated AKI and need of renal replacement therapy in a large health system in New York City, the first COVID-19 epicenter in United States.

Federated Learning of Electronic Health Records Improves Mortality Prediction in Patients Hospitalized with COVID-19

Machine learning (ML) models require large datasets which may be siloed across different healthcare institutions. Using federated learning, a ML technique that avoids locally aggregating raw clinical data across multiple institutions, we predict mortality within seven days in hospitalized COVID-19 patients. Patient data was collected from Electronic Health Records (EHRs) from five hospitals within the Mount Sinai Health System (MSHS). Logistic Regression with L1 regularization (LASSO) and Multilayer Perceptron (MLP) models were trained using local data at each site, a pooled model with combined data from all five sites, and a federated model that only shared parameters with a central aggregator. Both the federated LASSO and federated MLP models performed better than their local model counterparts at four hospitals. The federated MLP model also outperformed the federated LASSO model at all hospitals. Federated learning shows promise in COVID-19 EHR data to develop robust predictive models without compromising patient privacy.