VSFL: Trajectory prediction framework based on validity-aware semi-asynchronous federated learning in internet of vehicles

Abstract

Accurate prediction of vehicle trajectories on the Internet of Vehicles is crucial for improving traffic safety. Federated learning has been proposed to address the issue of islands in shared data. Traditional synchronous federated learning requires waiting for each node to complete training before model aggregation can occur, and the aggregation algorithm of the server cannot adjust the aggregation weight in real-time, which leads to increased communication overhead, time cost, and decreased accuracy. To balance these issues, this paper proposes a Validity-aware Semi-asynchronous Federated Learning-based framework for predicting vehicle trajectories. The framework makes the following improvements - firstly, based on the asynchronous cooperation algorithm of generalization research, the models of vehicle nodes within the same RoadSide Unit (RSU) range are aggregated into this RSU to reduce communication overhead and time costs. To further ensure the privacy and security of all participants, we added differential privacy algorithms during the model training process. Secondly, the server uses a weighted aggregation algorithm based on effective perception to aggregate all RSU models into a global model to address potential accuracy issues. This article uses a Social-LSTM like model to validate the effectiveness of this framework in the Next Generation Simulation (NGSIM) Vehicle Trajectories and Support Data dataset. Compared with traditional federated averaging algorithms and trust aware federated learning algorithms, this framework reduces distance errors by 27.80 % and 19.23 %, respectively, and improves accuracy by 22.56 % and 13.59 %.