Coder Social home page Coder Social logo

drenches / federated-learning-research Goto Github PK

View Code? Open in Web Editor NEW

This project forked from ysyisyourbrother/federated-learning-research

0.0 0.0 0.0 44 KB

An implementation of federated learning research baseline methods based on FedML-core, which can be deployed on real distributed cluster and help researchers to explore more problems existing in real FL systems.

Shell 2.62% Python 97.38%

federated-learning-research's Introduction

Federated-Learning-Research

An implementation of Federated Learning research baseline methods based on FedML-core. This is not an implementation for only stand-alone simulation, but a distributed system that can be deployed on multiple real devices (or several docker containers on a same server), which can help researchers to explore more problems that may exist on real FL systems.

Here are the list of publications based on this repository:

  • #ICPP 2022# Our work "Eco-FL: Adaptive Federated Learning with Efficient Edge Collaborative Pipeline Training" has been accepted by International Conference on Parallel Processing(ICPP), 2022.

Quick Start

Here is demo to deploy our framework on a cluster with three devices. A (ip: 172.17.0.4) represents the server worker, B (ip: 172.17.0.13) and C (ip: 172.17.0.12) represent two client workers in FL system. All devices must have a python3 environment with pytorch and grpc installed. You can find the env-requirement under requirement.txt file.

Startup scripts for all methods are under experiment/ directory. We first need to modify the grpc_ipconfig.csv file as below:

receiver_id,ip
0,172.17.0.4
1,172.17.0.13
2,172.17.0.12

Receiver_id represents the worker_id for each worker in FL system. Usually, worker_id of server worker is 0 and client workers' id will start from 1.

After that, we will start the worker process on each devices. All client workers need to be started up before server worker. Remember to execute all commands below at the root direcroy of this project:

python experiment/fedprox/run_fedprox_distributed.sh $worker_id

Then the training process will begin and the checkpoint of global model will be saved under checkpoint/fedprox/. You can change the experiment settings by editing the pre-set arguments in run_fedprox_distributed.sh.

Reproduced FL Algorithms

Method Reference
FedAvg McMahan et al., 2017
FedProx Li et al., 2020
FedAsync Wang et al., 2021
FedAT Zheng Chai et al., 2021
On going ...

Contacts

federated-learning-research's People

Contributors

ysyisyourbrother avatar acomze avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.