arrow
Return

Accelerating Split Federated Learning Over Wireless Communication Networks

delete2024-06-01
delete4
delete
OA
AI
C
Ce Xu
J
Jinxuan Li
Y
Yuan Liu *
M
Miaowen Wen
DOI:10.1109/TWC.2023.3327372delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
The development of artificial intelligence (AI) provides opportunities for the promotion of deep neural network (DNN)-based applications. However, the large amount of parameters and computational complexity of DNN makes it difficult to deploy it on edge devices which are resource-constrained. An efficient method to address this challenge is model partition/splitting, in which DNN is divided into two parts which are deployed on device and server respectively for co-training or co-inference. In this paper, we consider a split federated learning (SFL) framework that combines the parallel model training mechanism of federated learning (FL) and the model splitting structure of split learning (SL). We consider a practical scenario of heterogeneous devices with individual split points of DNN. We formulate a joint problem of split point selection and bandwidth allocation to minimize the system latency. By using alternating optimization, we decompose the problem into two sub-problems and solve them optimally. Experiment results demonstrate the superiority of our work in latency reduction and accuracy improvement.
Keywords:
Split federated learning
model splitting
resource allocation

Journal

IEEE Transactions on Wireless Communications cover
IEEE Transactions on Wireless Communications
IF:
10.7
Papers:
1.3W
Citations:
5.3W

Organization

C
China Southern Power Grid
Scholars:
3.4K
Papers: 2.4K
Citations: 8
S
south china university of technology
Scholars:
6.7W
Papers: 5.1W
Citations: 85