Enhanced Federated Learning for Edge Data Security in Intelligent Transportation Systems

Abstract

Federated learning (FL) provides a promising solution to meet the requirements of data privacy and security in intelligent transportation systems (ITS), which enables edge devices and road side units (RSUs) to collaboratively train learning models without exposing the raw data. However, the deep leakage from gradients (DLG) still leads to the risk of divulging the original data. Meanwhile, the existing gradient protection methods based on secure multi-party computation (SMC) result in huge communication overheads and latency, which are difficult to satisfy the real-time demands of both FL and the diverse services in ITS. Focusing on improving edge data security in ITS, this paper proposes an enhanced federated learning (FL) model (SemBroc-RF) with reinforcement learning, which considers the advantages of both end-to-end homomorphic encryption (HE) and SMC. To reduce communication overheads and strengthen data security in RSUs and end devices simultaneously, a partially encrypted secure multi-party broadcast computation algorithm (SemBroc) is designed, which achieves the time complexity O(n) by constructing the decoding function and sharing the gradients among the local models. To improve the model accuracy by gradients aggregation, a FL algorithm (GreFLa) with reinforcement learning is proposed based on the adaptive assigned weight of the local gradients. Theoretical analysis and detailed simulation results verify that SemBroc-RF can effectively prevent gradient leakage. On the MNIST and CIFAR-10 datasets, compared with the benchmark, the accuracy of SemBroc-RF is increased by 3.63% and 1.35%, and the training round of SemBroc-RF is reduced by 70.8% and 45.6%, respectively.