Semi-Asynchronous Federated Learning with Trajectory Prediction for Vehicular Edge Computing

Yuxuan Deng,Xiuhua Li,Chuan Sun,Qilin Fan,Xiaofei Wang,Victor C. M. Leung
DOI: https://doi.org/10.1109/iwqos61813.2024.10682953
2024-01-01
Abstract:Federated learning, as a distributed machine learning paradigm, offers promising solutions for vehicular edge computing (VEC) networks. However, federated learning in VEC with classification tasks still faces two key challenges: i) Delayed data labeling hampers supervised training; ii) Dynamic vehicle behavior complicates training scheduling and model uploads to edge servers. In this paper, we propose a semi-asynchronous federated learning algorithm for VEC. Specifically, it utilizes knowledge distillation to generate soft labels from raw data for supervised training, and estimates model training and uploading time through trajectory prediction. We further logically group vehicles based on the characteristics of their dynamic behavior. We then employ synchronous aggregation within groups and asynchronous aggregation between groups to optimize model performance while reducing latency. Finally, we conduct separate comparative experiments for all components, demonstrating that each component possesses unique advantages. Experiment results show that the proposed algorithm outperforms existing schemes in terms of accuracy and latency. The code is available at: https://github.com/dyxcode/Semi-Asynchronous-Federated-Learning.
What problem does this paper attempt to address?