Skip to content
Open access

FedTraffic: A Hierarchical Federated Learning Framework for Traffic Flow Prediction in Intelligent Transportation Systems

Jul 2026 · Engineer · 0 citations · 24 references

TL;DR

FedTraffic, a hierarchical federated learning framework for traffic flow forecasting that integrates Edge–Fog–Cloud computing, hybrid deep learning, adaptive federated optimization, and Explainable Artificial Intelligence, is proposed.

Abstract

The rapid growth of Intelligent Transportation Systems (ITSs) and Internet of Things (IoT) technologies has generated massive volumes of distributed traffic data, creating significant challenges related to privacy, scalability, communication overhead, and heterogeneous traffic patterns. To address these challenges, this paper proposes FedTraffic, a hierarchical federated learning framework for traffic flow forecasting that integrates Edge–Fog–Cloud computing, hybrid deep learning, adaptive federated optimization, and Explainable Artificial Intelligence (XAI). The proposed framework combines a Temporal Convolutional Network–Conditional Variational Autoencoder (TCN–CVAE) with traffic-behavior clustering, adaptive client selection, and hierarchical model aggregation to enable accurate, privacy-preserving, and interpretable traffic prediction under heterogeneous non-IID environments. Extensive experiments demonstrate that FedTraffic achieves a best Mean Absolute Error (MAE) of 2.12, a Root Mean Square Error (RMSE) of 4.28, a Mean Absolute Percentage Error (MAPE) of 5.47%, and an R2 score of 0.966. Compared with the strongest federated baseline, it improves MAE by up to 18.77%, RMSE by 16.41%, and MAPE by more than 22%, while reducing communication overhead through an 8:1 latent representation compression ratio. These results demonstrate the effectiveness of FedTraffic as a scalable, privacy-preserving, and interpretable solution for next-generation intelligent transportation systems.

Read PDF

Similar papers

FedTP: A Federated Learning Framework for Traffic Prediction

FedTP is proposed, a federated learning framework that integrates gradient conflict elimination into the aggregation process and harmonizes local updates, thereby improving fairness across clients without compromising overall predictive accuracy.

Baobao Chai, Zhongyuan Yu, Tianqing He et al. · 0 citations
Aug 2026

Traffic congestion management using attention based federated hybrid deep learning model in Vehicle-to-Everything (V2X) systems

A hybrid deep learning framework that integrates Bidirectional Long Short-Term Memory and Gated Recurrent Unit networks with an attention mechanism within a federated learning paradigm is proposed, which enables decentralized model training across multiple data sources without requiring raw data sharing, thereby preserving privacy while maintaining predictive performance.

Apurba Nandi, Shaoni Banerjee, Avik Kumar Das et al. · 0 citations
Open access Aug 2026

Federated Learning Approach for Multi-Regional Traffic Flow Prediction

Experiments on controlled synthetic data and the Q-Traffic real-world dataset demonstrate that the proposed framework improves predictive accuracy, cross-client stability, and robustness under heterogeneous federated traffic scenarios.

Zhi-Cheng Wang, Tao Zhang, Yi-Meng Zhu et al. · 0 citations
Open access 2026

Hierarchical Graph Federated Learning for Cross-Domain Intelligent Transportation Networks

Experiments demonstrate that HG-FL achieves superior performance compared to centralized, flat federated, and graph-only baselines, and highlight the framework’s scalability, robustness, and effectiveness in achieving cross-domain generalization and service-level assurance within privacy-preserving ITS environments.

Mohammed Saad Javeed, Jannatul Maua, Md Shafiq Ullah et al. · 0 citations
Conference Jul 2026

Online Federated Learning on Resource-Limited IoT Devices for Traffic Flow Prediction in Smart Mobility Ecosystems

Traffic congestion significantly impacts safety and urban livability in smart cities, motivating the development of accurate Traffic Flow Prediction (TFP) systems. Traditional deep learning approaches typically rely on centralized training, which is difficult to scale in distributed Internet of Things (IoT) environments. To address these limitations, decentralized paradigms such as Local Learning (LL) and Federated Learning (FL) enable on-device training and collaborative model updates while preserving data locality. For real-time TFP, the inherently non-stationary nature of traffic data necessitates continuous model adaptation, making online federated learning essential for scalable and collaborative deployment. However, this setting remains challenging because traffic data are typically non-IID across clients, with local patterns varying significantly across locations and devices. This paper presents an exploratory study of online federated learning for TFP on resource-constrained IoT devices. Using a GRU-based network as a common backbone, the Online Federated Learning paradigm is benchmarked relative to Centralized and Local Learning as reference baselines. A performance evaluation was conducted by evaluating RMSE and MAE on the PEMS-BAY dataset. Robustness to non-IID data is further assessed using FedProx and SCAFFOLD. Results show that LL achieves the lowest prediction error, whereas FL degrades as the number of local epochs increases, and non-IID mitigation strategies provide limited improvements under low-latency constraints. Overall, online federated learning is a viable approach for real-time TFP, but its performance is highly sensitive to client heterogeneity.

M. Pizzolante, A. Shumba, T. Montanaro et al. · 0 citations