# Privacy-preserving Traffic Flow Prediction: A Federated Learning Approach

arXiv: https://arxiv.org/abs/2003.08725 · HTML: https://mlsanity.com/paper/2003.08725/

## Key facts

- Cited by 83 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 19 Mar 2020.

## Built on by (most cited first, 30 of 83)

- [Federated Learning: Opportunities and Challenges](https://mlsanity.com/paper/2101.05428/) (2021)
- [Communication Efficiency in Federated Learning: Achievements and Challenges](https://mlsanity.com/paper/2107.10996/) (2021)
- [Federated Learning for Connected and Automated Vehicles: A Survey of Existing Approaches and Challenges](https://mlsanity.com/paper/2308.10407/) (2023)
- [Federated Learning for Medical Applications: A Taxonomy, Current Trends, Challenges, and Future Research Directions](https://mlsanity.com/paper/2208.03392/) (2022)
- [Fusion of Federated Learning and Industrial Internet of Things: A Survey](https://mlsanity.com/paper/2101.00798/) (2021)
- [FedDPGAN: Federated Differentially Private Generative Adversarial Networks Framework for the Detection of COVID-19 Pneumonia](https://mlsanity.com/paper/2104.12581/) (2021)
- [LightTR: A Lightweight Framework for Federated Trajectory Recovery](https://mlsanity.com/paper/2405.03409/) (2024)
- [A Survey of Federated Learning for Connected and Automated Vehicles](https://mlsanity.com/paper/2303.10677/) (2023)
- [Traffic Prediction using Artificial Intelligence: Review of Recent Advances and Emerging Opportunities](https://mlsanity.com/paper/2305.19591/) (2023)
- [Federated Learning for 5G Base Station Traffic Forecasting](https://mlsanity.com/paper/2211.15220/) (2022)
- [Federated Learning in Intelligent Transportation Systems: Recent Applications and Open Problems](https://mlsanity.com/paper/2309.11039/) (2023)
- [PerFED-GAN: Personalized Federated Learning via Generative Adversarial Networks](https://mlsanity.com/paper/2202.09155/) (2022)
- [Deep Learning Serves Traffic Safety Analysis: A Forward-looking Review](https://mlsanity.com/paper/2203.10939/) (2022)
- [Federated Learning Priorities Under the European Union Artificial Intelligence Act](https://mlsanity.com/paper/2402.05968/) (2024)
- [Efficient Federated Learning with Spike Neural Networks for Traffic Sign Recognition](https://mlsanity.com/paper/2205.14315/) (2022)
- [Wind Power Forecasting Considering Data Privacy Protection: A Federated Deep Reinforcement Learning Approach](https://mlsanity.com/paper/2211.02674/) (2022)
- [Failure Prediction in Production Line Based on Federated Learning: An Empirical Study](https://mlsanity.com/paper/2101.11715/) (2021)
- [A Survey on Federated Learning and its Applications for Accelerating Industrial Internet of Things](https://mlsanity.com/paper/2104.10501/) (2021)
- [FedParking: A Federated Learning based Parking Space Estimation with Parked Vehicle assisted Edge Computing](https://mlsanity.com/paper/2110.12876/) (2021)
- [Deep Learning for Time Series Forecasting: A Survey](https://mlsanity.com/paper/2503.10198/) (2025)
- [Asynchronous Decentralized Federated Learning for Collaborative Fault Diagnosis of PV Stations](https://mlsanity.com/paper/2202.13606/) (2022)
- [Enabling Resource-efficient AIoT System with Cross-level Optimization: A survey](https://mlsanity.com/paper/2309.15467/) (2023)
- [A Comprehensive Survey of Federated Transfer Learning: Challenges, Methods and Applications](https://mlsanity.com/paper/2403.01387/) (2024)
- [Personalized Federated Learning for Spatio-Temporal Forecasting: A Dual Semantic Alignment-Based Contrastive Approach](https://mlsanity.com/paper/2404.03702/) (2024)
- [Robust Semi-supervised Federated Learning for Images Automatic Recognition in Internet of Drones](https://mlsanity.com/paper/2201.01230/) (2022)
- [A Systematic Literature Review on Blockchain Enabled Federated Learning Framework for Internet of Vehicles](https://mlsanity.com/paper/2203.05192/) (2022)
- [SemiPFL: Personalized Semi-Supervised Federated Learning Framework for Edge Intelligence](https://mlsanity.com/paper/2203.08176/) (2022)
- [Online Spatio-Temporal Correlation-Based Federated Learning for Traffic Flow Forecasting](https://mlsanity.com/paper/2302.08658/) (2023)
- [Advancements in Federated Learning: Models, Methods, and Privacy](https://mlsanity.com/paper/2302.11466/) (2023)
- [Federated Learning for Big Data: A Survey on Opportunities, Applications, and Future Directions](https://mlsanity.com/paper/2110.04160/) (2021)

## Abstract (arXiv)

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.