project screenshot 1
project screenshot 2
project screenshot 3

nfl

Use Nillion for training federated learning models while keeping datasets private.

Project Description

This project uses NIllion to perform a federated learning training procedure for neural networks. It provides a framework for arbitrary networks once Nillion scales to more computations.

It provides a script to autogenerate a program that can perform federated learning updates for a model as well as functionality to store local weights as secrets, share computation permissions for these secrets with the global party, and then update the local weights at every training epoch.

How it's Made

This project is entirely using Nillion and Python. Pytorch was the deep learning framework used, as well as a few other utility libraries in Python.

Notable things include autogenerated Nillion program for arbitrary neural networks. This allows the approach to be scalable to many more complex architectures. It also includes the entire end to end training loop in a series of scripts.

background image mobile

Join the mailing list

Get the latest news and updates