FedOpt

Implementation of the [FedOpt21] algorithm.

References

[FedOpt21]

Sashank Reddi, Zachary Charles, Manzil Zaheer, Zachary Garrett, Keith Rush, Jakub Konečný, Sanjiv Kumar, H. Brendan McMahan. Adaptive Federated Optimization. In ICLR (2021). URL: https://openreview.net/pdf?id=LkFG3lB13U5

Classes included in fluke.algorithms.fedopt

FedOptServer
FedOpt

Classes

class fluke.algorithms.fedopt.FedOptServer(model: Module, test_set: FastDataLoader, clients: Collection[Client], mode: str = 'adam', lr: float = 0.001, beta1: float = 0.9, beta2: float = 0.999, tau: float = 0.0001, weighted: bool = True)

Bases: Server

class fluke.algorithms.fedopt.FedOpt(n_clients: int, data_splitter: DataSplitter, hyper_params: DDict | dict[str, Any], **kwargs)

Bases: CentralizedFL