/Communication-Efficient-FL-In-IoT

Primary LanguagePython

Communication Efficient Federated Learning

This repository contains the code to run simulations from the Communication-Efficient Federated Learning for Wireless Edge Intelligence in IoT paper in IEEE IoT journal.

Requirements

  • python = 3.7
  • tensorflow = 2.1.0
  • numpy = 1.17
  • bitarray = 1.2.1

Running

To run a non-IID MNIST CE-FedAvg simulation with 10 workers, C = 0.5, E = 1, sparsity = 0.6 using the MNIST-2NN model, run

python main.py

The main.py file is easily edited to run experiments with the CIFAR dataset and different FL parameters.