Conference Paper2021Cited by 47

SST-GNN: simplified spatio-temporal traffic forecasting model using graph neural network

Amit Roy, Kashob Kumar Roy, Amin Ahsan Ali, M Ashraful Amin, AKM Mahbubur Rahman

Pacific-Asia Conference on Knowledge Discovery and Data Mining

Springer International Publishing, pp. 90–102, ISBN: 9783030757670

  • Sustainable Cities and Communities — Relevance: 63%
DOI: 10.1007/978-3-030-75768-7_8Open AccessURL

Keywords

  • graph neural networks
  • positional encoding
  • traffic prediction
  • temporal dependencies
  • periodic traffic patterns
  • PEMS
  • long-range dependencies
  • street intersections
  • spatial dependence

CCDS Authors

Amit Roy
Amit Roy

Amit Roy

  • Alumni, CCDS

  • PhD Student

    Purdue University, USA

  • BSc

    University of Dhaka, Bangladesh

Published at

PAKDD, IJCNN

Kashob Kumar Roy
Kashob Kumar Roy

Kashob Kumar Roy

  • Alumni, CCDS

  • PhD Student

    University of Illinois Urbana-Champaign, USA

  • BSc

    University of Dhaka, Bangladesh

Published at

PAKDD, IJCNN

Prof. Amin Ahsan Ali, PhD
Amin Ahsan Ali, PhD

Prof. Amin Ahsan Ali, PhD

  • Director, AI/ML Wing
  • Supervisor, ComBio Wing

  • Professor

    Department of Computer Science and Engineering

    Independent University, Bangladesh (IUB)

Published at

ACL, CHI, UbiComp, ECCV, IJCAI, ECAI, WACV, IPSN, PAKDD, LREC, IJCNN, ICPR, ICIP, BEA, BLP, ICMLA, Scientific Reports, Scientific Data, Pattern Recognition, Information Sciences, Ad Hoc Networks, Drug and Alcohol Dependence, PLOS ONE

Prof. M Ashraful Amin, PhD
M Ashraful Amin, PhD

Prof. M Ashraful Amin, PhD

  • Founder Director, CCDS
  • Director, AI/ML, HCI & DS Wings

Published at

ACL, CHI, UbiComp, ECAI, WACV, PAKDD, IJCNN, ICPR, ICIP, HCII, CLEF, BLP, Scientific Reports, Pattern Recognition, JMIR, IEEE Access, Smart Agricultural Technology, Frontiers in Computational Neuroscience, Diabetes Research and Clinical Practice, PLOS ONE

AKM Mahbubur Rahman, PhD
AKM Mahbubur Rhman, PhD

AKM Mahbubur Rahman, PhD

  • Director, DS & IPT Wings

  • Associate Professor

    Department of Computer Science and Engineering

    Independent University, Bangladesh (IUB)

Published at

CHI, ECCV, WACV, PAKDD, LREC, IJCNN, ICPR, ICIP, HCII, BEA, Scientific Reports, Information Sciences

