Privacy-preserving Traffic Flow Prediction: A Federated Learning Approach

03/19/2020
by   YI LIU, et al.
0

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 learning models, which confirm that FedGRU can achieve accurate and timely traffic prediction without compromising the privacy and security of raw data.

READ FULL TEXT

page 1

page 2

research
05/28/2023

BFRT: Blockchained Federated Learning for Real-time Traffic Flow Prediction

Accurate real-time traffic flow prediction can be leveraged to relieve t...
research
06/21/2019

Privacy Preserving QoE Modeling using Collaborative Learning

Machine Learning based Quality of Experience (QoE) models potentially su...
research
08/21/2023

Split Learning for Distributed Collaborative Training of Deep Learning Models in Health Informatics

Deep learning continues to rapidly evolve and is now demonstrating remar...
research
10/26/2020

Communication-efficient learning of traffic flow in a network of wireless presence sensors

Current traffic management systems learn global traffic flow models base...
research
05/30/2021

FED-χ^2: Privacy Preserving Federated Correlation Test

In this paper, we propose the first secure federated χ^2-test protocol F...
research
09/03/2019

Energy Demand Prediction with Federated Learning for Electric Vehicle Networks

In this paper, we propose novel approaches using state-of-the-art machin...
research
02/17/2023

Online Spatio-Temporal Correlation-Based Federated Learning for Traffic Flow Forecasting

Traffic flow forecasting (TFF) is of great importance to the constructio...

Please sign up or login with your details

Forgot password? Click here to reset