Efficient Resource Optimization for Split Federated Learning
Organizations: Department of Electrical and Computer Engineering, the University of Hong Kong, Pok Fu Lam, Hong Kong SAR, 999077
Abstract
Split federated learning (SFL) has emerged as a powerful paradigm for model training at the edge. However, SFL inherently involves discrete decision variables for model splitting and resource allocation, resulting in a challenging mixed-integer problem. Consequently, prior optimization schemes for SFL are either heuristic or computationally inefficient, which cannot handle large-scale user populations. To address this limitation, this work establishes an efficient optimization framework for SFL under resource-constrained networks. Our framework jointly optimizes model splitting and resource allocation to minimize training cost, which is defined as the weighted sum of latency and energy costs. We first study the model splitting problem and develop a polynomial-time algorithm that achieves the global optimum. Then, we extend the approach to the joint model splitting and resource allocation problem. In this case, we formulate it as a two-dimensional master problem and develop an efficient approximation method with a -approximation guarantee. Extensive experiments show that the proposed approach provides efficient solutions to strike the optimal energy--latency tradeoff.
Figures & tables
| Notation | Description |
| The set of all clients | |
| The number of clients | |
| The number of layers in the global model | |
| The cut layer of client | |
| The number of GPU FLOPs per cycle of client | |
| The number of GPU FLOPs per cycle of the edge server |
| Parameter | Value | Parameter | Value |
Split Federated Learning
Split federated learning divides models between clients and servers to train collaboratively while keeping raw data local.
See all 37 papers