Skip to content
/ vFedCCE Public

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

License

Notifications You must be signed in to change notification settings

zhxchd/vFedCCE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.