References

  1. 1.Chen, Tianyu, Wang, Zeyu, Li, Lin, Li, Ding, Li, Zongyang, Chang, Xiaoning, Bian, Pan, Liang, Guangtai, Wang, Qianxiang, Xie, Tao. (2025). Detecting Functionality-Specific Vulnerabilities via Retrieving Individual Functionality-Equivalent APIs in Open-Source Repositories. Dagstuhl Research Online Publication Server[10.4230/lipics.ecoop.2025.6]
  2. 2.Michaël Defferrard. (2016). Convolutional Neural Networks On Graphs With Fast Localized Spectral Filtering (Gdl Seminar). Figshare[10.5281/zenodo.1318397]
  3. 3.Bing Yu, Haoteng Yin, Zhanxing Zhu. (2018). Spatio-Temporal Graph Convolutional Networks: A Deep Learning Framework for Traffic Forecasting. , 3634–3640[10.24963/ijcai.2018/505]
  4. 4.Shengnan Guo, Youfang Lin, Ning Feng, Chao Song, Huaiyu Wan. (2019). Attention Based Spatial-Temporal Graph Convolutional Networks for Traffic Flow Forecasting. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 922–929[10.1609/aaai.v33i01.3301922]
  5. 5.Zonghan Wu, Shirui Pan, Guodong Long, Jing Jiang, Chengqi Zhang. (2019). Graph WaveNet for Deep Spatial-Temporal Graph Modeling. , 1907–1913[10.24963/ijcai.2019/264]
  6. 6.Deli Chen, Yankai Lin, Wei Li, Peng Li, Jie Zhou, Xu Sun. (2020). Measuring and Relieving the Over-Smoothing Problem for Graph Neural Networks from the Topological View. Proceedings of the AAAI Conference on Artificial Intelligence, 34(04), 3438–3445[10.1609/aaai.v34i04.5747]
  7. 7.Rongzhou Huang, Chuyin Huang, Yubao Liu, Genan Dai, Weiyang Kong. (2020). LSGCN: Long Short-Term Traffic Prediction with Graph Convolutional Networks. , 2355–2361[10.24963/ijcai.2020/326]
  8. 8.Cen Chen, Kenli Li, Sin G. Teo, Xiaofeng Zou, Kang Wang, Jie Wang, Zeng Zeng. (2019). Gated Residual Recurrent Graph Neural Networks for Traffic Prediction. Proceedings of the AAAI Conference on Artificial Intelligence, 33(01), 485–492[10.1609/aaai.v33i01.3301485]
  9. 9.Shen Fang, Qi Zhang, Gaofeng Meng, Shiming Xiang, Chunhong Pan. (2019). GSTNet: Global Spatial-Temporal Network for Traffic Flow Prediction. , 2286–2293[10.24963/ijcai.2019/317]
  10. 10.Cheonbok Park, Chunggi Lee, Hyojin Bahng, Yunwon Tae, Seungmin Jin, Ki-Hwan Kim, Sungahn Ko, Jaegul Choo. (2020). ST-GRAT: A Novel Spatio-temporal Graph Attention Networks for Accurately Forecasting Dynamically Changing Road Speed. , 1215–1224[10.1145/3340531.3411940]
  11. 11.William L. Hamilton, Rex Ying, Jure Leskovec. (2017). Inductive Representation Learning on Large Graphs. arXiv (Cornell University), 30, 1024–1034[10.48550/arxiv.1706.02216]
  12. 12.Yaguang Li, Rose Yu, Cyrus Shahabi, Yan Liu. (2017). Diffusion Convolutional Recurrent Neural Network: Data-Driven Traffic\n Forecasting. arXiv (Cornell University)[10.48550/arxiv.1707.01926]
  13. 13.Jiong Zhu, Yujun Yan, Lingxiao Zhao, Mark Heimann, Leman Akoglu, Danai Koutra. (2020). Beyond Homophily in Graph Neural Networks: Current Limitations and\n Effective Designs. arXiv (Cornell University)[10.48550/arxiv.2006.11468]

