A traffic flow prediction framework based on integrated federated learning and Recurrent Long short-term networks

被引:1
作者
Pulligilla, Manoj Kumar [1 ]
Vanmathi, C. [1 ]
机构
[1] Vellore Inst Technol, Sch Comp Sci Engn & Informat Syst, Vellore, Tamil Nadu, India
关键词
Traffic flow prediction; Recurrent long short-term capture network; Federated learning; Spatio and temporal information; Edge computing; Traffic congestion; Intelligent transportation systems; SPEEDS;
D O I
10.1007/s12083-024-01792-x
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
For smart cities, predicting traffic flow is crucial to lower traffic jams and enhancing transportation efficiency. The smart city needs effective models, highly dependable networks, and data privacy for traffic flow prediction (Traff-FP). The majority of current research uses a central training mode and ignores privacy issue conveyed by distributed traffic data. In this paper, an effective traffic flow prediction (ETraff-FP) is proposed to forecast traffic flow using actual historical traffic data. Initially, pre-processing is carried out using data normalization and handling missing value. The three major components of Traff-FP framework for each local Traff-FP model are recurrent long short-term capture network (RLSCN), federated gated graph attentive network (FGAN) and semantic connection relationship capture network (SCRCN). The long-term spatio and temporal information in each location has been captured by RLSCN, which encompasses constituents like fully connected (FC) layers, convolution, and bidirectional long short term memory (BiLSTM) to collect short-term information. FGAN, which incorporates bi-directional gated recurrent unit (Bi-GRU), exchanges short-term spatio-temporal hidden information while it trains local Traff-FP model using elliptic curve diffie-hellman (ECDiff-H) algorithm. Accordingly, the hyper parameters of ETraff-FP are tuned using extended remora optimization algorithm (EReOA). The ETraff-FP framework is trained and tested with TaxiNYC and TaxiBJ datasets. For simulation, python platform is utilized and various evaluation metrics are analysed. Accordingly, the ETraff-FP framework has reached better improvements with MSE of 8.98% and 10.57%, RMSE of 8.62% and 18.65%, MAE of 2.11% and 10.57%, R2-score of 0.959% and 0.913%, and MAPE of 21.12% and 24.89% against the existing methods using TaxiNYC and TaxiBJ datasets. Overall, the proposed work not only advances the state-of-the-art in traffic flow prediction but also proves the value of enabling effective and efficient traffic management systems in urban and smart city environments.
引用
收藏
页码:4131 / 4155
页数:25
相关论文
共 42 条
[1]   Exploiting dynamic spatio-temporal graph convolutional neural networks for citywide traffic flows prediction [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Zakarya, Muhammad .
NEURAL NETWORKS, 2022, 145 :233-247
[2]   Exploiting dynamic spatio-temporal correlations for citywide traffic flow prediction using attention based neural networks [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Zakarya, Muhammad .
INFORMATION SCIENCES, 2021, 577 :852-870
[3]   A data aggregation based approach to exploit dynamic spatio-temporal correlations for citywide crowd flows prediction in fog computing [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Zakarya, Muhammad .
MULTIMEDIA TOOLS AND APPLICATIONS, 2021, 80 (20) :31401-31433
[4]   Leveraging Spatio-Temporal Patterns for Predicting Citywide Traffic Crowd Flows Using Deep Hybrid Neural Networks [J].
Ali, Ahmad ;
Zhu, Yanmin ;
Chen, Qiuxia ;
Yu, Jiadi ;
Cai, Haibin .
2019 IEEE 25TH INTERNATIONAL CONFERENCE ON PARALLEL AND DISTRIBUTED SYSTEMS (ICPADS), 2019, :125-132
[5]   Cooperative Position Prediction: Beyond Vehicle-to-Vehicle Relative Positioning [J].
Ansari, Keyvan .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2020, 21 (03) :1121-1130
[6]   Modeling Dynamic Spatio-Temporal Correlations for Urban Traffic Flows Prediction [J].
Awan, Nabeela ;
Ali, Ahmad ;
Khan, Fazlullah ;
Zakarya, Muhammad ;
Alturki, Ryan ;
Kundi, Mahwish ;
Alshehri, Mohammad Dahman ;
Haleem, Muhammad .
IEEE ACCESS, 2021, 9 :26502-26511
[7]   Artificial intelligence-based vehicular traffic flow prediction methods for supporting intelligent transportation systems [J].
Boukerche, Azzedine ;
Tao, Yanjie ;
Sun, Peng .
COMPUTER NETWORKS, 2020, 182 (182)
[8]   A Multiscale-Grid-Based Stacked Bidirectional GRU Neural Network Model for Predicting Traffic Speeds of Urban Expressways [J].
Chen, Deqi ;
Yan, Xuedong ;
Liu, Xiaobing ;
Li, Shurong ;
Wang, Liwei ;
Tian, Xinmei .
IEEE ACCESS, 2021, 9 :1321-1337
[9]   XGBoost: A Scalable Tree Boosting System [J].
Chen, Tianqi ;
Guestrin, Carlos .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :785-794
[10]  
Ding Y, 2023, IEEE Internet Things J