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 条
[21]   Online Incremental Machine Learning Platform for Big Data-Driven Smart Traffic Management [J].
Nallaperuma, Dinithi ;
Nawaratne, Rashmika ;
Bandaragoda, Tharindu ;
Adikari, Achini ;
Su Nguyen ;
Kempitiya, Thimal ;
De Silva, Daswin ;
Alahakoon, Damminda ;
Pothuhera, Dakshan .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2019, 20 (12) :4679-4690
[22]   Privacy-preserving blockchain-based federated learning for traffic flow prediction [J].
Qi, Yuanhang ;
Hossain, M. Shamim ;
Nie, Jiangtian ;
Li, Xuandi .
FUTURE GENERATION COMPUTER SYSTEMS-THE INTERNATIONAL JOURNAL OF ESCIENCE, 2021, 117 :328-337
[23]   Cybersecurity in Big Data Era: From Securing Big Data to Data-Driven Security [J].
Rawat, Danda B. ;
Doku, Ronald ;
Garuba, Moses .
IEEE TRANSACTIONS ON SERVICES COMPUTING, 2021, 14 (06) :2055-2072
[24]  
Sepasgozar SS, 2022, IEEE ACCESS, V10, P119607, DOI 10.1109/ACCESS.2022.3221970
[25]  
Shi XJ, 2015, ADV NEUR IN, V28
[26]   Digital Twins-Based Automated Pilot for Energy-Efficiency Assessment of Intelligent Transportation Infrastructure [J].
Tu, Zhen ;
Qiao, Liang ;
Nowak, Robert ;
Lv, Haibin ;
Lv, Zhihan .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2022, 23 (11) :22320-22330
[27]   STTF: An Efficient Transformer Model for Traffic Congestion Prediction [J].
Wang, Xing ;
Zeng, Ruihao ;
Zou, Fumin ;
Liao, Lyuchao ;
Huang, Faliang .
INTERNATIONAL JOURNAL OF COMPUTATIONAL INTELLIGENCE SYSTEMS, 2023, 16 (01)
[28]   RSAB-ConvGRU: A hybrid deep-learning method for traffic flow prediction [J].
Xia, Dawen ;
Chen, Yan ;
Zhang, Wenyong ;
Hu, Yang ;
Li, Yantao ;
Li, Huaqing .
MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (07) :20559-20585
[29]   Short-Term Traffic Flow Prediction Based on Graph Convolutional Networks and Federated Learning [J].
Xia, Mengran ;
Jin, Dawei ;
Chen, Jingyu .
IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (01) :1191-1203
[30]   An Abnormal Traffic Detection Method for IoT Devices Based on Federated Learning and Depthwise Separable Convolutional Neural Networks [J].
Xia, Qinyu ;
Dong, Shi ;
Peng, Tao .
2022 IEEE INTERNATIONAL PERFORMANCE, COMPUTING, AND COMMUNICATIONS CONFERENCE, IPCCC, 2022,