Federated learning‐based trajectory prediction model with privacy preserving for intelligent vehicle

Abstract

The existing trajectory prediction is mainly for specific road sections, which is poor to adapt to complex and changing traffic scenarios. Meanwhile, decentralized trajectory data is hard to be fully utilized in the data silo environment. To solve data silos in the intelligent vehicle industry, introducing federated learning methods in vehicular edge computing has attracted extensive attention. But traditional federated learning still has the potential to suffer from the mining of training data in the case of model privacy leakage. In this paper, a vehicle trajectory prediction method based on federated learning and homomorphic encryption has been presented, which adopts a three-layer architecture with a vehicle cluster, edge computing server, and cloud core network. Compared with traditional centralized deep learning methods, this approach enables joint modeling of multiple parties to improve the model's generalization performance. We used a proxy re-encryption algorithm to implement key distribution and also designed an encrypted federated network algorithm FAHEFL, which uses FAHE1 homomorphic encryption to protect the privacy of the model in parameter transmission. Each local model includes a driving behavior recognition module and trajectory output module. The driving behavior recognition module uses a 1D convolutional neural network to recognize the driving behavior, then input recognition results and historical trajectories to the trajectory output module, which uses LSTM neural network to output predicted trajectories. The experiment results show that the model built with FAHEFL has no more 1% error in the driving behavior recognition module than concentrated learning, while the minimum mean square error of the trajectory prediction module increased by only 2.5%. This article also discusses the performance between FAHEFL and well-known cryptographic federation network algorithms.