Cited By

  1. 1.Yinxin Bao, Jiashuang Huang, Qinqin Shen, Yang Cao, Weiping Ding, Zhenquan Shi, Zhenquan Shi, Quan Shi. (2023). Spatial–Temporal Complex Graph Convolution Network for Traffic Flow Prediction. Engineering Applications of Artificial Intelligence, 121, 106044[10.1016/j.engappai.2023.106044]
  2. 2.Shiyu Yang, Qunyong Wu. (2025). SDSINet: A spatiotemporal dual-scale interaction network for traffic prediction. Applied Soft Computing, 173, 112892[10.1016/j.asoc.2025.112892]
  3. 3.Shipeng Liu, Xingjian Wang. (2025). An improved transformer based traffic flow prediction model. Scientific Reports, 15(1), 8284[10.1038/s41598-025-92425-7]
  4. 4.Guangrui Fan, Aznul Qalid Md. Sabri, Siti Soraya Abdul Rahman, Lihu Pan, Susanto Rahardja. (2025). Emerging Trends in Graph Neural Networks for Traffic Flow Prediction: A Survey. Archives of Computational Methods in Engineering, 32(8), 4811–4855[10.1007/s11831-025-10286-9]
  5. 5.Xingyu Zhao, Jianpeng Qi, Yanwei Yu, Lei Zhou. (2024). Deep learning for ocean temperature forecasting: a survey. Intelligent Marine Technology and Systems, 2(1)[10.1007/s44295-024-00042-3]
  6. 6.Hanen Balti, Ali Ben Abbes, Yanfang Sang, Nedra Mellouli, Imed Riadh Farah. (2023). Spatio-temporal heterogeneous graph using multivariate earth observation time series: Application for drought forecasting. Computers & Geosciences, 179, 105435[10.1016/j.cageo.2023.105435]
  7. 7.Chuan Qin, Anurag K. Srivastava, Ahmed Y. Saber, Dax Matthews, Kevin Davies. (2023). Geometric Deep-Learning-Based Spatiotemporal Forecasting for Inverter-Based Solar Power. IEEE Systems Journal, 17(3), 3425–3435[10.1109/jsyst.2023.3250403]
  8. 8.Jiahui Chen, Lina Yang, Yi Yang, Ling Peng, Xingtong Ge. (2024). Spatio-temporal graph neural networks for missing data completion in traffic prediction. International Journal of Geographical Information Systems, 39(5), 1057–1075[10.1080/13658816.2024.2381221]
  9. 9.Rudai Shan, Hao Ning, Qianhui Xu, Xuehua Su, Mengjin Guo, Xiaohan Jia. (2025). Physics-Informed and Explainable Graph Neural Networks for Generalizable Urban Building Energy Modeling. Applied Sciences, 15(16), 8854[10.3390/app15168854]
  10. 10.Xiyuan Jin, Jing Wang, Shengnan Guo, Tonglong Wei, Yiji Zhao, Youfang Lin, Huaiyu Wan. (2023). Spatial–temporal uncertainty-aware graph networks for promoting accuracy and reliability of traffic forecasting. Expert Systems with Applications, 238, 122143[10.1016/j.eswa.2023.122143]
  11. 11.Chi Zhang, Hong-Yu Zhou, Qiang Qiu, Zhichun Jian, Daoye Zhu, Chengqi Cheng, Liesong He, Guoping Liu, Xiang Wen, Runbo Hu. (2022). Augmented Multi-Component Recurrent Graph Convolutional Network for Traffic Flow Forecasting. ISPRS International Journal of Geo-Information, 11(2), 88[10.3390/ijgi11020088]
  12. 12.Sungchul Hong, Yunjin Choi, Jong-June Jeon. (2024). Interpretable water level forecaster with spatiotemporal causal attention mechanisms. International Journal of Forecasting, 41(3), 1037–1054[10.1016/j.ijforecast.2024.10.003]
  13. 13.Wencheng Huang, Yanhui Yin, Haoran Li, Anhao Xie, Yuzhou Fan. (2024). High-speed railway express delivery volume forecast based on data-driven ensemble forecast approaches: The China case. Expert Systems with Applications, 258, 125196[10.1016/j.eswa.2024.125196]
  14. 14.Francesco Piccialli, Marzia Canzaniello, Diletta Chiaro, Stefano Izzo, Pian Qi. (2024). GRAPHITE — Generative Reasoning and Analysis for Predictive Handling in Traffic Efficiency. Information Fusion, 106, 102265[10.1016/j.inffus.2024.102265]
  15. 15.Jiaming Ma, Binwu Wang, Pengkun Wang, Zhengyang Zhou, Yudong Zhang, Xu Wang, Yang Wang. (2025). MobiMixer: A Multi-Scale Spatiotemporal Mixing Model for Mobile Traffic Prediction. IEEE Transactions on Mobile Computing, 24(11), 11972–11986[10.1109/tmc.2025.3585007]
  16. 16.Rui Lian, Xin Wang. (2024). GCN-Bi-LSTM: A Neural Network Method Based on Spatiotemporal Features for Short-Term Traffic Flow Prediction. International Conference on Data-driven Optimization of Complex Systems (DOCS), 613–618[10.1109/docs63458.2024.10704288]
  17. 17.Rong Cao, Hongyang Chen, David Z.W. Wang. (2025). An attention-based dynamic graph model for on-street parking availability prediction. Transportation Research Part A Policy and Practice, 193, 104391[10.1016/j.tra.2025.104391]
  18. 18.Zhizhong Tan, Min Hu, Bin Liu, Guosheng Yin. (2024). Futures Quantitative Investment With Heterogeneous Continual Graph Neural Network. Proceedings, 851–856[10.1109/icdm59182.2024.00104]
  19. 19.Yu Liu, Lianzhen Zhang, Jiyu Xin, Sijie Peng. (2025). TPS-GNN: Predictive model for bridge health monitoring data based on temporal periodicity and global spatial distribution characteristics. Structural Health Monitoring, 25(5), 2925–2943[10.1177/14759217251330971]
  20. 20.Yuan Yao, Linlong Chen, Xianchen Wang, Xiaojun Wu. (2025). Traffic flow forecasting based on augmented multi-component recurrent graph attention network. Transportation Letters, 17(8), 1390–1398[10.1080/19427867.2025.2450577]
  21. 21.Javier García-Sigüenza, Manuel Curado, Faraon Llorens-Largo, Jose F. Vicent. (2025). Self explainable graph convolutional recurrent network for spatio-temporal forecasting. Machine Learning, 114(1)[10.1007/s10994-024-06725-6]
  22. 22.Weihan Dai, Xiaoyu He, Xiulin Geng, Shuai Zhang, Ziheng Gao. (2025). Sea Surface Temperature Prediction Based on Spatio-Temporal Graph Contrastive Learning Network. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 18, 14228–14239[10.1109/jstars.2025.3571494]
  23. 23.Basi Reddy. A, Jonnavada Jahnavi, J B Tejaswi, K M Asif, Jammulamudi Vishal. (2025). Graph Neural Network–Enhanced Traffic Flow Prediction using Dynamic Road Network Graphs. International Conference on Innovative Data Communication Technologies and Application (ICIDCA), 476–481[10.1109/icidca66325.2025.11280407]
  24. 24.Syed Rizvi, Akash Awasthi, Maria J. Peláez, Zhihui Wang, Vittorio Cristini, Hien Van Nguyen, Prashant Dogra. (2024). Deep learning-derived optimal aviation strategies to control pandemics. Scientific Reports, 14(1), 22926[10.1038/s41598-024-73639-7]
  25. 25.Yifan Wu, Zhiwei Xie, Zaiyang Ma, Min Chen, Fengyuan Zhang, Zhenguo Shi, Wengang Li, Shuaizhi Zhai. (2024). An Object-Oriented Semi-Supervised Land-Use/Land-Cover Change Detection Method Based on Siamese Autoencoder Graph Attention Network. IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, 17, 6478–6500[10.1109/jstars.2024.3373039]
  26. 26.Junwei Zhou, Xizhong Qin, Kun Yu, Zhenhong Jia, Yan Du. (2022). STSGAN: Spatial-Temporal Global Semantic Graph Attention Convolution Networks for Urban Flow Prediction. ISPRS International Journal of Geo-Information, 11(7), 381[10.3390/ijgi11070381]
  27. 27.Xiuwen Li, Weiwei Fang, Liang Qian, Haoyuan Li, Yanming Chen, Neal N. Xiong. (2025). Automated Design for Hardware-Aware Graph Neural Networks on Edge Devices. IEEE Transactions on Network Science and Engineering, 13, 828–840[10.1109/tnse.2025.3587645]
  28. 28.Ching-Hao Fan, Hao Zhou, Yao Sun, Geovanny Palomino Roldan, Olga Kokshagina, Marc Santolini, Lijing Wang. (2025). KEGNN: Knowledge-Enhanced Graph Neural Networks for User Engagement Prediction. International Conference on Multimedia Retrieval, 275–283[10.1145/3731715.3733368]
  29. 29.Min Hu, Zhizhong Tan, Bin Liu, Guosheng Yin. (2025). Graph Portfolio: High-Frequency Factor Predictors via Heterogeneous Continual GNNs. IEEE Transactions on Knowledge and Data Engineering, 1–12[10.1109/tkde.2025.3566111]
  30. 30.Siteng Cai, Gang Liu, Jing He, Yulun Du, Zhichao Si, Yunhao Jiang. (2024). Temporal-Spatial Traffic Flow Prediction Model Based on Prompt Learning. ISPRS International Journal of Geo-Information, 14(1), 11[10.3390/ijgi14010011]
  31. 31.Yanling Zhang, Xiaojie Guo. (2024). Research on Smart City Road Network Capacity Optimization Configuration Based on Deep Learning Algorithms. International Journal of High Speed Electronics and Systems, 34(01)[10.1142/s0129156425400567]
  32. 32.Arun Kumar Garov, A. K. Awasthi, Ram Kumar, Monica Sankat. (2023). Application and Some Fundamental Study of GNN In Forecasting. Advances in systems analysis, software engineering, and high performance computing book series, 132–154[10.4018/978-1-6684-6903-3.ch009]
  33. 33.Seth Ockerman, Amal Gueroudji, Tanwi Mallick, Yixuan He, Line Pouchard, Robert Ross, Shivaram Venkataraman. (2025). PGT-I: Scaling Spatiotemporal GNNs with Memory-Efficient Distributed Training. 2012 International Conference for High Performance Computing, Networking, Storage and Analysis, 217–236[10.1145/3712285.3759880]
  34. 34.Xinyang Zhang, Ming Wen, Jiaxi Li, Hongyi Huang. (2023). Distributed PV Power Forecasting Considering Spatio-temporal Dependence. International Conference on Smart Grid and Smart Cities (ICSGSC), 9, 487–493[10.1109/icsgsc59580.2023.10319128]
  35. 35.Enjie Ye, Kun Guo, Wenzhong Guo, Dangrun Chen, Zihan Zhang, Fuan Li, JiaChen Zheng. (2023). Horizontal Federated Traffic Speed Prediction Base on Secure Node Attribute Aggregation. Communications in computer and information science, 645–659[10.1007/978-981-99-2385-4_49]
  36. 36.Tzu‐Hsien Tsai, Ming‐Yen Chiang, Ding‐Yong Hong, Pangfeng Liu, Jan‐Jan Wu. (2026). Optimizing Pipeline Parallelism for Deep Learning With Activation Checkpointing. Concurrency and Computation Practice and Experience, 38(19)[10.1002/cpe.70975]
  37. 37.Yiming Lei, Qiannan Shen, Junhao Song. (2026). A multi-scale graph learning framework with temporal consistency constraints for financial fraud detection in transaction networks under non-stationary conditions. International Journal of Data Science and Analytics, 22(1)[10.1007/s41060-026-01256-w]
  38. 38.Qianru Zhang, Xinyi Gao, Alexander Zhou, Reynold Cheng, Siu-Ming Yiu, Hongzhi Yin. (2026). Efficient prompt learning for traffic forecasting. The VLDB Journal, 35(5)[10.1007/s00778-026-00983-7]
  39. 39.Mohammadreza Akbari Lor, Shu-Ching Chen, Mei-Ling Shyu. (2026). CAP-STN: Continuous Adaptive Spatial Propagation for Large-Scale Spatio-Temporal Forecasting. IEEE International Conference on Information Reuse and Integration for Data Science (IRI), 98–103[10.1109/iri69576.2026.00029]
  40. 40.Tianze Deng, Zengni Zhang, Kai Zhang, Yuhan Dong. (2026). PVDSF: A Photovoltaic Generation Forecasting Network With Dynamic-Static Correlation Fusion on Endogenous and Exogenous Variables. IEEE Transactions on Industrial Informatics, 22(8), 6941–6952[10.1109/tii.2026.3681961]
  41. 41.Yi Li, MingZe Zhang, BoLin Chen, WenYuan Wu, MingQuan Shi. (2026). Regional Bus Travel Time Prediction Using Graph Neural Networks. Journal of Transportation Engineering Part A Systems, 152(4)[10.1061/jtepbs.teeng-9006]
  42. 42.Chang Liu, Yonghao Wu. (2026). STUAformer: Spatio-Temporal Unified Aggregate Transformer Model for Traffic Speed Prediction. Communications in computer and information science, 127–143[10.1007/978-981-95-5294-8_10]
  43. 43.Ming-Yen Chiang, Tzu-Hsien Tsai, Ding-Yong Hong, Pangfeng Liu, Jan-Jan Wu. (2025). Optimizing Pipeline Parallelism for Deep Learning with Activation Checkpointing. , 20–28[10.1109/candar68384.2025.00011]
  44. 44.Benedek Máth, Ádám Nemes, Balázs Varga. (2025). Iterative GNN-Based Traffic Flow Prediction and Sensor Placement Optimization. IEEE International Symposium on Applied Computational Intelligence and Informatics (SACI), 1–6[10.1109/saci66288.2025.11030106]
  45. 45.Santhosh Sachin, Varun Sankara Narayanan, Madhiraju Sai Trisha, K. Harshavardhan Reddy, Anbazhagan Mahadevan. (2025). AGNet: An Adaptive Encoder-Decoder GNN for Spatio-Temporal Forecasting. Lecture notes in networks and systems, 131–149[10.1007/978-3-032-05548-4_11]
  46. 46.Lei Chang, Wenxi Yang, Kaiyuan Qi, Tao Cui, Lianfei Yu, Zhijian Qu, Chongguang Ren. (2024). Multiscale Spatio-Temporal Aware Graph Recurrent Neural Network for Traffic Prediction. Research Square[10.21203/rs.3.rs-5257185/v1]
  47. 47.Ritwik Ojha, Harshit Poddar, Aparna Mohanty, Vinay Jadhav, Aarthi Gunasekar, Gerardine Immaculate Mary, Vaibhav Kejriwal, Rajesh M Kumar. (2023). Comparative Analysis of Two Graph Neural Network-Based Approaches to Traffic Speed Forecasting. International Conference on Next Generation Electronics (NELEX), 1–6[10.1109/nelex59773.2023.10420996]
← All publications