Towards a Unified Framework for Split Learning
Samuel Horvath Praneeth Vepakomma
Samuel Horvath
Praneeth Vepakomma
Author
Supervisor
Department
Machine Learning
Embargo End Date
Type
Workshop
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
Keywords
Distributed Collaborative Machine Learning, Split Learning