Federated Learning Can Find Friends That Are Advantageous

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tupitsa, Nazarii, Horváth, Samuel, Takáč, Martin, Gorbunov, Eduard
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929423274475520
author Tupitsa, Nazarii
Horváth, Samuel
Takáč, Martin
Gorbunov, Eduard
author_facet Tupitsa, Nazarii
Horváth, Samuel
Takáč, Martin
Gorbunov, Eduard
contents In Federated Learning (FL), the distributed nature and heterogeneity of client data present both opportunities and challenges. While collaboration among clients can significantly enhance the learning process, not all collaborations are beneficial; some may even be detrimental. In this study, we introduce a novel algorithm that assigns adaptive aggregation weights to clients participating in FL training, identifying those with data distributions most conducive to a specific learning objective. We demonstrate that our aggregation method converges no worse than the method that aggregates only the updates received from clients with the same data distribution. Furthermore, empirical evaluations consistently reveal that collaborations guided by our algorithm outperform traditional FL approaches. This underscores the critical role of judicious client selection and lays the foundation for more streamlined and effective FL implementations in the coming years.
format Preprint
id arxiv_https___arxiv_org_abs_2402_05050
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Federated Learning Can Find Friends That Are Advantageous
Tupitsa, Nazarii
Horváth, Samuel
Takáč, Martin
Gorbunov, Eduard
Machine Learning
Optimization and Control
In Federated Learning (FL), the distributed nature and heterogeneity of client data present both opportunities and challenges. While collaboration among clients can significantly enhance the learning process, not all collaborations are beneficial; some may even be detrimental. In this study, we introduce a novel algorithm that assigns adaptive aggregation weights to clients participating in FL training, identifying those with data distributions most conducive to a specific learning objective. We demonstrate that our aggregation method converges no worse than the method that aggregates only the updates received from clients with the same data distribution. Furthermore, empirical evaluations consistently reveal that collaborations guided by our algorithm outperform traditional FL approaches. This underscores the critical role of judicious client selection and lays the foundation for more streamlined and effective FL implementations in the coming years.
title Federated Learning Can Find Friends That Are Advantageous
topic Machine Learning
Optimization and Control
url https://arxiv.org/abs/2402.05050