We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:

References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Machine Learning

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

Abstract: 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.
Comments: ICLR 2021 Workshop on Distributed and Private Machine Learning (DPML 2021)
Subjects: Machine Learning (cs.LG); Cryptography and Security (cs.CR); Distributed, Parallel, and Cluster Computing (cs.DC)
Cite as: arXiv:2104.00489 [cs.LG]
  (or arXiv:2104.00489v3 [cs.LG] for this version)

Submission history

From: Pavlos Papadopoulos [view email]
[v1] Thu, 1 Apr 2021 14:21:33 GMT (1879kb,D)
[v2] Mon, 12 Apr 2021 18:15:08 GMT (1044kb,D)
[v3] Wed, 14 Apr 2021 08:05:04 GMT (1044kb,D)

Link back to: arXiv, form interface, contact.