Stalactite: Toolbox for Fast Prototyping of Vertical Federated Learning Systems

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Zakharova, Anastasiia, Alexandrov, Dmitriy, Khodorchenko, Maria, Butakov, Nikolay, Vasilev, Alexey, Savchenko, Maxim, Grigorievskiy, Alexander
Formato: Preprint
Publicado: 2024
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910630390267904
author Zakharova, Anastasiia
Alexandrov, Dmitriy
Khodorchenko, Maria
Butakov, Nikolay
Vasilev, Alexey
Savchenko, Maxim
Grigorievskiy, Alexander
author_facet Zakharova, Anastasiia
Alexandrov, Dmitriy
Khodorchenko, Maria
Butakov, Nikolay
Vasilev, Alexey
Savchenko, Maxim
Grigorievskiy, Alexander
contents Machine learning (ML) models trained on datasets owned by different organizations and physically located in remote databases offer benefits in many real-world use cases. State regulations or business requirements often prevent data transfer to a central location, making it difficult to utilize standard machine learning algorithms. Federated Learning (FL) is a technique that enables models to learn from distributed datasets without revealing the original data. Vertical Federated learning (VFL) is a type of FL where data samples are divided by features across several data owners. For instance, in a recommendation task, a user can interact with various sets of items, and the logs of these interactions are stored by different organizations. In this demo paper, we present \emph{Stalactite} - an open-source framework for VFL that provides the necessary functionality for building prototypes of VFL systems. It has several advantages over the existing frameworks. In particular, it allows researchers to focus on the algorithmic side rather than engineering and to easily deploy learning in a distributed environment. It implements several VFL algorithms and has a built-in homomorphic encryption layer. We demonstrate its use on a real-world recommendation datasets.
format Preprint
id arxiv_https___arxiv_org_abs_2409_15558
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Stalactite: Toolbox for Fast Prototyping of Vertical Federated Learning Systems
Zakharova, Anastasiia
Alexandrov, Dmitriy
Khodorchenko, Maria
Butakov, Nikolay
Vasilev, Alexey
Savchenko, Maxim
Grigorievskiy, Alexander
Machine Learning
Distributed, Parallel, and Cluster Computing
Information Retrieval
Machine learning (ML) models trained on datasets owned by different organizations and physically located in remote databases offer benefits in many real-world use cases. State regulations or business requirements often prevent data transfer to a central location, making it difficult to utilize standard machine learning algorithms. Federated Learning (FL) is a technique that enables models to learn from distributed datasets without revealing the original data. Vertical Federated learning (VFL) is a type of FL where data samples are divided by features across several data owners. For instance, in a recommendation task, a user can interact with various sets of items, and the logs of these interactions are stored by different organizations. In this demo paper, we present \emph{Stalactite} - an open-source framework for VFL that provides the necessary functionality for building prototypes of VFL systems. It has several advantages over the existing frameworks. In particular, it allows researchers to focus on the algorithmic side rather than engineering and to easily deploy learning in a distributed environment. It implements several VFL algorithms and has a built-in homomorphic encryption layer. We demonstrate its use on a real-world recommendation datasets.
title Stalactite: Toolbox for Fast Prototyping of Vertical Federated Learning Systems
topic Machine Learning
Distributed, Parallel, and Cluster Computing
Information Retrieval
url https://arxiv.org/abs/2409.15558