Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologies

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Guo, Yongxin, Tang, Xiaoying, Lin, Tao
Natura: Preprint
Pubblicazione: 2023
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913713513037824
author Guo, Yongxin
Tang, Xiaoying
Lin, Tao
author_facet Guo, Yongxin
Tang, Xiaoying
Lin, Tao
contents Federated Learning (FL) is an evolving distributed machine learning approach that safeguards client privacy by keeping data on edge devices. However, the variation in data among clients poses challenges in training models that excel across all local distributions. Recent studies suggest clustering as a solution to address client heterogeneity in FL by grouping clients with distribution shifts into distinct clusters. Nonetheless, the diverse learning frameworks used in current clustered FL methods create difficulties in integrating these methods, leveraging their advantages, and making further enhancements. To this end, this paper conducts a thorough examination of existing clustered FL methods and introduces a four-tier framework, named HCFL, to encompass and extend the existing approaches. Utilizing the HCFL, we identify persistent challenges associated with current clustering methods in each tier and propose an enhanced clustering method called HCFL$^{+}$ to overcome these challenges. Through extensive numerical evaluations, we demonstrate the effectiveness of our clustering framework and the enhanced components. Our code is available at https://github.com/LINs-lab/HCFL.
format Preprint
id arxiv_https___arxiv_org_abs_2310_05397
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologies
Guo, Yongxin
Tang, Xiaoying
Lin, Tao
Machine Learning
Federated Learning (FL) is an evolving distributed machine learning approach that safeguards client privacy by keeping data on edge devices. However, the variation in data among clients poses challenges in training models that excel across all local distributions. Recent studies suggest clustering as a solution to address client heterogeneity in FL by grouping clients with distribution shifts into distinct clusters. Nonetheless, the diverse learning frameworks used in current clustered FL methods create difficulties in integrating these methods, leveraging their advantages, and making further enhancements. To this end, this paper conducts a thorough examination of existing clustered FL methods and introduces a four-tier framework, named HCFL, to encompass and extend the existing approaches. Utilizing the HCFL, we identify persistent challenges associated with current clustering methods in each tier and propose an enhanced clustering method called HCFL$^{+}$ to overcome these challenges. Through extensive numerical evaluations, we demonstrate the effectiveness of our clustering framework and the enhanced components. Our code is available at https://github.com/LINs-lab/HCFL.
title Enhancing Clustered Federated Learning: Integration of Strategies and Improved Methodologies
topic Machine Learning
url https://arxiv.org/abs/2310.05397