Expand description
Federated learning utilities.
This module provides:
- Differential privacy mechanisms (
PrivacyBudget,DPMechanism,DifferentialPrivacy) - Secure aggregation (
SecureAggregation) - Client lifecycle management (
ClientState,ClientInfo) - Round coordination (
FederatedRound,ConvergenceDetector,ModelSyncProtocol) - Standalone helpers:
federated_average,clip_gradient_norm DistributedGradientAccumulatorfor peer-to-peer gradient exchange
Structs§
- Client
Info - Client information in federated learning
- Convergence
Config - Configuration for
ConvergenceDetector. - Convergence
Detector - Convergence detection for federated learning.
- Differential
Privacy - Differential privacy for gradient protection
- Distributed
Gradient Accumulator - Accumulates gradients from distributed peers via content-addressed storage.
- Federated
Round - Federated learning round.
- Gossip
Model Sync - GossipSub-based model synchronisation protocol.
- Model
Sync Protocol - Model synchronization protocol for federated learning
- Model
Update - A model-update announcement broadcast via GossipSub.
- Privacy
Budget - Privacy budget for differential privacy
- Round
Stats - Statistics snapshot for a completed federated round.
- Secure
Aggregation - Secure aggregation for federated learning (simplified)
Enums§
- Client
State - Client state in federated learning
- DPMechanism
- Differential privacy mechanism types
- Federated
Error - Errors specific to the federated round lifecycle.