FedBN

Implementation of the FedBN [FedBN21] algorithm.

References

[FedBN21]

Xiaoxiao Li, Meirui JIANG, Xiaofei Zhang, Michael Kamp, and Qi Dou. FedBN: Federated Learning on Non-IID Features via Local Batch Normalization. In ICLR (2021). URL: https://openreview.net/pdf?id=6YEQUn0QICG

Classes included in fluke.algorithms.fedbn

FedBNClient
FedBN

Classes

class fluke.algorithms.fedbn.FedBNClient(index: int, train_set: FastDataLoader, test_set: FastDataLoader, optimizer_cfg: OptimizerConfigurator, loss_fn: Module, local_epochs: int = 3, fine_tuning_epochs: int = 0, **kwargs: dict[str, Any])

Bases: Client

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

Bases: CentralizedFL