An open API service indexing awesome lists of open source software.

https://github.com/sattviksahai/collaborative-learning-simulator

A python library to simulate Collaborative Deep Learning. It provides the flexibility to simulate various network architectures, Collaborative Learning Stratergies, and privacy invasion attacks. Currently built for Pytorch, this library is a work in progress.
https://github.com/sattviksahai/collaborative-learning-simulator

collaborative-learning deep-learning federated-learning pytorch-implementation simulator

Last synced: 21 days ago
JSON representation

A python library to simulate Collaborative Deep Learning. It provides the flexibility to simulate various network architectures, Collaborative Learning Stratergies, and privacy invasion attacks. Currently built for Pytorch, this library is a work in progress.

Awesome Lists containing this project

README

          

# Collaborative-Learning-Simulator

A python library to simulate Collaborative Deep Learning. It provides the flexibility to simulate various network architectures, Collaborative Learning Stratergies, and privacy invasion attacks. Currently built for Pytorch, this library is a work in progress.

## Pre-requisites
It is recommended to install the simulator in a virtual environment, and install the dependencies in this environment. In a Linux system, this can be done using terminal as follows:
```bash
python3 -m pip install --user virtualenv
python3 -m venv my_env
source my_env/bin/activate
pip install -r requirements.txt
```

## Installation
```bash
pip install cl_simulator
```

## Example Implementations
The repository contains two example implementation.
### Federated Averaging:
Implemented in the "federated_averaging.ipynb" file, this simulates a two part setting where we train an image based gender classifier.
### Reference Script:
Implemented in the "reference_training_script.ipynb" file, this demonstrates training a gender classifer on a single system.

## License

MIT