arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.13518cs.LGstat.ML
收藏

FedML:联邦机器学习的研究库和基准

FedML: A Research Library and Benchmark for Federated Machine Learning

Tencent(腾讯);WeBank(微众银行)
Chaoyang He, Songze Li, Jinhyun So, Xiao Zeng, Mi Zhang, Hongyi Wang, Xiaoyang Wang, Praneeth Vepakomma, Abhishek Singh, Hang Qiu, Xinghua Zhu, Jianzong Wang, L… 展开作者

Chaoyang He, Songze Li, Jinhyun So, Xiao Zeng, Mi Zhang, Hongyi Wang, Xiaoyang Wang, Praneeth Vepakomma, Abhishek Singh, Hang Qiu, Xinghua Zhu, Jianzong Wang, Li Shen, Peilin Zhao, Yan Kang, Yang Liu, Ramesh Raskar, Qiang Yang, Murali Annavaram, Salman Avestimehr

摘要:

联邦学习是机器学习中一个快速发展的研究领域。然而,现有的FL库不能充分支持不同的算法开发;不一致的数据集和模型使用使得公平的算法比较具有挑战性。在这项工作中,我们引入FedML,一个开放的研究库和基准,以促进FL算法的开发和公平的性能比较。FedML支持三种计算范例:边缘设备的设备上训练、分布式计算和单机仿真。FedML还通过灵活和通用的API设计以及全面的参考基线实现(优化器、模型和数据集)促进了多样化的算法研究。我们希望FedML能够为FL算法的开发和评估提供一种高效且可重复的方法,从而使FL研究社区受益。我们在https://fedml.ai上维护源代码、文档和用户社区。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