arXiv · 2403.06298
Analysis of Total Variation Minimization for Clustered Federated Learning
Abstract
A key challenge in federated learning applications is the statistical heterogeneity of local datasets. Clustered federated learning addresses this challenge by identifying clusters of local datasets that are approximately homogeneous. One recent approach to clustered federated learning is generalized total variation minimization (GTVMin). This approach requires a similarity graph which can be obtained by domain expertise or in a data-driven fashion via graph learning techniques. Under a widely applicable clustering assumption, we derive an upper bound the deviation between GTVMin solutions and their cluster-wise averages. This bound provides valuable insights into the effectiveness and robustness of GTVMin in addressing statistical heterogeneity within federated learning environments.
Explore related subjects
Keep this discovery
A. Jung. 2024-03-10. Analysis of Total Variation Minimization for Clustered Federated Learning. https://arxiv.org/abs/2403.06298
Cite the original work for its findings. Save a collection to share your selection of sources.