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*


研究成果: Article同行評審


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.

期刊Health Information Science and Systems
出版狀態Published - 12月 2023


深入研究「Federated machine learning for predicting acute kidney injury in critically ill patients: a multicenter study in Taiwan」主題。共同形成了獨特的指紋。