Distributionally Robust Clustered Federated Learning: A Case Study in Healthcare
Fuente:
arXiv
Salvato in:
| Autori principali: | , , , , , , |
|---|---|
| Natura: | Preprint |
| Pubblicazione: |
2024
|
| Soggetti: | |
| Accesso online: | |
| Tags: |
Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
|
| _version_ | 1866916429490552832 |
|---|---|
| author | Konti, Xenia Riess, Hans Giannopoulos, Manos Shen, Yi Pencina, Michael J. Economou-Zavlanos, Nicoleta J. Zavlanos, Michael M. |
| author_facet | Konti, Xenia Riess, Hans Giannopoulos, Manos Shen, Yi Pencina, Michael J. Economou-Zavlanos, Nicoleta J. Zavlanos, Michael M. |
| contents | In this paper, we address the challenge of heterogeneous data distributions in cross-silo federated learning by introducing a novel algorithm, which we term Cross-silo Robust Clustered Federated Learning (CS-RCFL). Our approach leverages the Wasserstein distance to construct ambiguity sets around each client's empirical distribution that capture possible distribution shifts in the local data, enabling evaluation of worst-case model performance. We then propose a model-agnostic integer fractional program to determine the optimal distributionally robust clustering of clients into coalitions so that possible biases in the local models caused by statistically heterogeneous client datasets are avoided, and analyze our method for linear and logistic regression models. Finally, we discuss a federated learning protocol that ensures the privacy of client distributions, a critical consideration, for instance, when clients are healthcare institutions. We evaluate our algorithm on synthetic and real-world healthcare data. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2410_07039 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Distributionally Robust Clustered Federated Learning: A Case Study in Healthcare Konti, Xenia Riess, Hans Giannopoulos, Manos Shen, Yi Pencina, Michael J. Economou-Zavlanos, Nicoleta J. Zavlanos, Michael M. Machine Learning In this paper, we address the challenge of heterogeneous data distributions in cross-silo federated learning by introducing a novel algorithm, which we term Cross-silo Robust Clustered Federated Learning (CS-RCFL). Our approach leverages the Wasserstein distance to construct ambiguity sets around each client's empirical distribution that capture possible distribution shifts in the local data, enabling evaluation of worst-case model performance. We then propose a model-agnostic integer fractional program to determine the optimal distributionally robust clustering of clients into coalitions so that possible biases in the local models caused by statistically heterogeneous client datasets are avoided, and analyze our method for linear and logistic regression models. Finally, we discuss a federated learning protocol that ensures the privacy of client distributions, a critical consideration, for instance, when clients are healthcare institutions. We evaluate our algorithm on synthetic and real-world healthcare data. |
| title | Distributionally Robust Clustered Federated Learning: A Case Study in Healthcare |
| topic | Machine Learning |
| url | https://arxiv.org/abs/2410.07039 |