Byzantine-Robust Aggregation in Federated Learning Empowered Industrial IoT

This repository contains the code and experiments for the paper:

Byzantine-Robust Aggregation in Federated Learning Empowered Industrial IoT

Datasets

1. Bosch

• Overview: Image Dataset. See kaggle
• Details: 2 different classes.
• Task: Binary Classification

2. FEMNIST

• Overview: Image Dataset
• Details: 62 different classes (10 digits, 26 lowercase, 26 uppercase), images are 28 by 28 pixels (with option to make them all 128 by 128 pixels), 3500 users
• Task: Image Classification

Notes

• Install the libraries listed in requirements.txt
  • I.e. with pip: run pip3 install -r requirements.txt

Ref

LEAF benchmark

• Homepage: leaf.cmu.edu
• Paper: "LEAF: A Benchmark for Federated Settings"