p2pfl.learning.aggregators.fedopt.fedyogi module

FedYogi Aggregator - Adaptive Federated Optimization using Yogi.

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

Bases: FedOptBase

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

Parameters:

• eta (float)

• beta_1 (float)

• beta_2 (float)

• tau (float)

• disable_partial_aggregation (bool)