Joint Optimization of Model Partition and Resource Allocation for Split Federated Learning Over Vehicular Edge Networks

Maoqiang Wu,Jiawen Kang,Yuan Wu,Ruibin Yang,Shengli Xie,Xumin Huang
DOI: https://doi.org/10.1109/TVT.2024.3399011
IF: 6.8
2024-10-01
IEEE Transactions on Vehicular Technology
Abstract:Split federated learning (SFL) has been regarded as an efficient paradigm to enable federated learning and reduce the computation burdens at the devices by allowing them to train parts of the model. However, deploying SFL over resource-constrained vehicular edge networks is challenging, and a cost-effective scheme is necessitated to minimize the total time and energy consumption of vehicular devices. To this end, we use an improved reinforcement learning method to present a joint optimization scheme that can efficiently determine the optimal model partition point for each vehicular device and the optimal allocations of the computing resource and bandwidth resource among all vehicular devices. Experimental results validate the effectiveness and performance advantages of our proposed scheme.
Computer Science,Engineering
What problem does this paper attempt to address?