|
Algorithm 1 pFL -GMCO Algorithm |
-
Require:
Private datasets in multiple sites, pFL models , number of epochs L, local learning rate , federated learning rate , variance of Gaussian noise, local learning rate , federated learning rate .
-
Ensure:
Optimal weights of pFL model in site ;
-
1:
for do
-
2:
Local updating:
-
3:
for do
-
4:
Compute local classification loss:
-
5:
Update pFL models locally:
-
6:
end for
-
7:
Federated adaptive aggregation:
-
8:
for do
-
9:
Compute gradient matching loss:
-
10:
Compute the adaptive aggregating weights:
-
11:
Federated aggregation:
-
12:
end for
-
13:
Federated optimization:
-
14:
for do
-
15:
Compute gradient matching loss:
-
16:
Compute model contrastive loss:
-
17:
Compute classification loss:
-
18:
Compute federated optimization loss:
-
19:
Update personalized federated parameters:
-
20:
end for
-
21:
Update multimodal classifier:
-
22:
end for
-
23:
Return the pFL model for each federated site.
|