Patient Clustering Improves Efficiency of Federated Machine Learning to predict mortality and hospital stay time using distributed Electronic Medical Records

2019-03-22 · Li Huang, Dianbo Liu

Electronic medical records (EMRs) supports the development of machine learning algorithms for predicting disease incidence, patient response to treatment, and other healthcare events. But insofar most algorithms have been centralized, taking little account of the decentralized, non-identically independently distributed (non-IID), and privacy-sensitive characteristics of EMRs that can complicate data collection, sharing and learning. To address this challenge, we introduced a community-based federated machine learning (CBFL) algorithm and evaluated it on non-IID ICU EMRs. Our algorithm clustered the distributed data into clinically meaningful communities that captured similar diagnoses and geological locations, and learnt one model for each community. Throughout the learning process, the data was kept local on hospitals, while locally-computed results were aggregated on a server. Evaluation results show that CBFL outperformed the baseline FL algorithm in terms of Area Under the Receiver Operating Characteristic Curve (ROC AUC), Area Under the Precision-Recall Curve (PR AUC), and communication cost between hospitals and the server. Furthermore, communities' performance difference could be explained by how dissimilar one community was to others.

📄 PDF Abstract BibTeX arXiv:1903.09296

Code (0)

등록된 구현이 없습니다.

Tasks

BIG-bench Machine LearningClustering

Federated unsupervised random forest for privacy-preserving patient stratification

2024-01-29 · Bastian Pfeifer, Christel Sirocchi, Marcus D. Bloice, Markus Kreuzthaler 외

In the realm of precision medicine, effective patient stratification and disease subtyping demand innovative methodologies tailored for multi-omics data. Clustering techniques applied to multi-omics data have become inst…

Clustering Feature Importance Privacy Preserving

Privacy-preserving patient clustering for personalized federated learning

2023-07-17 · Ahmed Elhussein, Gamze Gursoy

Federated Learning (FL) is a machine learning framework that enables multiple organizations to train a model without sharing their data with a central server. However, it experiences significant performance degradation i…

Clustering Federated Learning Mortality Prediction Personalized Federated Learning+1

FedHK-MVFC: Federated Heat Kernel Multi-View Clustering

2025-09-19 · Kristina P. Sinaga arxiv

In the realm of distributed artificial intelligence (AI) and privacy-focused medical applications, this paper proposes a multi-view clustering framework that links quantum field theory with federated healthcare analytics…

Federated Learning

Personalized Federated Learning with Clustering: Non-IID Heart Rate Variability Data Application

2021-08-04 · Joo Hun Yoo, Ha Min Son, Hyejun Jeong, Eun-Hye Jang 외

While machine learning techniques are being applied to various fields for their exceptional ability to find complex relations in large datasets, the strengthening of regulations on data ownership and privacy is causing i…

Clustering Federated Learning Heart Rate Variability Personalized Federated Learning+1

A new type of federated clustering: A non-model-sharing approach

2025-06-11 · Yuji Kawamata, Kaoru Kamijo, Maki Kihira, Akihiro Toyoda 외

In recent years, the growing need to leverage sensitive data across institutions has led to increased attention on federated learning (FL), a decentralized machine learning paradigm that enables model training without sh…

Clustering Federated Learning