Skip to main content

Research Repository

Advanced Search

PyVertical: A Vertical Federated Learning Framework for Multi-headed SplitNN

Romanini, Daniele; Hall, Adam James; Papadopoulos, Pavlos; Titcombe, Tom; Ismail, Abbas; Cebere, Tudor; Sandmann, Robert; Roehm, Robin; Hoeh, Michael A.


Daniele Romanini

Adam James Hall

Tom Titcombe

Abbas Ismail

Tudor Cebere

Robert Sandmann

Robin Roehm

Michael A. Hoeh


We introduce PyVertical, a framework supporting vertical federated learning using split neural networks. The proposed framework allows a data scientist to train neural networks on data features vertically partitioned across multiple owners while keeping raw data on an owner's device. To link entities shared across different datasets' partitions, we use Private Set Intersection on IDs associated with data points. To demonstrate the validity of the proposed framework, we present the training of a simple dual-headed split neural network for a MNIST classification task, with data samples vertically distributed across two data owners and a data scientist.

Presentation Conference Type Poster
Conference Name ICLR 2021 Workshop on Distributed and Private Machine Learning (DPML 2021)
Start Date May 7, 2021
Deposit Date Oct 31, 2022
Publicly Available Date Nov 1, 2022
Public URL
Publisher URL
Related Public URLs


PyVertical: A Vertical Federated Learning Framework For Multi-headed SplitNN (990 Kb)

You might also like

Downloadable Citations