zhxchd / vFedCCE

A vertical federated learning algorithm for classfication problems with gradient-based optimization.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

vFedCCE

A vertical federated learning algorithm for classfication problems with gradient-based optimization.

Data set structure

This algorithm is going to train a shared model on the common IDs (aligned by some private entity resolution technique) between vertically partitioned datasets, as shown in the following figure.

The Proposed vFedCCE Algorithm

The algorithm itself can be described by the following sequence diagram.

Details of the algorithm can be found in the report.

Details of the implementation (with TensorFlow and Keras) can be found under the directory src.

About

A vertical federated learning algorithm for classfication problems with gradient-based optimization.

License:MIT License


Languages

Language:Jupyter Notebook 100.0%