FLBench: A Benchmark Suite for Federated Learning

Abstract

Federated learning is a new machine learning paradigm. The goal is to build a machine learning model from the data sets distributed on multiple devices so-called an isolated data island, while keeping their data secure and private. Most existing federated learning benchmarks work manually splits commonly used public datasets into partitions to simulate real world isolated data island scenarios. Still, this simulation fails to capture real world isolated data island intrinsic characteristics. This paper presents a federated learning (FL) benchmark suite named FLBench. FLBench contains three domains: medical, financial, and AIoT. By configuring various domains, FLBench is qualified to evaluate federated learning systems and algorithms essential aspects, like communication, scenario transformation, privacy-preserving, data distribution heterogeneity, and cooperation strategy. Hence, it becomes a promising platform for developing novel federated learning algorithms. Currently, FLBench is open sourced and in fast evolution. We package it as an automated deployment tool. The benchmark suite is available from https://www.benchcouncil.org/flbench.html.

arXiv comment12 pages, 1 figures

References in corpus (10)

• Federated Learning: Strategies for Improving Communication Efficiency
• Federated Optimization: Distributed Machine Learning for On-Device Intelligence
• How To Backdoor Federated Learning
• FedML: A Research Library and Benchmark for Federated Machine Learning
• Agnostic Federated Learning
• Federated Learning with Bayesian Differential Privacy
• Bayesian Nonparametric Federated Learning of Neural Networks
• Federated Learning with Matched Averaging
• Real-World Image Datasets for Federated Learning
• The OARF Benchmark Suite: Characterization and Implications for Federated Learning Systems