MOON

Implementation of the [Moon21] algorithm.

References

[Moon21]

Qinbin Li, Bingsheng He, and Dawn Song. Model-Contrastive Federated Learning. In CVPR (2021). URL: https://arxiv.org/abs/2103.16257

Classes included in fluke.algorithms.moon

MOONClient
MOON

Classes

class fluke.algorithms.moon.MOONClient(index: int, train_set: FastDataLoader, test_set: FastDataLoader, optimizer_cfg: OptimizerConfigurator, loss_fn: Module, local_epochs: int, mu: float, tau: float, fine_tuning_epochs: int = 0, **kwargs: dict[str, Any])

Bases: Client

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

Bases: CentralizedFL