Computer Science ›› 2026, Vol. 53 ›› Issue (8): 245-256.doi: 10.11896/jsjkx.250900014

• Artificial Intelligence • Previous Articles     Next Articles

Leveraging Multi-source Contextual Knowledge-enhanced Graph for Traffic Forecasting

QUAN Jingtao, ZHANG Lei, LIU Bailong, WANG Feifan   

  1. Engineering Research Center of Mine Digitalization of Ministry of Education, China University of Mining and Technology, Xuzhou, Jiangsu 221116, China
    School of Computer Science and Technology, China University of Mining and Technology, Xuzhou, Jiangsu 221116, China
  • Received:2025-09-01 Revised:2025-12-18 Online:2026-08-15 Published:2026-08-17
  • About author:QUAN Jingtao,born in 2001,postgra-duate.His main research interests include knowledge graph and traffic prediction.
    ZHANG Lei,born in 1977,associate professor,is a member of CCF(No.12434S).His main research interest is spatio-temporal data mining.
  • Supported by:
    Postgraduate Innovation Program of China Universityof Mining and Technology(2024WLJCRCZL253) and Postgraduate Research & Practice Innovation Program of Jiangsu Province(SJCX24_1389).

Abstract: Traffic prediction aims to forecast future traffic conditions based on historical traffic conditions and is a crucial research topic in the field of intelligent transportation systems.Although existing traffic prediction models have achieved considerable progress in modeling complex spatio-temporal patterns,they still heavily rely on advanced deep learning techniques and exhibit limitations in representing multi-source contextual knowledge and integrating global graph-topological information.Consequently,they fail to fully capture the diverse context about external environments and internal networks,which constrains their perfor-mance.To address these issues,this paper proposes MCK-GWN(Multi-source Contextual Knowledge-enhanced Graph WaveNet),a graph-based traffic prediction model enhanced by multi-source contextual knowledgegraph.In terms of multi-source knowledge representation,a multi-source contextual knowledge graph is constructed to enrich the POI(Point of Interest) context of traffic nodes and the semantic context between nodes,effectively characterizing both external and internal contextual knowledge.In terms of multi-source knowledge fusion,the POI heterogeneity-aware unit focuses on modeling external environmental attributes,while the semantic path-aware unit focuses on internal network semantic relations,thereby enhancing the integration capability of global graph-topological information.Experimental results on the SZ-TAXI dataset demonstrate that MCK-GWN achieves the best performance across all evaluation metrics,reducing the mean absolute error(MAE) by 3.15% compared with the state-of-the-art baseline model KMHNet.

Key words: Traffic prediction, Knowledge graph, Graph neural network, External factor fusion

CLC Number: 

  • TP391
