A Hybrid Spatiotemporal Framework with Memory and Diffusion Convolution for Traffic Flow Prediction
DOI:
https://doi.org/10.65455/6rkv7629关键词:
Traffic Flow Prediction, Spatiotemporal Forecasting, Memory-Augmented Transformer, Diffusion Convolution摘要
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.
参考
[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
##submission.downloads##
已出版
##submission.dataAvailability##
The data that support the findings of this study are available from the corresponding author upon reasonable request.
##submission.license##
##submission.copyrightStatement##
##submission.license.cc.by4.footer##