Communication-and-Computation Efficient Split Federated Learning: Gradient Aggregation and Resource Management

TL;DR

This paper introduces a novel split federated learning framework that optimizes communication and computation efficiency through dynamic model splitting and gradient broadcasting, improving convergence and reducing latency.

Contribution

It proposes a joint optimization strategy for model splitting, resource allocation, and privacy preservation, integrating DDQN with convex optimization to enhance SFL performance.

Findings

The framework achieves lower communication overhead compared to existing methods.

Optimized model splitting improves convergence rates.

Experimental results validate the effectiveness of the proposed approach.

Abstract

With the prevalence of Large Learning Models (LLM), Split Federated Learning (SFL), which divides a learning model into server-side and client-side models, has emerged as an appealing technology to deal with the heavy computational burden for network edge clients. However, existing SFL frameworks would frequently upload smashed data and download gradients between the server and each client, leading to severe communication overheads. To address this issue, this work proposes a novel communication-and-computation efficient SFL framework, which allows dynamic model splitting (server- and client-side model cutting point selection) and broadcasting of aggregated smashed data gradients. We theoretically analyze the impact of the cutting point selection on the convergence rate of the proposed framework, revealing that model splitting with a smaller client-side model size leads to a better convergence performance and vise versa. Based on the above insights, we formulate an optimization problem to minimize the model convergence rate and latency under the consideration of data privacy via a joint Cutting point selection, Communication and Computation resource allocation (CCC) strategy. To deal with the proposed mixed integer nonlinear programming optimization problem, we develop an algorithm by integrating the Double Deep Q-learning Network (DDQN) with convex optimization methods. Extensive experiments validate our theoretical analyses across various datasets, and the numerical results demonstrate the effectiveness and superiority of the proposed communication-efficient SFL compared with existing schemes, including parallel split learning and traditional SFL mechanisms.