[1] LUO X,ZHU C,ZHANG D,et al.Stg4traffic:A survey andbenchmark of spatial-temporal graph neural networks for traffic prediction[J].arXiv:2307.00495,2023.
[2] YIN X,WU G,WEI J,et al.Deep learning on traffic prediction:Methods,analysis,and future directions[J].IEEE Transactions on Intelligent Transportation Systems,2021,23(6):4927-4943.
[3] SAYED S A,ABDEL-HAMID Y,HEFNY H A.Artificial intelligence-based traffic flow prediction:a comprehensive review[J].Journal of Electrical Systems and Information Technology,2023,10(1):13.
[4] WILLIAMS B M,HOEL L A.Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process:Theoretical basis and empirical results[J].Journal of Transportation Engineering,2003,129(6):664-672.
[5] OKUTANI I,STEPHANEDES Y J.Dynamic prediction of traffic volume through Kalman filtering theory[J].Transportation Research Part B:Methodological,1984,18(1):1-11.
[6] WU C H,HO J M,LEE D T.Travel-time prediction with support vector regression[J].IEEE Transactions on Intelligent Transportation Systems,2004,5(4):276-281.
[7] TEDJOPURNOMO D A,BAO Z,ZHENG B,et al.A survey on modern deep neural network for traffic prediction:Trends,methods and challenges[C]//2023 IEEE 39th International Conference on Data Engineering(ICDE).IEEE,2023:3795-3796.
[8] SHAO Z,WANG F,XU Y,et al.Exploring progress in multivariate time series forecasting:Comprehensive benchmarking and heterogeneity analysis[J].IEEE Transactions on Knowledge and Data Engineering,2024,37(1):291-305.
[9] CHEN M,GAN K,LI K,et al.A Spatial-temporal Graph Convolutional Network Model for Accurate and Robust Traffic Flow Prediction Based on Real-time Multimodal Spatial-temporal Data[J].Journal of Highway and Transportation Research and Development,2021,38(8):134-139,158.
[10] ZHANG X,JIANG Z,LOU P.Triple Dynamic Graph Convolutional Recurrent Network for Traffic Prediction[J].IEEE Transactions on Intelligent Transportation Systems,2025,26(8):12647-12660.
[11] LIU H,DONG Z,JIANG R,et al.Staeformer:Spatio-temporal adaptive embedding makes vanilla transformer SOTA for traffic forecasting[J].arXiv:2308.10425,2023.
[12] SHUAI S,KONG X,LIU L,et al.Hypernetwork-Enhanced Hierarchical Federated Learning for Long-Term Traffic Prediction with Transformer[J].ACM Transactions on Intelligent Systems and Technology,2025,17(3):1-24.
[13] ALI A,ULLAH I,AHMAD S,et al.An attention-driven spatio-temporal deep hybrid neural networks for traffic flow prediction in transportation systems[J].IEEE Transactions on Intelligent Transportation Systems,2025,26(9):14154-14168.
[14] ZHU J,WANG Q,TAO C,et al.AST-GCN:Attribute-augmented spatiotemporal graph convolutional network for traffic forecasting[J].IEEE Access,2021,9:35973-35983.
[15] ZHANG Y,ZHAO T,GAO S,et al.Incorporating multimodal context information into traffic speed forecasting through graph deep learning[J].International Journal of Geographical Information Science,2023,37(9):1909-1935.
[16] ZHU J,HAN X,DENG H,et al.KST-GCN:A knowledge-driven spatial-temporal graph convolutional network for traffic forecasting[J].IEEE Transactions on Intelligent Transportation Systems,2022,23(9):15055-15065.
[17] HAN X,ZHANG X,WU Y,et al.Knowledge-based multiple relations modeling for traffic forecasting[J].IEEE Transactions on Intelligent Transportation Systems,2024,25(9):11844-11857.
[18] WANG S,LV Y,PENG Y,et al.Metro traffic flow prediction via knowledge graph and spatiotemporal graph neural network[J].Journal of Advanced Transportation,2022,2022(1):2348375.
[19] ZHOU Y,LIU Y,NING N,et al.Integrating knowledge representation into traffic prediction:a spatial-temporal graph neural network with adaptive fusion features[J].Complex & Intelligent Systems,2024,10(2):2883-2900.
[20] XIONG H,SHEN G,LAN X,et al.Hit-gcn:spatial-temporal graph convolutional network embedded with heterogeneous information of road network for traffic forecasting[J].Electronics,2023,12(6):1306.
[21] NING Y,LIU H,WANG H,et al.UUKG:unified urban knowledge graph dataset for urban spatiotemporal prediction[J].Advances in Neural Information Processing Systems,2023,36:62442-62456.
[22] ZHANG Y,WANG Y,GAO S,et al.Context-aware knowledge graph framework for traffic speed forecasting using graph neural network[J].IEEE Transactions on Intelligent Transportation Systems,2024,26(3):3885-3902.
[23] WU Z,PAN S,LONG G,et alGraph wavenet for deep spatial-temporal graph modeling[J].arXiv:1906.00121,2019.
[24] LIU Z,DU W,YAN D,et al.Short-term traffic flow forecasting based on combination of k-nearest neighbor and support vector regression[J].Journal of Highway and Transportation Research and Development(English Edition),2018,12(1):89-96.
[25] YU B,YIN H,ZHU Z.Spatio-temporal graph convolutional networks:A deep learning framework for traffic forecasting[J].arXiv:1709.04875,2017.
[26] SONG C,LIN Y,GUO S,et al.Spatial-temporal synchronous graph convolutional networks:A new framework for spatial-temporal network data forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020:914-921.
[27] ZHANG X,PAN L,SHEN Q,et al.Trend-aware spatio-temporal fusion graph convolutional network with self-attention for traffic prediction[J].Neurocomputing,2025,637:130040.
[28] ZHOU Z,ZHU R,CHEN B,et al.Dynamic Graph Convolutional Traffic Flow Prediction Based on Multi-scale Downsampling Convolutional Interaction[J].Journal of Highway and Transportation Research and Development,2025,42(7):1-12.
[29] GUO S,LIN Y,WAN H,et al.Learning dynamics and heterogeneity of spatial-temporal graph data for traffic forecasting[J].IEEE Transactions on Knowledge and Data Engineering,2021,34(11):5415-5428.
[30] JIANG J,HAN C,ZHAO W X,et al.Pdformer:Propagation delay-aware dynamic long-range transformer for traffic flow prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2023:4365-4373
[31] GUO H F,XU H W,ZHOU Z S.Urban Traffic PredictionBased on Deep Spatio-temporal Hybrid Graph Convolution[J] Journal of Chinese Computer Systems,2025,46(1):97-103.
[32] ZHAO L,SONG Y,ZHANG C,et al.T-GCN:A temporal graph convolutional network for traffic prediction[J].IEEE Transactions on Intelligent Transportation Systems,2019,21(9):3848-3858.
[33] LONG W,XIAO Z,WANG D,et al.Unified spatial-temporal neighbor attention network for dynamic traffic prediction[J].IEEE Transactions on Vehicular Technology,2022,72(2):1515-1529.
[34] LI J,DONG W,GUI X.uTransformer:unified spatial-temporal transformer with external factors for traffic flow forecasting[J].The Journal of Supercomputing,2025,81(1):281.
[35] YANG R,SALIM F D,XUE H.Sstkg:Simple spatio-temporal knowledge graph for intepretable and versatile dynamic information embedding[C]//Proceedings of the ACM Web Conference 2024.2024:551-559.
[36] GONG J,LIU Y,LI T,et al.Empowering spatial knowledge graph for mobile traffic prediction[C]//Proceedings of the 31st ACM International Conference on Advances in Geographic Information Systems.2023:1-11.
[37] WANG Y,HU J,TENG F,et al.KaTaGCN:knowledge-aug-mented and time-aware graph convolutional network for efficient traffic forecasting[J].Information Fusion,2024,111:102542.
[38] WANG S,ZHANG Y,HU Y,et al.Knowledge fusion enhanced graph neural network for traffic flow prediction[J].Physica A:Statistical Mechanics and its Applications,2023,623:128842.
[39] LI M,ZHU Z.Spatial-temporal fusion graph neural networksfor traffic flow forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2021:4189-4196.
[40] BERNDT D J,CLIFFORD J.Using dynamic time warping tofind patterns in time series[C]//Proceedings of the 3rd International Conference on Knowledge Discovery and Data Mining.1994:359-370.
[41] LIN Y,LIU Z,SUN M.Knowledge representation learning with entities,attributes and relations[J].Ethnicity,2016,1:41-52.
[42] BORDES A,USUNIER N,GARCIA-DURAN A,et al.Translating embeddings for modeling multi-relational data[C]//Proceedings of the 27th International Conference on Neural Information Processing Systems.2013:2787-2795.
[43] SUN Z,DENG Z H,NIE J Y,et al.Rotate:Knowledge graph embedding by relational rotation in complex space[J].arXiv:1902.10197,2019.
[44] FLOYD R W.Algorithm 97:shortest path[J].Communications of the ACM,1962,5(6):345-345.
[45] VASWANI A,SHAZEER N,PARMAR N,et al.Attention is all you need[C]//Proceedings of the 31st International Confe-rence on Neural Information Processing Systems.2017:6000-6010.
[46] LI Y,YU R,SHAHABI C,et al.Diffusion convolutional recurrent neural network:Data-driven traffic forecasting[J].arXiv:1707.01926,2017.
[47] BAI L,YAO L,LI C,et al.Adaptive graph convolutional recurrent network for traffic forecasting[J].Advances in Neural Information Processing Systems,2020,33:17804-17815.
[48] JIANG J,HAN C,JIANG W,et al.Libcity:A unified library towards efficient and comprehensive urban spatial-temporal prediction[J].arXiv:2304.14343,2023.
[1] PAN Yuquan, YUAN Deyu, WANG Anran, JIA Yuan. Enhanced GNNs Across Social Networks User Identity Linkage Algorithm Based on HiddenFeatures [J]. Computer Science, 2026, 53(8): 50-60.
[2] JIA Zishuo, ZHANG Jian’ge, HE Haofeng, FENG Shizhong, LIU Yilin. Survey on Mutually Augmenting Technologies and Applications of Large Models and KnowledgeGraphs [J]. Computer Science, 2026, 53(8): 219-228.
[3] ZHOU Haobin, LU Yunhao, QIN Jun, JIAO Xintao, ZENG Biqing. CCSFR:Collaborative-Content Semantic Fusion for Review-enhanced Recommendation [J]. Computer Science, 2026, 53(8): 276-284.
[4] ZHANG Haoran, HAO Wenning, JIN Dawei, CHENG Kai, LIU Junyang. Agentic Retrieval Augmented Generation Framework Based on Retrieval Task Planning and Reflection Mechanism [J]. Computer Science, 2026, 53(8): 285-297.
[5] LI Xiaochao, YUAN Zisu, LI Qianmu, LIU Fan, CHE Xun. Survey on Code Representation Learning for Vulnerability Detection [J]. Computer Science, 2026, 53(8): 388-402.
[6] WANG Xinlin, LI Yan, MA Chaofan, LI Shuo. Retrieval-Augmented Generation:Survey of Methods and Applications [J]. Computer Science, 2026, 53(7): 101-117.
[7] ZHAO Xingbo, LIAN Defu. Traffic Flow Forecasting Based on Dynamic Graph Convolution and Hypergraph Learning [J]. Computer Science, 2026, 53(7): 205-212.
[8] LI Kaiju, YIN Chenyang, CHENG Zhangtao, LIU Xueting, ZHOU Fan. Causal Subgraph Learning for Cascade Popularity Prediction [J]. Computer Science, 2026, 53(7): 213-221.
[9] WU Kai, SUN Zhe, ZHANG Xu, CAO Yadong, SUN Zhixin. Research on Modeling and Scheduling Methods for Intra-city Delivery Based on Heterogeneous Graph Neural Networks [J]. Computer Science, 2026, 53(7): 242-250.
[10] WANG Jiajun, JIAO Pengfei, ZHANG Xinxun, LI Tianpeng, GAO Mengzhou. Unsupervised Dynamic Graph Change Point Detection Method Based on Variational Graph Auto-encoder [J]. Computer Science, 2026, 53(7): 272-279.
[11] TAI Wenxin, LIU Xueting, WANG Xiaohan, ZHONG Ting, WANG Yong, ZHOU Fan. Trustworthy IP Geolocation Method via Graph Neural Networks and Conformalized Quantile Regression [J]. Computer Science, 2026, 53(7): 308-314.
[12] SHEN Jianwei, CHEN Jiawen, CHEN Hanlin, MA Xinjian, CHEN Xing. Construction and Application of Dataset Knowledge Graph Based on Metadata Semantic Enhancement [J]. Computer Science, 2026, 53(6A): 250500052-10.
[13] HAN Zhigeng, FU Chunshuo. Fraud Detection Model Based on Dual-space Heterogeneous Graph Neural Network [J]. Computer Science, 2026, 53(6A): 250600050-6.
[14] SHEN Yajie, WANG Jishu, JIN Kui, ZI Tong, TANG Mingjing. Second-order Multi-channel Directed Graph Convolution for Gene Regulatory Inference [J]. Computer Science, 2026, 53(6A): 250700116-10.
[15] ZHANG Xinliang, LIU Lilong, CHEN Shangheng, CHEN Ziyang, QIAN Shengsheng. Dual-stream Heterogeneous Social Graph for Micro-video Popularity Prediction [J]. Computer Science, 2026, 53(6A): 250800073-8.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!