FedTraffic: A Hierarchical Federated Learning Framework for Traffic Flow Prediction in Intelligent Transportation Systems
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.