|
Algorithm 2: Hierarchical Federated Learning with Gradient Compression |
Initial model ,
Edge nodes ,
Fog groups ,
Learning rate
Updated global model
Train local model on
Compute local gradient
Compress gradient:
Send to fog node
Aggregate compressed gradients:
Update fog model:
Aggregate fog-level models at cloud:
Broadcast to all nodes
End if
End for
θ
|