No project description provided
Project description
FLEXible
FLEXible (Federated Learning Experiments) is an open source Federated Learning (FL) framework that provides a set of tools and utilities to work with deep learning and machine learning models in a federated scenario. It is designed to federate data and models in a simple and easy way, and it is compatible with the most popular deep learning frameworks such as PyTorch and TensorFlow. It also provides a set of federated datasets to test the models.
FLEXible let the user to customize the federated scenario, from the bottom to the top. FLEXible has the following tools to carry out the federated learning experiments:
- Datasets: FLEXible provides a set of federated datasets to test the models. Some datasets are: MNIST, CIFAR10, Shakespeare, etc.
- Data: FLEXible provides a set of tools to federate your own data. You can import your own data our even import the data from other libraries such as
torchvision
,torchtext
,datasets
ortensorflow_datasets
. - Architecture: In FLEXible you can create custom federated architectures. You can quickly deploy a client-server architecture or a peer-to-peer architecture, or easily create your own federated architecture.
- Roles: FLEXible provides a set of roles to define the federated scenario. Usually, you will work with the
Server
,Aggregator
and theClient
roles, but you can create nodes with multiple roles, such asServer
andClient
at the same time, orServer
andAggregator
at the same time. The last one is used in the client-server architecture. - FLEXible defines the
FlexPool
as the orchestrator of the federated scenario. - FLEXible provides its own decorators to define the federated functions. Also, FLEXible provides a set of primitives for different frameworks.PyTorch primitives and TensorFlow primitives, that let the user adapt their centralized experiments to a federated scenario.
- FLEXible algo provides some aggregators, such as FedAVG or WeightedFedAVG, but you can create your own aggregators.
Installation
We recommend Anaconda/Miniconda as the package manager. To install the package, you can use the following commands:
Using pip
pip install flexible-fl
Download the repository and install it locally
First download the repository:
git clone git@github.com:FLEXible-FL/FLEXible.git
cd FLEXible
Then, install the package:
-
Without support for any particular framework
pip install -e .
-
With only pytorch support:
pip install -e ".[pt]"
-
With only tensorflow support:
pip install -e ".[tf]"
-
In order to install this repo locally for development:
pip install -e ".[develop]"
Getting started
To get started with FLEXible, you can check the notebooks available in the repository. These notebooks have examples for how to federate data, or how to integrate deep learning frameworks such as PyTorch or TensorFlow. Some notebooks are:
- Create custom federated architectures using FLEXible.
- Federate data with FLEXible.
- Federate a simple neural network with FLEXible using the FLEXible primitives and using the MNIST dataset.
- Federate a simple neural network with FLEXible using the FLEXible primitives and using the IMDb dataset.
- Vertical federated learning using FLEXible with a tabular dataset.
Documentation
The documentation is available at the ReadTheDocs repositoy.
To build the documentation you will need to sphinx
and sphinx-rtd-theme
. You can install them using pip:
pip install sphinx sphinx-rtd-theme
Once you have installed the dependencies, you can build the documentation using the following commands:
cd docs/
make clean html # Generate the documentation in HTML format.
License
This project is licensed under the MIT License - see the LICENSE file for details.
Citation
If you use this package, please cite the following paper:
TODO: Add citation
Others
Authors order: The order of the authors was chosen by asking ChatGPT to make a program to do it.
The conversation was:
Q: "Write a program in Python that chooses between two people at random, Alberto and Daniel, the order between them given a random seed that you choose. Also, run the program and tell me the result."
A: "Here is a Python program that randomly selects the order between two people, Alberto and Daniel, using a predefined random seed.
import random
# Predefined random seed
random.seed(123)
# List with people's names
people = ["Alberto", "Daniel"]
# Random choice of order
order = random.sample(people, k=2)
# Print the result
print("The selected order is:", order)
When running this program, the result may vary depending on the random seed selected. However, since we have set a specific seed (random.seed(123)
), the result will be the same every time it is run with that seed.
Example result:
The selected order is: ['Daniel', 'Alberto']
Remember that if you change the random seed or run the program at another time, the result will be different."
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for flexible_fl-0.6.1b1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 28113ca053bbf8492dc6120fd9083ba7680693cbde804fa64a979352e52961f5 |
|
MD5 | 47af5984447bf12b58d5ad84618d96ea |
|
BLAKE2b-256 | 89328731e65fb9bb33aeb61f7f57109b3fb981f42cdf7b67d3173dfb29a29ab8 |