weimingwill / EasyFL

An easy-to-use federated learning platform
Apache License 2.0
293 stars 50 forks source link
computer-vision data-privacy deep-learning federated-learning machine-learning

EasyFL: A Low-code Federated Learning Platform

[![PyPI](https://img.shields.io/pypi/v/easyfl)](https://pypi.org/project/easyfl) [![docs](https://img.shields.io/badge/docs-latest-blue)](https://easyfl.readthedocs.io/en/latest/) [![license](https://img.shields.io/github/license/easyfl-ai/easyfl.svg)](https://github.com/easyfl-ai/easyfl/blob/master/LICENSE) [![maintained](https://img.shields.io/badge/Maintained%3F-YES-yellow.svg)](https://github.com/easyfl-ai/easyfl/graphs/commit-activity) [![Downloads](https://pepy.tech/badge/easyfl)](https://pepy.tech/project/easyfl) [📘 Documentation](https://easyfl.readthedocs.io/en/latest/) | [🛠️ Installation](https://easyfl.readthedocs.io/en/latest/get_started.html)

Introduction

EasyFL is an easy-to-use federated learning (FL) platform based on PyTorch. It aims to enable users with various levels of expertise to experiment and prototype FL applications with little/no coding.

You can use it for:

We currently focus on horizontal FL, supporting both cross-silo and cross-device FL. You can learn more about federated learning from these resources.

Major Features

Easy to Start

EasyFL is easy to install and easy to learn. It does not have complex dependency requirements. You can run EasyFL on your personal computer with only three lines of code (Quick Start).

Out-of-the-box Functionalities

EasyFL provides many out-of-the-box functionalities, including datasets, models, and FL algorithms. With simple configurations, you simulate different FL scenarios using the popular datasets. We support both statistical heterogeneity simulation and system heterogeneity simulation.

Flexible, Customizable, and Reproducible

EasyFL is flexible to be customized according to your needs. You can easily migrate existing CV or NLP applications into the federated manner by writing the PyTorch codes that you are most familiar with.

Multiple Training Modes

EasyFL supports standalone training, distributed training, and remote training. By developing the code once, you can easily speed up FL training with distributed training on multiple GPUs. Besides, you can even deploy it to Kubernetes with Docker using remote training.

Getting Started

You can refer to Get Started for installation and Quick Run for the simplest way of using EasyFL.

For more advanced usage, we provide a list of tutorials on:

Projects & Papers

We have released the source code for the following papers under the applications folder:

:bulb: We will release the source codes of these projects in this repository. Please stay tuned.

We have been doing research on federated learning for several years, the following are our additional publications.

Join Our Community

Please join our community on Slack: easyfl.slack.com

We will post updated features and answer questions on Slack.

License

This project is released under the Apache 2.0 license.

Citation

If you use this platform or related projects in your research, please cite this project.

@article{zhuang2022easyfl,
  title={Easyfl: A low-code federated learning platform for dummies},
  author={Zhuang, Weiming and Gan, Xin and Wen, Yonggang and Zhang, Shuai},
  journal={IEEE Internet of Things Journal},
  year={2022},
  publisher={IEEE}
}

Main Contributors

Weiming Zhuang :octocat:
Xin Gan :octocat: