LG-FedAVG

Implementation of the [LG-FedAVG20] algorithm.

References

[LG-FedAVG20]

Paul Pu Liang, Terrance Liu, Liu Ziyin, Nicholas B. Allen, Randy P. Auerbach, David Brent, Ruslan Salakhutdinov, Louis-Philippe Morency. Think Locally, Act Globally: Federated Learning with Local and Global Representations. In arXiv (2020). URL: https://arxiv.org/abs/2001.01523

Classes included in fluke.algorithms.lg_fedavg

LGFedAVGClient

LGFedAVGServer

LGFedAVG

Classes

class fluke.algorithms.lg_fedavg.LGFedAVGClient(index: int, model: EncoderHeadNet, train_set: FastDataLoader, test_set: FastDataLoader, optimizer_cfg: OptimizerConfigurator, loss_fn: Module, local_epochs: int = 3, fine_tuning_epochs: int = 0, **kwargs)[source]

Bases: Client

class fluke.algorithms.lg_fedavg.LGFedAVGServer(model: Module, test_set: FastDataLoader, clients: Collection[Client], weighted: bool = False)[source]

Bases: Server

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

Bases: PersonalizedFL