Towards Straggler-Resilient Split Federated Learning: An Unbalanced Update Approach

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Liang, Dandan, Zhang, Jianing, Chen, Evan, Li, Zhe, Li, Rui, Yang, Haibo
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866918169339232256
author Liang, Dandan
Zhang, Jianing
Chen, Evan
Li, Zhe
Li, Rui
Yang, Haibo
author_facet Liang, Dandan
Zhang, Jianing
Chen, Evan
Li, Zhe
Li, Rui
Yang, Haibo
contents Split Federated Learning (SFL) enables scalable training on edge devices by combining the parallelism of Federated Learning (FL) with the computational offloading of Split Learning (SL). Despite its great success, SFL suffers significantly from the well-known straggler issue in distributed learning systems. This problem is exacerbated by the dependency between Split Server and clients: the Split Server side model update relies on receiving activations from clients. Such synchronization requirement introduces significant time latency, making straggler a critical bottleneck to the scalability and efficiency of the system. To mitigate this problem, we propose MU-SplitFed, a straggler-resilient SFL algorithm in zeroth-order optimization that decouples training progress from straggler delays via a simple yet effective unbalanced update mechanism. By enabling the server to perform $τ$ local updates per client round, MU-SplitFed achieves a convergence rate of $O(\sqrt{d/(τT)})$ for non-convex objectives, demonstrating a linear speedup of $τ$ in communication rounds. Experiments demonstrate that MU-SplitFed consistently outperforms baseline methods with the presence of stragglers and effectively mitigates their impact through adaptive tuning of $τ$. The code for this project is available at https://github.com/Johnny-Zip/MU-SplitFed.
format Preprint
id arxiv_https___arxiv_org_abs_2510_21155
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Towards Straggler-Resilient Split Federated Learning: An Unbalanced Update Approach
Liang, Dandan
Zhang, Jianing
Chen, Evan
Li, Zhe
Li, Rui
Yang, Haibo
Distributed, Parallel, and Cluster Computing
Artificial Intelligence
Machine Learning
Split Federated Learning (SFL) enables scalable training on edge devices by combining the parallelism of Federated Learning (FL) with the computational offloading of Split Learning (SL). Despite its great success, SFL suffers significantly from the well-known straggler issue in distributed learning systems. This problem is exacerbated by the dependency between Split Server and clients: the Split Server side model update relies on receiving activations from clients. Such synchronization requirement introduces significant time latency, making straggler a critical bottleneck to the scalability and efficiency of the system. To mitigate this problem, we propose MU-SplitFed, a straggler-resilient SFL algorithm in zeroth-order optimization that decouples training progress from straggler delays via a simple yet effective unbalanced update mechanism. By enabling the server to perform $τ$ local updates per client round, MU-SplitFed achieves a convergence rate of $O(\sqrt{d/(τT)})$ for non-convex objectives, demonstrating a linear speedup of $τ$ in communication rounds. Experiments demonstrate that MU-SplitFed consistently outperforms baseline methods with the presence of stragglers and effectively mitigates their impact through adaptive tuning of $τ$. The code for this project is available at https://github.com/Johnny-Zip/MU-SplitFed.
title Towards Straggler-Resilient Split Federated Learning: An Unbalanced Update Approach
topic Distributed, Parallel, and Cluster Computing
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2510.21155