FedML: A Research Library and Benchmark for Federated Machine Learning

Chaoyang He, Songze Li, Jinhyun So, Xiao Zeng, Mi Zhang +15 more
7/27/2020
cs.LGstat.ML

Official code: chaoyanghe/Awesome-Federated-Learning (2,023 stars, last commit 2022-09-03).

Abstract

Federated learning (FL) is a rapidly growing research field in machine learning. However, existing FL libraries cannot adequately support diverse algorithmic development; inconsistent dataset and model usage make fair algorithm comparison challenging. In this work, we introduce FedML, an open research library and benchmark to facilitate FL algorithm development and fair performance comparison. FedML supports three computing paradigms: on-device training for edge devices, distributed computing, and single-machine simulation. FedML also promotes diverse algorithmic research with flexible and generic API design and comprehensive reference baseline implementations (optimizer, models, and datasets). We hope FedML could provide an efficient and reproducible means for developing and evaluating FL algorithms that would benefit the FL research community. We maintain the source code, documents, and user community at https://fedml.ai.

View on arXivView PDF

Code Implementations(1)

chaoyanghe/Awesome-Federated-LearningOfficial100%

FedML - The Research and Production Integrated Federated Learning Library: https://fedml.ai

2,023333Jul 3, 20204 years ago
adversarial-attack-and-defensecommunication-efficiencycomputation-efficiencycomputer-visioncontinual-learning+15 more

Similar Papers

DataPerf: Benchmarks for Data-Centric AI Development

86% match1 repo252022

MLPerf Training Benchmark

84% match1 repo1.8k2019

AutoGluon-Tabular: Robust and Accurate AutoML for Structured Data

83% match1 repo222020

Procedural Generation of Algorithm Discovery Tasks in Machine Learning

83% match1 repo502026

MOABB: Trustworthy algorithm benchmarking for BCIs

83% match1 repo1.0k2018

SLIDE : In Defense of Smart Algorithms over Hardware Acceleration for Large-Scale Deep Learning Systems

83% match1 repo1.1k2019

Cite this paper

@article{he2020fedml,
  title  = {FedML: A Research Library and Benchmark for Federated Machine Learning},
  author = {Chaoyang He and Songze Li and Jinhyun So and Xiao Zeng and Mi Zhang and Hongyi Wang and Xiaoyang Wang and Praneeth Vepakomma and Abhishek Singh and Hang Qiu and Xinghua Zhu and Jianzong Wang and Li Shen and Peilin Zhao and Yan Kang and Yang Liu and Ramesh Raskar and Qiang Yang and Murali Annavaram and Salman Avestimehr},
  year   = {2020},
  eprint = {2007.13518},
  archivePrefix = {arXiv},
  url    = {https://arxiv.org/abs/2007.13518}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2007.13518.svg)](https://codeofpaper.com/paper/2007.13518)

Discussion