p2pfl.learning.aggregators.fedprox module

FedProx Aggregator - Federated Proximal.

class p2pfl.learning.aggregators.fedprox.FedProx(proximal_mu=0.01, disable_partial_aggregation=False)

Bases: FedAvg

FedProx - Federated Proximal [Li et al., 2018].

FedProx extends FedAvg by adding a proximal term to the local objective function to handle system and statistical heterogeneity.

Paper: https://arxiv.org/abs/1812.06127

aggregate(*args: Any, **kwargs: Any) → Any

Return type:

Any

get_required_callbacks(*args: Any, **kwargs: Any) → Any

Return type:

Any