FedProto
¶
Implementation of the FedProto [FedProto22] algorithm. .. rubric:: References
[FedProto22]
Yue Tan, Guodong Long, Lu Liu, Tianyi Zhou, Qinghua Lu, Jing Jiang, Chengqi Zhang. FedProto: Federated Prototype Learning across Heterogeneous Clients. In AAAI (2022). URL: https://arxiv.org/abs/2105.00243
Classes included in fluke.algorithms.fedproto
Classes¶
- class fluke.algorithms.fedproto.FedProtoModel(model: EncoderHeadNet, prototypes: dict[int, Tensor], device: device)[source]¶
Bases:
Module
- class fluke.algorithms.fedproto.FedProtoClient(index: int, model: Module, train_set: FastDataLoader, test_set: FastDataLoader, optimizer_cfg: OptimizerConfigurator, loss_fn: Module, local_epochs: int, n_protos: int, lam: float, fine_tuning_epochs: int = 0, clipping: float = 0, **kwargs: dict[str, Any])[source]¶
Bases:
Client