Loading...
Thumbnail Image
Item

Towards a Unified Framework for Split Learning

Radovic, Boris
Canini, Marco
Horvath, Samuel
Pejovic, Veljko
Vepakomma, Praneeth
Supervisor
Department
Machine Learning
Embargo End Date
Type
Journal Article
Date
2025
License
Language
English
Collections
Research Projects
Organizational Units
Journal Issue
Abstract
Split Learning (SL) is a principled approach for training models on data distributed across multiple devices without sharing training data. While SL emerged as an alternative to federated learning to reduce the compute burden on devices, it also enables to redistribute work across nodes. Despite its potential, there is no unified framework for implementing and deploying SL algorithms, leaving several research questions underexplored. To address this gap, we introduce SplitBud, a versatile framework to implement virtually any SL algorithm. By supporting various variants of SL, SplitBud facilitates research and development in the field. In this paper, we demonstrate its flexibility by implementing and evaluating multiple SL algorithms, and we discuss future directions for the field.
Citation
B. Radovi?, M. Canini, S. Horváth, V. Pejovi?, and P. Vepakomma, “Towards a Unified Framework for Split Learning,” Proceedings of the 5th Workshop on Machine Learning and Systems, pp. 183–191, Mar. 2025, doi: 10.1145/3721146.3721936.
Source
Proceedings of the 5th Workshop on Machine Learning and Systems
Conference
Keywords
Distributed Collaborative Machine Learning, Split Learning
Subjects
Source
Publisher
Association for Computing Machinery
Full-text link