p2pfl.learning.aggregators.fedyogi module

FedYogi Aggregator - Adaptive Federated Optimization using Yogi.

class p2pfl.learning.aggregators.fedyogi.FedYogi(eta=0.01, eta_l=0.0316, beta_1=0.9, beta_2=0.99, tau=0.001, disable_partial_aggregation=False)

Bases: Aggregator

FedYogi - Adaptive Federated Optimization using Yogi [Reddi et al., 2020].

FedYogi adapts the Yogi optimizer to federated settings, maintaining adaptive learning rates on the server side to handle heterogeneous data distributions.

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

SUPPORTS_PARTIAL_AGGREGATION: bool = False

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

Return type:

Any