ASFL: An Adaptive Model Splitting and Resource Allocation Framework for Split Federated Learning
Organizations: Department of Electrical and Computer Engineering, The University of British Columbia, Vancouver, Canada · Department of Computer Science and Engineering, Southern University of Science and Technology, China
Abstract
Federated learning (FL) enables multiple clients to collaboratively train a machine learning model without sharing their raw data. However, the limited computation resources of the clients may result in a high delay and energy consumption on training. In this paper, we propose an adaptive split federated learning (ASFL) framework over wireless networks. ASFL exploits the computation resources of the central server to train part of the model and enables adaptive model splitting as well as resource allocation during training. To optimize the learning performance (i.e., convergence rate) and efficiency (i.e., delay and energy consumption) of ASFL, we theoretically analyze the convergence rate and formulate a joint learning performance and resource allocation optimization problem. Solving this problem is challenging due to the long-term delay and energy consumption constraints as well as the coupling of the model splitting and resource allocation decisions. We propose an online optimization enhanced block coordinate descent (OOE-BCD) algorithm to solve the problem iteratively. Experimental results show that when compared with five baseline schemes, our proposed ASFL framework converges faster and reduces the total delay and energy consumption by up to 75% and 80%, respectively.
Figures & tables
| Notation | Definition | Notation | Definition |
|---|---|---|---|
| Bandwidth of a resource block | Output size of the -th layer (in bits) | ||
| Downlink bandwidth | Set of all training rounds | ||
| Downlink transmission rate for client in the -th training round | Number of training rounds | ||
| Uplink transmission rate for client in the -th training round | Packet error rate of the intermediate output transmission of client in the -th training round | ||
| Number of training samples of client | Allocation decision of the -th RB for client in the -th training round | ||
| CPU frequency of client | Computation workload for FP of the -th layer (in FLOPs) |
| Para. | Value | Para. | Value | Para. | Value |
| 1 MHz | 1.5 W | 8 | |||
| 8 MHz | cycles/FLOP | 5 W | |||
| 173 dBm/Hz | cycles/FLOP | 0.05 | |||
| 1 | cycles/s | 10 | |||
| 20 s | J |
| Base Model | Baseline | CIFAR-10 (%) | CIFAR-100 (%) |
|---|---|---|---|
| VGG-19 | ASFL | 89.20 | 62.10 |
| FedAvg [ 1 ] | 89.69 | 61.35 | |
| SL [ 3 ] | 83.41 | 58.30 | |
| SFL [ 9 ] | 88.77 | 62.43 | |
| ACC-SFL [ 19 ] | 88.02 | 61.96 | |
| EPSL [ 20 ] | 87.59 | 61.58 |
Split Federated Learning
Split federated learning divides models between clients and servers to train collaboratively while keeping raw data local.
See all 37 papers