A Hybrid Spatiotemporal Framework with Memory and Diffusion Convolution for Traffic Flow Prediction
DOI:
https://doi.org/10.65455/6rkv7629Keywords:
Traffic Flow Prediction, Spatiotemporal Forecasting, Memory-Augmented Transformer, Diffusion ConvolutionAbstract
Accurate traffic flow prediction is pivotal for intelligent transportation systems, yet it remains inherently challenging due to dynamic spatial correlations and long-range temporal dependencies. While existing forecasting paradigms predominantly rely on static, pre-defined graph structures, they often overlook the direct functional connections between non-adjacent time steps.This paper proposes a novel spatiotemporal forecasting framework, termed Memory-augmented Diffusion Convolutional LSTM network (MDC-LSTM), which integrates a MemBART-inspired memory mechanism, diffusion convolution, regional attention, and LSTM-based temporal modeling. By synthesizing these components with Long Short-Term Memory (LSTM) units, the proposed model effectively captures both localized spatial patterns and deep inter-temporal relationships. Empirical evaluations conducted on the METR-LA benchmark dataset demonstrate that our framework significantly enhances predictive accuracy across multiple metrics, consistently outperforming several state-of-the-art (SOTA) baselines and exhibiting strong robustness in long-term forecasting scenarios.
References
[1]WANG X. A Case Study of the Influence of Multifarious Factors on Traffic Flow Forecasting. Applied Artificial Intelligence Research, 2026, 2(2). DOI: https://doi.org/10.65455/sbd3t486
[2]VLAHOGIANNI E I, KARLAFTIS M G, GOLIAS J C. Short-term traffic forecasting: Where we are and where we are going. Transportation Research Part C: Emerging Technologies, 2014, 43: 3-19. DOI: https://doi.org/10.1016/j.trc.2014.01.005
[3]LV Y, DUAN Y, KANG W, et al. Traffic flow prediction with big data: A deep learning approach. IEEE Transactions on Intelligent Transportation Systems, 2015, 16(2): 865-873.
[4]CHANDRA S R, AL-DEEK H. Predictions of freeway traffic speeds and volumes using vector autoregressive models. Journal of Intelligent Transportation Systems, 2009, 13(2): 53-72. DOI: https://doi.org/10.1080/15472450902858368
[5]HOCHREITER S, SCHMIDHUBER J. Long short-term memory. Neural Computation, 1997, 9(8): 1735-1780. DOI: https://doi.org/10.1162/neco.1997.9.8.1735
[6]LI Y, YU R, SHAHABI C, et al. Diffusion convolutional recurrent neural network: Data-driven traffic forecasting//Proceedings of the International Conference on Learning Representations, 2018.
[7]YU B, YIN H, ZHU Z. Spatio-temporal graph convolutional networks: A deep learning framework for traffic forecasting//Proceedings of the 27th International Joint Conference on Artificial Intelligence, 2018: 3634-3640. DOI: https://doi.org/10.24963/ijcai.2018/505
[8]WU Z, PAN S, LONG G, et al. Graph WaveNet for deep spatial-temporal graph modeling//Proceedings of the 28th International Joint Conference on Artificial Intelligence, 2019: 1907-1913. DOI: https://doi.org/10.24963/ijcai.2019/264
[9]GUO S, LIN Y, FENG N, et al. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting//Proceedings of the AAAI Conference on Artificial Intelligence, 2019: 922-929. DOI: https://doi.org/10.1609/aaai.v33i01.3301922
[10]WU Z, PAN S, LONG G, et al. Connecting the dots: Multivariate time series forecasting with graph neural networks//Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, 2020: 753-763. DOI: https://doi.org/10.1145/3394486.3403118
[11]BAI L, YAO L, LI C, et al. Adaptive graph convolutional recurrent network for traffic forecasting//Advances in Neural Information Processing Systems, 2020: 17804-17815.
[12]SHAO Z, ZHANG Z, WEI W, et al. Decoupled dynamic spatial-temporal graph neural network for traffic forecasting. Proceedings of the VLDB Endowment, 2022, 15(11): 2733-2746. DOI: https://doi.org/10.14778/3551793.3551827
[13]VASWANI A, SHAZEER N, PARMAR N, et al. Attention is all you need//Advances in Neural Information Processing Systems, 2017: 5998-6008.
[14]ZHOU H, ZHANG S, PENG J, et al. Informer: Beyond efficient Transformer for long sequence time-series forecasting//Proceedings of the AAAI Conference on Artificial Intelligence, 2021: 11106-11115. DOI: https://doi.org/10.1609/aaai.v35i12.17325
[15]WU H, XU J, WANG J, et al. Autoformer: Decomposition Transformers with auto-correlation for long-term series forecasting//Advances in Neural Information Processing Systems, 2021: 22419-22430.
[16]ZHOU T, MA Z, WEN Q, et al. FEDformer: Frequency enhanced decomposed Transformer for long-term series forecasting//Proceedings of the 39th International Conference on Machine Learning, PMLR, 2022: 27268-27286.
[17]ZHENG C, FAN X, WANG C, et al. GMAN: A graph multi-attention network for traffic prediction//Proceedings of the AAAI Conference on Artificial Intelligence, 2020: 1234-1241. DOI: https://doi.org/10.1609/aaai.v34i01.5477
[18]JIANG J, HAN C, ZHAO W X, et al. PDFormer: Propagation delay-aware dynamic long-range Transformer for traffic flow prediction//Proceedings of the AAAI Conference on Artificial Intelligence, 2023: 4365-4373. DOI: https://doi.org/10.1609/aaai.v37i4.25556
[19]LIU H, DONG Z, JIANG R, et al. Spatio-temporal adaptive embedding makes vanilla Transformer SOTA for traffic forecasting//Proceedings of the 32nd ACM International Conference on Information and Knowledge Management, 2023: 4125-4129. DOI: https://doi.org/10.1145/3583780.3615160
[20]DAI Z, YANG Z, YANG Y, et al. Transformer-XL: Attentive language models beyond a fixed-length context//Proceedings of the 57th Annual Meeting of the Association for Computational Linguistics, 2019: 2978-2988. DOI: https://doi.org/10.18653/v1/P19-1285
[21]WU A, SZEGEDY C, MISRA D, et al. Memorizing Transformers//Proceedings of the International Conference on Learning Representations, 2022.
[22]LEWIS M, LIU Y, GOYAL N, et al. BART: Denoising sequence-to-sequence pre-training for natural language generation, translation, and comprehension//Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics, 2020: 7871-7880. DOI: https://doi.org/10.18653/v1/2020.acl-main.703
Downloads
Published
Data Availability Statement
The data that support the findings of this study are available from the corresponding author upon reasonable request.
Issue
Section
License
Copyright (c) 2026 The Author(s). Applied Artificial Intelligence Research published by CSTDP
This work is licensed under a Creative Commons Attribution 4.0 International License.
