Federated machine learning for predicting acute kidney injury in critically ill patients: a multicenter study in Taiwan

Chun Te Huang, Tsai Jung Wang, Li Kuo Kuo, Ming Ju Tsai, Cong Tat Cia, Dung Hung Chiang, Po Jen Chang, Inn Wen Chong, Yi Shan Tsai, Yuan Chia Chu, Chia Jen Liu, Cheng Hsu Chen, Kai Chih Pai, Chieh Liang Wu

Research output: Contribution to journalArticlepeer-review

Abstract

Purpose: To address the contentious data sharing across hospitals, this study adopted a novel approach, federated learning (FL), to establish an aggregate model for acute kidney injury (AKI) prediction in critically ill patients in Taiwan. Methods: This study used data from the Critical Care Database of Taichung Veterans General Hospital (TCVGH) from 2015 to 2020 and electrical medical records of the intensive care units (ICUs) between 2018 and 2020 of four referral centers in different areas across Taiwan. AKI prediction models were trained and validated thereupon. An FL-based prediction model across hospitals was then established. Results: The study included 16,732 ICU admissions from the TCVGH and 38,424 ICU admissions from the other four hospitals. The complete model with 60 features and the parsimonious model with 21 features demonstrated comparable accuracies using extreme gradient boosting, neural network (NN), and random forest, with an area under the receiver-operating characteristic (AUROC) curve of approximately 0.90. The Shapley Additive Explanations plot demonstrated that the selected features were the key clinical components of AKI for critically ill patients. The AUROC curve of the established parsimonious model for external validation at the four hospitals ranged from 0.760 to 0.865. NN-based FL slightly improved the model performance at the four centers. Conclusion: A reliable prediction model for AKI in ICU patients was developed with a lead time of 24 h, and it performed better when the novel FL platform across hospitals was implemented.

Original languageEnglish
Article number48
JournalHealth Information Science and Systems
Volume11
Issue number1
DOIs
Publication statusPublished - 2023 Dec

All Science Journal Classification (ASJC) codes

  • Health Informatics
  • Health Information Management
  • Information Systems

Fingerprint

Dive into the research topics of 'Federated machine learning for predicting acute kidney injury in critically ill patients: a multicenter study in Taiwan'. Together they form a unique fingerprint.

Cite this