Fed-BioMed is an open source project focused on empowering biomedical research using non-centralized approaches for statistical analysis and machine learning.
The project is currently based on Python, PyTorch and Scikit-learn, and enables developing and deploying collaborative learning analysis in real-world machine learning applications, including federated learning and federated analytics.
The code is regularly released and available on the master branch of this repository. The documentation of the releases can be found at https://fedbiomed.org
Curious users may also be interested by the current developments, occurring in the develop branch (https://github.com/fedbiomed/fedbiomed/tree/develop) According to our coding rules, the develop branch is usable, tests and tutorials will run, but the documentation may be not fully available or desynchronizing with the code. We only provide support for the last release aka the master branch.
Fed-BioMed is developped under Linux Fedora & Ubuntu, should be easily ported to other Linux distributions. It runs also smoothly on macOSX, and in Windows WSL2.
This README.md file provide a quick start/installation guide for Linux.
Full installation instruction are also available at: https://fedbiomed.org/latest/tutorials/installation/0-basic-software-installation/
An installation guide is also provided for Windows11, which relies on WSL2: https://fedbiomed.org/latest/user-guide/installation/windows-installation/
To ensure fedbiomed will work fine, you need to install before :
- conda
- git
docker is not needed anymore for using development environment, unless you use secure aggregation with MP-SPDZ for multi party computation and plan to rebuild the Shamir protocol binary
Clone the Fed-BioMed repository for running the software :
git clone -b master https://github.com/fedbiomed/fedbiomed.git
Fed-BioMed developers clone of the repository :
git clone [email protected]:fedbiomed/fedbiomed.git
- to create or update the environments, you can use the configure_conda script:
$ ./scripts/configure_conda
-
this script will create/update the conda environments
-
there is one specific environment for each component:
- fedbiomed-node.yaml : environment for the node part
- fedbiomed-researcher.yaml : environment for the researcher part
- fedbiomed-gui.yaml : environment for the data management gui on the node
Remark:
- this script can also be used to update only some of the environments
- for some components, we provide different versions of yaml files depending of the operating system of your host
- in case of (conda or python) errors, we advice to remove all environments and restart from fresh (use the -c flag of configure_conda)
- general usage for this script is:
Usage: configure_conda [-n] [-c] [-t] [ENV ENV ..]
Install/update conda environments for fedbiomed. If several ENV
are provided, only these components will be updated. If no ENV is
provided, all components will be updated.
ENV can be node, researcher, gui (or a combination of them)
-h, --help this help
-n, --dry-run do nothing, just print what the script would do
-c, --clean remove environment before reinstallating it
-t, --test test the environment at the end of installation
(this only tests the researcher environment for now)
In a terminal, you can configure environments to work interactively inside a specific repository, with the right conda environment and the right PYTHONPATH environment.
WARNING: this script only work for bash, ksh and zsh. It is not compliant with c variant of shell (csh/tcsh/etcsh/...)
source ./scripts/fedbiomed_environment ENV
where ENV chosen from:
- node
- researcher
- gui
- in a new terminal:
$ ./scripts/fedbiomed_run node start
-
this will launch a new node
-
you may also upload new data on this node with:
$ ./scripts/fedbiomed_run node dataset add
- you may also specify a new config file for the node (usefull then running multiple test nodes on the same host)
$ ./scripts/fedbiomed_run node --config another_config.ini start
- if you want to change the default IP address used to join the fedbiomed researcher component (localhost), you can provide it at launch time:
$ RESEARCHER_SERVER_HOST=192.168.0.100 ./scripts/fedbiomed_run node start
$ RESEARCHER_SERVER_HOST=192.168.0.100./scripts/fedbiomed_run researcher start
(adjust the 192.168.0.100 IP address to your configuration)
If this option is given at the first launch or after a clean, it is saved in the configuration file and becomes the default for subsequent launches. If this option is given at a subsequent launch, it only affects this launch.
- in a new terminal:
$ ./scripts/fedbiomed_run researcher start
-
this will launch a new jupyter notebook working in the notebooks repository. Some notebooks are available:
101_getting-started.ipynb: training a simplenet + federated average on MNIST datapytorch-local-training.ipynb: comparing the simplenet + federated average on MNIST data with its local training equivalent
- in a new terminal:
$ source ./scripts/fedbiomed_environment researcher- convert the notebook to a python script
jupyter nbconvert --output=101_getting-started --to script ./notebooks/101_getting-started.ipynb- then you can use any researcher script
$ python ./notebooks/101_getting-started.pyBy default, fedbiomed-node contacts fedbiomed-researcher on localhost.
To configure your current shell to use another IP address for joining fedbiomed-researcher (e.g. 192.168.0.100):
source ./scripts/fedbiomed_environment node 192.168.0.100
source ./scripts/fedbiomed_environment researcher 192.168.0.100Then launch the components with usual commands while you are in the current shell.
Warning: this option does not modify the existing configuration file (.ini file).
De-configure environments, remove all configuration files and caches
source ./scripts/fedbiomed_environment clean
To use the docker + VPN mode you need to install:
- docker
- docker compose v2 (aka docker compose plugin)
The envs/vpn directory contains all material for VPN support. A full technical description is provided in envs/vpn/README.md
The ./scripts/fedbiomed_vpn script is provided to ease the deployment of a set of docker container(s) with VPN support. The provided containers are:
- fedbiomed/vpn-vpnserver: WireGuard server
- fedbiomed/vpn-researcher: a researcher jupyter notebooks
- fedbiomed/vpn-node: a node component
- fedbiomed/vpn-gui: a GUI for managing node component data
All these containers are communicating through the Wireguard VPN server.
To setup all these components, you should:
- clean all containers and files
./scripts/fedbiomed_vpn clean
- build all the docker containers
./scripts/fedbiomed_vpn build
- configure the wireguard encryption keys of all containers
./scripts/fedbiomed_vpn configure
- start the containers
./scripts/fedbiomed_vpn start
- check the containers status (presence and Wireguard configuration)
./scripts/fedbiomed_vpn status
- run a fedbiomed_run command inside the node component. Eg:
./scripts/fedbiomed_vpn node dataset add --mnist /data
./scripts/fedbiomed_vpn node list
./scripts/fedbiomed_vpn node start
-
connect to the researcher jupyter at http://127.0.0.1:8888 (Remark: the researcher* docker automatically starts a jupyter notebook inside the container)
-
manage data inside the node with the node GUI at http://127.0.0.1:8484
-
stop the containers:
./scripts/fedbiomed_vpn stop
You can manage individually the containers for the build/stop/start phases, by passing the name of the container(s) on the command line.
For example, to build only the node, you can use:
./scripts/fedbiomed_vpn build node
You can build/configure/stop/start/check more than one component at a time. Example:
./scripts/fedbiomed_vpn build gui node
This will stop and build the node container.
The list of the container names is:
- vpnserver
- researcher
- node
- gui
Remarks:
- the configuration files are keeped then rebuilding individual containers
- to remove the old config files, you should do a clean
- restarting only vpnserver when others are running may lead to unpredictable behavior. In this case, it is adviced to restart from scratch (clean/build/configure/start)
list the content of a message queue (as used in fedbiomed.node and fedbiomed.researcher)
usage: lqueue directory or lqueue dir1 dir2 dir3 ...
Run a full (end to end) test by launching:
- a researcher (running a python script or a notebook script)
- several nodes, providing data
Usefully for continuous integration tests and notebook debugging. Full documentation in tests/README.md file.
Required python modules should be installed to be able to build or serve the documentation page. These packages can be installed using conda environment to serve or build the documentation (recommended).
conda env update -f envs/build/conda/fedbiomed-doc.yaml
conda activate fedbiomed-doc
They can also be installed using pip (required python version 3.11), as in the real build process (if you know what you're doing).
- Warning: if not using a
condaorpipvirtual environment, your global settings are modified.
pip install -r envs/development/docs-requirements.txt
Please use following command to serve documentation page. This will allow you to test/verify changes in docs and also in doc-strings.
cd ${FEDBIOMED_DIR}
./scripts/docs/fedbiomed_doc.sh servePlease see usage for additional options.
cd ${FEDBIOMED_DIR}
./scripts/docs/fedbiomed_doc.sh --help
To enable tensorboard during training routine to see loss values, you need to set tensorboard parameter to True while initializing Experiment class.
exp = Experiment(tags=tags,
#nodes=None,
model_args=model_args,
training_plan_class=MyTrainingPlan,
training_args=training_args,
round_limit=round_limit,
aggregator=FedAverage(),
node_selection_strategy=None,
tensorboard=True
)
Or after initialization :
exp.set_tensorboard(True)
During training, the scalar values (loss) will be writen in the runs directory. You can either start tensorboard from jupyter notebook or terminal window.
Start tensorboard from notebook
First you should import TENSORBOARD_RESULTS_DIR from researcher environment in another cell
from fedbiomed.researcher.environ import environ
tensorboard_dir = environ['TENSORBOARD_RESULTS_DIR']Load tensorboard extension in a different code block.
%load_ext tensorboardRun following command to start tensorboard
tensorboard --logdir "$tensorboard_dir"Start tensorboard from terminal command line
Open new terminal and change directory to Fed-BioMed base directory (${FEDBIOMED_DIR})
Make sure that already activated fedbiomed researcher conda environment :
source ./scripts/fedbiomed_environment researcherLaunch tensorboard with the following command :
tensorboard --logdir "$tensorboard_dir"`Fed-BioMed offers optional training plan approval feature to approve the training plans requested by the researcher. This training plan approval process is done by hashing/checksum operation by the ModelManager of node instance. When the TRAINING_PLAN_APPROVAL mode is enabled, node should register/approve training plan files before performing the training. For testing and easy development, there are already presented default training plans by Fed-BioMed for the tutorials that we provide in the notebooks directory. However, node can also enable or disable the mode for allowing default training plans to perform training.
Enabling training plan approval mode, allowing default Fed-BioMed training plans and the hashing algorithm that will be performed for the checksum operation can be configurred from the config file of the node. The following code snippet represents an example security section of config file with default values.
[default]
# ....
[security]
hashing_algorithm = SHA256
allow_default_training_plans = True
training_plan_approval = False
By default, when node is launched for the first time without additional security parameters, training_plan_approval mode comes as disabled. If training_plan_approval is disabled the status of allow_defaults_training_plans will have no effect. To enable training_plan_approval you should set training_plan_approval to True and if it is desired allow_default_training_plans can be set to False for not accepting training plans of default Fed-BioMed examples.
The default hashing algorithm is SHA256 and it can also be changed to other hashing algorithms that are provided by Fed-BioMed. You can see the list of Hashing algorithms in the following section.
ModelManager provides different hashing algorithms, and the algorithm can be changed through the config file of the node. The name of the algorithms should be typed with capital letters. However, after changing hashing algorithm node should be restarted because it checks/updates hashing algorithms of the register/default training plans during the starting process.
Provided hashing algorithms are SHA256, SHA384, SHA512, SHA3_256, SHA3_384, SHA3_512, BLAKE2B and BLAKE2S. These are the algorithms that has been guaranteed by hashlib library of Python.
To enable training_plan_approval mode and allow_default_training_plans node, start the following command.
ENABLE_TRAINING_PLAN_APPROVAL=True ALLOW_DEFAULT_TRAINING_PLANS=True ./scripts/fedbiomed_run node --config config-n1.ini startThis command will start the node with training plan approval activated mode even the config file has been set as training_plan_aproval = False. However it doesn't change the config file. If there is no config file named config-n1.ini it creates a config file for the node with enabled training plan approval mode.
[security]
hashing_algorithm = SHA256
allow_default_training_plans = True
training_plan_approval = True
For starting node with disabled training plan approval and default training plans;
```shell
ENABLE_TRAINING_PLAN_APPROVAL=False ALLOW_DEFAULT_TRAINING_PLANS=False ./scripts/fedbiomed_run node --config config-n1.ini start
Default training plans are located in the envs/common/default_training_plans/ directory as txt files. Each time the node starts with the training_plan_approval = True and allow_default_training_plan = True modes, hashing of the training plan files are checked to detect if the file is modified, the hashing algorithm has changed or is there any new training plan file added. If training plan files are modified ModelManager updates hashes for these training plans in the database. If the hashing algorithm of the training plan is different from the active hashing algorithm, hashes also get updated. This process only occurs when both training_plan_approval and allow_default_training_plan modes are activated. To add new default training plan for the examples or for testing, training plan files should be saved as txt and copied into the envs/common/default_training_plans directory. After the copy/save operation node should be restarted.
New training plans can be registered using fedbiomed_run scripts with training-plan register option.
./scripts/fedbiomed_run node --config config-n1.ini training-plan registerThe CLI asks for the name of the training plan, description and the path where training plan file is stored. Model files should be saved as txt in the file system for registration. This is because these files are for only hashing purposes not for loading modules.
Following command is used for deleting registered training plans.
./scripts/fedbiomed_run node --config config-n1.ini training-plan delete
Output of this command will list registered training plans with their name and id. It will ask to select training plan file you would like to remove. For example, in the follwing example, typing 1 will remove the MyModel from registered/approved list of training plans.
Select the training plan to delete:
1) MyModel Model ID training_plan_98a1e68d-7938-4889-bc46-357e4ce8b6b5
Select:
Default training plans can not be removed using fedbiomed CLI. They should be removed from the envs/common/default_training_plans directory. After restarting the node, deleted training plan files will be also removed from the TrainingPlans table of the node DB.
Following command is used for updating registered training plans. It updates chosen training plan with provided new training plan file. User also can provide same training plan file to update its content.
./scripts/fedbiomed_run node --config config-n1.ini training-plan update
Node GUI provides an interface for Node to manage datasets and deploy new ones. GUI consists of two components, Server and UI. Server is developed on Flask framework and UI is developed using ReactJS. Flask provides API
services that use Fed-BioMed's DataManager for deploying and managing dataset. All the source files for GUI has been
located on the ${FEDBIOMED_DIR}/gui directory.
Node GUI can be started using Fed-BioMed CLI.
${FEDBIOMED_DIR}/scripts/fedbiomed_run node [--config [CONFIG_FILE_NAME]] gui --data-folder '<path-for-data-folder>' startArguments:
data-folder: Data folder represents the folder path where datasets have been stored. It can be absolute or relative path. If it is relative path, Fed-BioMed base directory is going to be used as reference. Ifdatafolderis not provided. Script will look fordatafolder in the Fed-BioMed root directory and if it doesn't exist it will raise an error.--config: Config file represents the name of the configuration file which is going to be used for GUI. If it is not provided, default will beconfig_node.ini.
It is also possible to start GUI on specific host and port, By default it is started localhost as host and 8484 as port. To change
it you can modify following command.
The GUI is based on HTTPS and by default, it will generate a self-signed certificate for you. Butyou can also start GUI specifying the certificate and the private key
names you want to use for HTTPS support. Please note that they must be in ${FEDBIOMED_DIR}/etc folder.
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config '<name-of-the-config-file> gui --data-folder '<path-for-data-folder>' ' cert '<name-of-certificate>' key '<name-of-private-key>' startIMPORTANT: Please always consider providing data-folder argument while starting the GUI.
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config config-n1.ini gui data-folder ../data --port 80 --host 0.0.0.0 start
When the Node GUI is started, it installs npm modules and builds ReactJS application in ${FEDBIOMED_DIR}/var/gui-build. If the GUI
is already built (means that gui/ui/node_modules and var/gui-build folders exist), it does not reinstall and rebuild ReactJS. If you want to
reinstall and rebuild, please add --recreate flag in the command same as below,
${FEDBIOMED_DIR}/scripts/fedbiomed_run node gui data-folder ../data --recreate startIt is possible to start multiple Node GUIs for different nodes as long as the http ports are different. The
commands below starts three Node GUI for the nodes; config-n1.ini, config-n2.ini and config-n3.ini on the ports respectively, 8181, 8282 and 8383.
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config config-n1.ini gui --data-folder ../data --port 8181 start
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config config-n2.ini gui --data-folder ../data --port 8282 start
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config config-n2.ini gui --data-folder ../data --port 8383 startIf you want to customize or work on user interface for debugging purposes, it is always better to use ReactJS in development mode, otherwise building GUI
after every update will take a lot of time. To launch user interface in development mode first you need to start Flask server. This can be
easily done with the previous start command. Currently, Flask server always get started on development mode. To enable debug mode you should add --debug
flag to the start command.
${FEDBIOMED_DIR}/scripts/fedbiomed_run node --config config-n1.ini gui --data-folder ../data --debug startImportant: Please do not change Flask port and host while starting it for development purposes. Because React (UI) will be calling
localhost:8484/api endpoint in development mode.
The command above will serve var/gui-build directory as well as API services. It means that on the URL localhost:8484 you will be able to
see the user interface. This user interface won't be updated automatically because it is already built. To have dynamic update for user interface you can start React with npm start.
source ${FEDBIOMED_DIR}/scripts/fedbiomed_environment gui
cd ${FEDBIOMED_DIR}/gui/ui
npm startAfter that if you go localhost:3000 you will see same user interface is up and running for development. When you change the source codes
in ${FEDBIOMED_DIR}/gui/ui/src it will get dynamically updated on localhost:3000.
Since Flask is already started in debug mode, you can do your development/update/changes for server side (Flask) in
${FEDBIOMED_DIR}/gui/server. React part (ui) on development mode will call API endpoint from localhost:8484, this is why
first you should start Flask server first.
After development/debugging is done. To update changes in built GUI, you need to start GUI with --recreate command. Afterward,
you will be able to see changes on the localhost:8484 URL which serve built UI files.
${FEDBIOMED_DIR}/scripts/fedbiomed_run data-folder ../data gui --recreate startFed-BioMed uses MP-SPDZ to provide secure aggregation of the model parameters. Running secure aggregation in Fed-BioMed is optional which makes MP-SPDZ installation/configuration optional as well. Fed-BioMed will be able to run FL experiment without MP-SPDZ as long as secure aggregation is not activated on the nodes and the researcher components.
Configuration or installation can be done with the following command by specifying the Fed-BioMed component.
If node and the researcher will be started in the same clone if Fed-BioMed running following command with once
(node or researcher) will be enough. For macOS, the operating system (Darwin) should higher than High Sierra (10.13)
${FEDBIOMED_DIR}/scripts/fedbiomed_configure_secagg (node|researcher)MP-SPDZ protocols for secure aggregation and multi party computation will be executed internally by
Fed-BioMed node and researcher components. The script for executing the protocols is located in
${FEDBIOMED_DIR}/scripts/fedbiomed_mpc. Please run following commands to see instructions and usage.
${FEDBIOMED_DIR}/scripts/fedbiomed_mpc (node | researcher) --help
${FEDBIOMED_DIR}/scripts/fedbiomed_mpc (node | researcher) *WORKDIR* compile --help
${FEDBIOMED_DIR}/scripts/fedbiomed_mpc (node | researcher) *WORKDIR* exec --help
${FEDBIOMED_DIR}/scripts/fedbiomed_mpc (node | researcher) *WORKDIR* shamir-server-key --help
React
Typescript
Django
Laravel
Facebook
Microsoft
Google
Alibaba
D3
Tencent