|
Algorithm 2 FedProx: federated proximal algorithm. |
-
1:
Input: Number of rounds T, number of clients K, learning rate , proximal term
-
2:
Initialize global model weights
-
3:
for each round do
-
4:
Sample a subset of K clients
-
5:
for each client in parallel do
-
6:
Download global model weights
-
7:
Compute local model weights by performing E epochs of proximal SGD on local data, minimizing the objective:
-
8:
end for
-
9:
Aggregate local models:
-
10:
end for
-
11:
Return: final global model weights
|