arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2003.08725cs.LGcs.CRstat.ML

Privacy-preserving Traffic Flow Prediction: A Federated Learning Approach

Southern University of Science and Technology(南方科技大学);Heilongjiang University(黑龙江大学);Nanyang Technological University(南洋理工大学)

Yi Liu, James J. Q. Yu, Jiawen Kang, Dusit Niyato, Shuyu Zhang

英文摘要:

Existing traffic flow forecasting approaches by deep learning models achieve excellent success based on a large volume of datasets gathered by governments and organizations. However, these datasets may contain lots of user's private data, which is challenging the current prediction approaches as user privacy is calling for the public concern in recent years. Therefore, how to develop accurate traffic prediction while preserving privacy is a significant problem to be solved, and there is a trade-off between these two objectives. To address this challenge, we introduce a privacy-preserving machine learning technique named federated learning and propose a Federated Learning-based Gated Recurrent Unit neural network algorithm (FedGRU) for traffic flow prediction. FedGRU differs from current centralized learning methods and updates universal learning models through a secure parameter aggregation mechanism rather than directly sharing raw data among organizations. In the secure parameter aggregation mechanism, we adopt a Federated Averaging algorithm to reduce the communication overhead during the model parameter transmission process. Furthermore, we design a Joint Announcement Protocol to improve the scalability of FedGRU. We also propose an ensemble clustering-based scheme for traffic flow prediction by grouping the organizations into clusters before applying FedGRU algorithm. Through extensive case studies on a real-world dataset, it is shown that FedGRU's prediction accuracy is 90.96% higher than the advanced deep learning models, which confirm that FedGRU can achieve accurate and timely traffic prediction without compromising the privacy and security of raw data.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