Logo image
Open Research University homepage
Surrey researchers Sign in
PipeSFL: A Fine-Grained Parallelization Framework for Split Federated Learning on Heterogeneous Clients
Journal article   Open access   Peer reviewed

PipeSFL: A Fine-Grained Parallelization Framework for Split Federated Learning on Heterogeneous Clients

Yunqi Gao, Bing Hu, Mahdi Boloursaz Mashhadi, Wei Wang and Mehdi Bennis
IEEE transactions on mobile computing, Vol.24(3), pp.1774-1791
03/2025

Abstract

Servers Training Computational modeling Federated learning Processor scheduling Mobile computing Artificial neural networks Scheduling Synchronization Peer-to-peer computing Split learning client heterogeneity parallel computing artificial intelligence of things (AIoT)

Split Federated Learning (SFL) improves scalability of Split Learning (SL) by enabling parallel computing of the learning tasks on multiple clients. However, state-of-the-art SFL schemes neglect the effects of heterogeneity in the clients’ computation and communication performance as well as the computation time for the tasks offloaded to the cloud server. In this paper, we propose a fine-grained parallelization framework, called PipeSFL, to accelerate SFL on heterogeneous clients. PipeSFL is based on two key novel ideas. First, we design a server-side priority scheduling mechanism to minimize per-iteration time. Second, we propose a hybrid training mode to reduce per-round time, which employs asynchronous training within rounds and synchronous training between rounds. We theoretically prove the optimality of the proposed priority scheduling mechanism within one round and analyze the total time per round for PipeSFL, SFL and SL. We implement PipeSFL on PyTorch. Extensive experiments on seven 64-client clusters with different heterogeneity demonstrate that at training speed, PipeSFL achieves up to 1.65x and 1.93x speedup compared to EPSL and SFL, respectively. At energy consumption, PipeSFL saves up to 30.8% and 43.4% of the energy consumed within each training round compared to EPSL and SFL, respectively. PipeSFL’s code is available at https://github.com/ZJU-CNLAB/PipeSFL.

pdf
manuscript (2)11.87 MBDownloadView
Author's Accepted Manuscript Open Access

Metrics

Details

Logo image

Usage Policy