FedLab: A Flexible Federated Learning Framework
arXiv:2107.11621
Abstract
Federated learning (FL) is a machine learning field in which researchers try to facilitate model learning process among multiparty without violating privacy protection regulations. Considerable effort has been invested in FL optimization and communication related researches. In this work, we introduce \texttt{FedLab}, a lightweight open-source framework for FL simulation. The design of \texttt{FedLab} focuses on FL algorithm effectiveness and communication efficiency. Also, \texttt{FedLab} is scalable in different deployment scenario. We hope \texttt{FedLab} could provide flexible API as well as reliable baseline implementations, and relieve the burden of implementing novel approaches for researchers in FL community.
References in corpus (15)
- PyTorch: An Imperative Style, High-Performance Deep Learning Library
- Deep Gradient Compression: Reducing the Communication Bandwidth for Distributed Training
- On the Convergence of FedAvg on Non-IID Data
- QSGD: Communication-Efficient SGD via Gradient Quantization and Encoding
- How To Backdoor Federated Learning
- Improving Federated Learning Personalization via Model Agnostic Meta Learning
- FedML: A Research Library and Benchmark for Federated Machine Learning
- A generic framework for privacy preserving deep learning
- LEAF: A Benchmark for Federated Settings
- Sparsified SGD with Memory
- Federated Learning Based on Dynamic Regularization
- Flower: A Friendly Federated Learning Research Framework
- 8-Bit Approximations for Parallelism in Deep Learning
- Federated Learning with Matched Averaging
- Federated Learning on Non-IID Data Silos: An Experimental Study