Abstract
Traffic flow prediction (TFP) plays a crucial role in intelligent transportation systems. Due to the need to collect data during the prediction process, TFP can face the risk of data leakage. Federated Graph Learning (FGL) is a novel method for protecting graph data privacy, but there are lower performance and Byzantine attacks challenges when applying FGL to TFP. In order to address these challenges, this paper proposes a distributed and secure traffic flow prediction (DS-TFP) framework based on federated graph learning, which includes three methods. Firstly, a New Spatial-Temporal Graph (NSTG) model is applied for the dynamic changes of temporal and spatial and simplifies the encoding procedures. Secondly, graph models are combined with federated method to ensure the protection of local data privacy. Thirdly, the resist Byzantine attacks method which utilizes clients dynamic cross-validation and a voting-based aggregation algorithm, is designed to enhance system security. Finally, extensive experiments are conducted on two datasets. The results show that compared with the baselines, the Fed-NSTG model improved accuracy by an average of 8.15%. The DS-TFP framework can maintain an average prediction accuracy of 87.57% even under the Byzantine attacks.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
Similar content being viewed by others
References
Zhang, H., Shen, T., Wu, F., Yin, M., Yang, H., Wu, C.: Federated graph learning - a position paper. arXiv preprint arXiv:2105.11099 (2021)
Bonawitz, K., Eichner, H., Grieskamp, W., Huba, D., Ingerman, A., Ivanov, V., Roselander, J.: Towards federated learning at scale: System design. In: Proceedings of Machine Learning and Systems, vol. 1, pp. 374-388 (2019)
Zaremba, W., Sutskever, I., Vinyals, O.: Recurrent neural network regularization. arXiv preprint arXiv:1409.2329 (2014)
Hochreiter, S., Schmidhuber, J.: Long short-term memory. Neural Comput. 9, 1735–1780 (1997)
Scarselli, F., Gori, M., Tsoi, A.C., Hagenbuchner, M., Monfardini, G.: The graph neural network model. IEEE Trans. Neural Netw. 20(1), 61–80 (2008)
Bruna, J., Zaremba, W., Szlam, A., LeCun, Y.: Spectral Networks and Locally Connected Networks on Graphs. arXiv preprint arXiv:1312.6203 (2013)
Defferrard, M., Bresson, X., Vandergheynst, P.: Convolutional neural networks on graphs with fast localized spectral filtering. Adv. Neural Inform. Process. Syst., 3844-3852 (2016)
Kipf, T., Welling, M.: Semi-Supervised Classification with Graph Convolutional Networks. arXiv preprint arXiv:1609.02907 (2016)
Yu, T., Yin, H., Zhu, Z.: Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting. arXiv preprint arXiv:1709.04875 (2017)
Guo, S., Lin, Y., Wan, H., Li, X., Cong, G.: Learning dynamics and heterogeneity of spatial-temporal graph data for traffic forecasting. IEEE Trans. Knowl. Data Eng. 34, 5415–5428 (2021)
Meese, C., Chen, H., Asif, S.A., Li, W., Shen, C. C., Nejad, M.: Bfrt: blockchained federated learning for real-time traffic flow prediction. In: IEEE International Symposium on Cluster, Cloud and Internet Computing, pp. 317-326. IEEE, Singapore (2022)
Xia, M., Jin, D., Chen, J.: Short-term traffic flow prediction based on graph convolutional networks and federated learning. IEEE Trans. Intell. Transp. Syst. 24, 1191–1203 (2023)
Dai, Y., Xu, D., Zhang, K., Maharjan, S., Zhang, Y.: Deep reinforcement learning and permissioned blockchain for content caching in vehicular edge computing and networks. IEEE Trans. Veh. Technol. 69(4), 4312–4324 (2020)
Lu, Y., Huang, X., Zhang, K., Maharjan, S., Zhang, Y.: Blockchain empowered asynchronous federated learning for secure data sharing in Internet of vehicles. IEEE Trans. Veh. Technol. 69, 4298–4311 (2020)
McMahan, H.B., Moore, E., Ramage, D., Hampson, S., Arcas, B.A.: Communication-efficient learning of deep networks from decentralized data. In: Artificial Intelligence and Statistics, pp. 1273-1282. PMLR, Lauderdale (2017)
Oord, A.V.D., Dieleman, S., Zen, H., Simonyan, K., Vinyals, O., Graves, A., Kavukcuoglu, K.: Wavenet: A generative model for raw audio. arXiv preprint arXiv:1609.03499 (2016)
Guo, S., Lin, Y., Feng, N., Song, C., Wan, H.: Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. In: Proceedings of the AAAI Conference on Artificial Intelligence, Hawaii, pp. 922-929 (2019)
Shaw, P., Uszkoreit, J., Vaswani, A.: Self-attention with relative position representations. arXiv preprint arXiv:1803.02155 (2018)
Acknowledgments
This work was supported by Natural Science Foundation of Qingdao City under Grant No. 23-2-1-164-zyyd-jch, National Natural Science Foundation of China No. 62172249, Natural Science Foundation of Shandong Province under Grant ZR2023MF082, Qingdao Science and Technology Plan Key Research and Development Project No. 22-3-4-xxgg-10-gx, Open Foundation of the State Key Laboratory under Grant SKLNST-2022-1-11.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2025 The Author(s), under exclusive license to Springer Nature Switzerland AG
About this paper
Cite this paper
Fu, Y., Sun, L., Ma, Z., Gao, H., Yan, C. (2025). DS-TFP: A Distributed and Secure Traffic Flow Prediction Framework Based on Federated Graph Learning. In: Cai, Z., Takabi, D., Guo, S., Zou, Y. (eds) Wireless Artificial Intelligent Computing Systems and Applications. WASA 2024. Lecture Notes in Computer Science, vol 14998. Springer, Cham. https://doi.org/10.1007/978-3-031-71467-2_12
Download citation
DOI: https://doi.org/10.1007/978-3-031-71467-2_12
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-031-71466-5
Online ISBN: 978-3-031-71467-2
eBook Packages: Computer ScienceComputer Science (R0)