Predicting Survival of Hemodialysis Patients using Federated Learning: A Nation-wide Study
Abhiram Raju, Praneeth Vepakomma · 2024
Hemodialysis patients who are on donor lists for kidney transplant may get misidentified, delaying their wait time. Thus, predicting their survival time is crucial for optimizing waiting lists and personalizing treatment plans. Predicting survival times for patients often requires large quantities of high-quality but sensitive data. This data is siloed and since individual datasets are smaller and less diverse, locally trained survival models don’t perform as well as centralized ones. Hence, we propose the use of Federated Learning (FL) in the context of predicting survival for hemodialysis patients. FL can have comparatively better performances than local models while not sharing data between centers. However, despite the increased use of such technologies, the application of FL in survival and even more, dialysis patients remains sparse. This paper studies the performance of FL for data of hemodialysis patients from NephroPlus, India’s largest network of dialysis centers.