GAPSL: A Gradient-Aligned Parallel Split Learning over Data-Heterogeneous Edge Computing Systems
Organizations: Department of Electrical and Computer Engineering, University of Hong Kong, Pok Fu Lam, Hong Kong, China · Interdisciplinary Centre for Security, Reliability and Trust (SnT), University of Luxembourg, L-1855 Luxembourg, Luxembourg · Hong Kong JC STEM Lab of Smart City and Department of Computer Science, City University of Hong Kong, Kowloon, Hong Kong SAR, China · School of Engineering, Edith Cowan University, Perth, WA 6027, Australia ·
Abstract
The increasing complexity of neural networks poses significant challenges for democratizing federated learning (FL) on resource-constrained edge devices. Parallel split learning (PSL) has emerged as a promising solution by offloading substantial computing workload to a server via model partitioning, shrinking client-side computing load, and eliminating the client-side model aggregation for reduced communication and deployment costs. However, the highly heterogeneous nature of client data in edge computing systems causes aggregation-free PSL to suffer from severe training divergence, stemming from gradient directional inconsistency across clients. To address this challenge, we propose GAPSL, a gradient-aligned PSL framework tailored for data-heterogeneous edge systems, which comprises two key components: leader gradient identification (LGI) and gradient direction alignment (GDA). LGI dynamically selects a set of directionally consistent device gradients to construct a leader gradient as a robust proxy for the global convergence trend. GDA employs a direction-aware regularization to align each client's gradient with the leader gradient, thereby mitigating inter-device gradient directional inconsistency and enhancing model convergence. We evaluate GAPSL on a prototype computing testbed. Extensive experiments demonstrate that GAPSL consistently outperforms state-of-the-art benchmarks in training accuracy, convergence latency, and system robustness under severe data heterogeneity.
Figures & tables
| Dataset | Model | Metric (Setting) | GAPSL | SFL | EPSL | Vanilla SL | PSL |
|---|---|---|---|---|---|---|---|
| CIFAR-10 | VGG-16 | Accuracy (IID) | 86.9 | 86.2 | 85.8 | 85.7 | 84.3 |
| Accuracy (Non-IID) | 63.3 | 60.7 | 59.4 | 57.3 | 47.3 | ||
| Time (IID) | 21 | 27 | 28 | 29 | 36 | ||
| Time (Non-IID) | 32 | 42 | 43 | 46 | 56 | ||
| CIFAR-100 | ViT-Base | Accuracy (IID) | 72.1 | 71.5 | 70.8 | 70.7 | 69.8 |
| Accuracy (Non-IID) | 57.0 | 53.4 | 51.4 | 50.3 | 43.2 |
| Setting | Devices Scale | GAPSL | SFL | EPSL | Vanilla SL | PSL |
|---|---|---|---|---|---|---|
| IID | 10 | 86.9 | 86.2 | 85.8 | 85.7 | 84.3 |
| 20 | 86.8 | 86.4 | 86.3 | 86.1 | 84.7 | |
| 30 | 86.7 | 86.2 | 86.0 | 85.8 | 84.6 | |
| 40 | 86.5 | 85.9 | 85.7 | 85.7 | 84.9 | |
| Non-IID | 10 | 63.3 | 60.7 | 59.4 | 57.3 | 47.3 |
| 20 | 63.1 | 60.1 | 57.2 | 56.3 | 45.5 |
| Dataset | Model | GAPSL | SFL | EPSL | Vanilla SL | PSL | |
|---|---|---|---|---|---|---|---|
| CIFAR-10 | VGG-16 | 0.9 | 85.0 | 84.1 | 82.8 | 82.7 | 81.3 |
| 0.7 | 78.1 | 76.6 | 74.9 | 73.4 | 70.3 | ||
| 0.5 | 73.2 | 71.2 | 68.0 | 66.2 | 60.9 | ||
| 0.3 | 69.3 | 65.1 | 61.8 | 58.9 | 53.5 | ||
| CIFAR-100 | ViT-Base | 0.9 | 70.1 | 69.3 | 68.2 | 67.3 | 66.1 |
| 0.7 | 66.2 | 63.1 | 62.3 | 62.0 | 57.4 |
Split Federated Learning
Split federated learning divides models between clients and servers to train collaboratively while keeping raw data local.
See all 37 papers