SYNTOLOGY HomeExplorerAtlasCodeMethodologyAboutDevelopersFeedPricing
Paper · 2202.06187 · 2022

On the Convergence of Clustered Federated Learning

arXiv · PDF · Open in the Atlas

Code that ran

We lifted 6 functions out of this paper's own repositories and ran 3 of them in a sandbox. "Ran" means the function executed on a synthesized input and returned a value. It is not a reproduction of the paper's results.

RepositoryRoleRan
jie-ma-ai/FedBase canonical 3 of 6
FunctionStatusWhere it lives
blackbox_mc Ran jie-ma-ai/FedBase/fedbase/utils/model_utils.py
code served (permissive licence) · get_code("ae1e2b2d017d9af0")
conv1x1 Ran jie-ma-ai/FedBase/fedbase/model/resnet.py
code served (permissive licence) · get_code("d9def42110729a85")
conv3x3 Ran jie-ma-ai/FedBase/fedbase/model/resnet.py
code served (permissive licence) · get_code("160bb14bd76201b4")
load_checkpoint Not yet run jie-ma-ai/FedBase/fedbase/utils/model_utils.py
code served (permissive licence) · get_code("1c44445eed1fe5bd")
resnet18 Not yet run jie-ma-ai/FedBase/fedbase/model/resnet.py
code served (permissive licence) · get_code("d9637905b644bd69")
similarity Not yet run jie-ma-ai/FedBase/fedbase/utils/model_utils.py
code served (permissive licence) · get_code("67933efee7675d5f")

Repositories linked to this paper

Some links come from the archived Papers with Code dataset (CC BY-SA 4.0): attribution and licence.

Abstract

Knowledge sharing and model personalization are essential components to tackle the non-IID challenge in federated learning (FL). Most existing FL methods focus on two extremes: 1) to learn a shared model to serve all clients with non-IID data, and 2) to learn personalized models for each client, namely personalized FL. There is a trade-off solution, namely clustered FL or cluster-wise personalized FL, which aims to cluster similar clients into one cluster, and then learn a shared model for all clients within a cluster. This paper is to revisit the research of clustered FL by formulating them into a bi-level optimization framework that could unify existing methods. We propose a new theoretical analysis framework to prove the convergence by considering the clusterability among clients. In addition, we embody this framework in an algorithm, named Weighted Clustered Federated Learning (WeCFL). Empirical analysis verifies the theoretical results and demonstrates the effectiveness of the proposed WeCFL under the proposed cluster-wise non-IID settings.

For agents

The same record, over MCP at https://syntology.ai/mcp:

get_harvested_code_for_paper("2202.06187")
get_code_for_paper("2202.06187")
have("2202.06187")

Connect an agent — have() is free.