Flower: A Friendly Federated Learning Research Framework

Abstract

Federated Learning (FL) has emerged as a promising technique for edge devices to collaboratively learn a shared prediction model, while keeping their training data on the device, thereby decoupling the ability to do machine learning from the need to store the data in the cloud. However, FL is difficult to implement realistically, both in terms of scale and systems heterogeneity. Although there are a number of research frameworks available to simulate FL algorithms, they do not support the study of scalable FL workloads on heterogeneous edge devices. In this paper, we present Flower -- a comprehensive FL framework that distinguishes itself from existing platforms by offering new facilities to execute large-scale FL experiments and consider richly heterogeneous FL device scenarios. Our experiments show Flower can perform FL experiments up to 15M in client size using only a pair of high-end GPUs. Researchers can then seamlessly migrate experiments to real devices to examine other parts of the design space. We believe Flower provides the community with a critical new tool for FL study and development.

Open-Source, mobile-friendly Federated Learning framework

References in corpus (4)

• Fashion-MNIST: a Novel Image Dataset for Benchmarking Machine Learning Algorithms
• Towards Federated Learning at Scale: System Design
• Deep ConvLSTM with self-attention for human activity decoding using wearables
• Visual Wake Words Dataset

Cited by in corpus (19)

• From Distributed Machine Learning to Federated Learning: A Survey
• FedBN: Federated Learning on Non-IID Features via Local Batch Normalization
• A Field Guide to Federated Optimization
• Artificial Intelligence of Things: A Survey
• FjORD: Fair and Accurate Federated Learning under heterogeneous targets with Ordered Dropout
• FedLab: A Flexible Federated Learning Framework
• A first look into the carbon footprint of federated learning
• Federated Learning for Big Data: A Survey on Opportunities, Applications, and Future Directions
• A Survey on Federated Learning and its Applications for Accelerating Industrial Internet of Things
• Federated Split Vision Transformer for COVID-19 CXR Diagnosis using Task-Agnostic Training
• Perfectly Accurate Membership Inference by a Dishonest Central Server in Federated Learning
• Federated Fine-Tuning of LLMs on the Very Edge: The Good, the Bad, the Ugly
• Federated Learning -- Methods, Applications and beyond
• HyFed: A Hybrid Federated Framework for Privacy-preserving Machine Learning
• FedCLF -- Towards Efficient Participant Selection for Federated Learning in Heterogeneous IoV Networks
• Syft 0.5: A Platform for Universally Deployable Structured Transparency
• On-device Federated Learning with Flower
• Can Federated Learning Save The Planet?
• Towards General-purpose Infrastructure for Protecting Scientific Data Under Study