p2pfl.learning.aggregators.scaffold module¶
Callback for SCAFFOLD operations.
- class p2pfl.learning.aggregators.scaffold.Scaffold(global_lr=0.1, disable_partial_aggregation=False)[source]¶
Bases:
AggregatorSCAFFOLD Aggregator.
Paper: https://arxiv.org/pdf/1910.06378 The aggregator acts like the server in centralized learning, handling both model and control variate updates.
Due to the complete decentralization of the enviroment, a global model is also maintained in the aggregator. This consumes additional bandwidth.
::todo:: Improve efficiency by sharing the global model only each n rounds.
- REQUIRED_INFO_KEYS = ['delta_y_i', 'delta_c_i']¶
-
SUPPORTS_PARTIAL_AGGREGATION:
bool= False¶
- aggregate(*args: Any, **kwargs: Any) Any¶
- Return type:
Any
- get_required_callbacks(*args: Any, **kwargs: Any) Any¶
- Return type:
Any