Federated Learning for Traffic Flow Prediction with Synthetic Data Augmentation

2024-12-11 · Fermin Orozco, Pedro Porto Buarque de Gusmão, Hongkai Wen, Johan Wahlström, Man Luo

Deep-learning based traffic prediction models require vast amounts of data to learn embedded spatial and temporal dependencies. The inherent privacy and commercial sensitivity of such data has encouraged a shift towards decentralised data-driven methods, such as Federated Learning (FL). Under a traditional Machine Learning paradigm, traffic flow prediction models can capture spatial and temporal relationships within centralised data. In reality, traffic data is likely distributed across separate data silos owned by multiple stakeholders. In this work, a cross-silo FL setting is motivated to facilitate stakeholder collaboration for optimal traffic flow prediction applications. This work introduces an FL framework, referred to as FedTPS, to generate synthetic data to augment each client's local dataset by training a diffusion-based trajectory generation model through FL. The proposed framework is evaluated on a large-scale real world ride-sharing dataset using various FL methods and Traffic Flow Prediction models, including a novel prediction model we introduce, which leverages Temporal and Graph Attention mechanisms to learn the Spatio-Temporal dependencies embedded within regional traffic flow data. Experimental results show that FedTPS outperforms multiple other FL baselines with respect to global model performance.

📄 PDF Abstract BibTeX arXiv:2412.08460

Code (0)

등록된 구현이 없습니다.

Tasks

Data Augmentation Federated LearningGraph AttentionPrediction Traffic Prediction

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…

Attention 설명 없음

Privacy-preserving Traffic Flow Prediction: A Federated Learning Approach

2020-03-19 · Yi Liu, James J. Q. Yu, Jiawen Kang, Dusit Niyato 외

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 us…

Clustering Federated Learning Prediction Privacy Preserving+1

Predicting Traffic Flow with Federated Learning and Graph Neural with Asynchronous Computations Network

2024-01-05 · Muhammad Yaqub, Shahzad Ahmad, Malik Abdul Manan, Imran Shabir Chuhan

Real-time traffic flow prediction holds significant importance within the domain of Intelligent Transportation Systems (ITS). The task of achieving a balance between prediction precision and computational efficiency pres…

Computational Efficiency Federated Learning Prediction

Federated Learning with Graph-Based Aggregation for Traffic Forecasting

2025-07-13 · Audri Banik, Glaucio Haroldo Silva de Carvalho, Renata Dividino

In traffic prediction, the goal is to estimate traffic speed or flow in specific regions or road segments using historical data collected by devices deployed in each area. Each region or road segment can be viewed as an …

Federated Learning Graph Learning Traffic Prediction

B^2SFL: A Bi-level Blockchained Architecture for Secure Federated Learning-based Traffic Prediction

2023-10-23 · Hao Guo, Collin Meese, Wanxin Li, Chien-Chung Shen 외

Federated Learning (FL) is a privacy-preserving machine learning (ML) technology that enables collaborative training and learning of a global ML model based on aggregating distributed local model updates. However, securi…

Federated Learning Prediction Privacy Preserving Traffic Prediction

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

2023-02-17 · Qingxiang Liu, Sheng Sun, Min Liu, Yuwei Wang 외

Traffic flow forecasting (TFF) is of great importance to the construction of Intelligent Transportation Systems (ITS). To mitigate communication burden and tackle with the problem of privacy leakage aroused by centralize…

Federated Learning Graph Attention