Search papers, labs, and topics across Lattice.
This paper introduces an efficient optimization framework for split federated learning (SFL) that addresses the challenges of discrete decision variables related to model splitting and resource allocation. By formulating the problem as a two-dimensional master problem, the authors develop a polynomial-time algorithm that guarantees a global optimum for model splitting and an efficient approximation method for the joint optimization of model splitting and resource allocation. Experimental results demonstrate that the proposed approach effectively minimizes training costs by optimizing the energy-latency tradeoff, making it suitable for large-scale user populations in resource-constrained networks.
Achieving global optimality in model splitting for split federated learning could revolutionize resource allocation strategies in edge computing environments.
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 \textit{heuristic} or \textit{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 $(1+蔚)$-approximation guarantee. Extensive experiments show that the proposed approach provides efficient solutions to strike the optimal energy--latency tradeoff.
