Computer Science ›› 2025, Vol. 52 ›› Issue (4): 110-118.doi: 10.11896/jsjkx.241000094

• Database & Big Data & Data Science • Previous Articles     Next Articles

SCFNet:Fusion Framework of External Spatial Features for Spatio-temporal Prediction

LIU Tengfei, CHEN Liyue, FANG Jiangyi, WANG Leye   

  1. Key Laboratory of High Confidence Software Technologies(Peking University),Ministry of Education,Beijing 100871,ChinaSchool of Computer Science,Peking University,Beijing 100871,China
  • Received:2024-10-20 Revised:2024-12-06 Online:2025-04-15 Published:2025-04-14
  • About author:LIU Tengfei,born in 2002,is a member of CCF(No.U9218G).His main research interests include urban computing and urban traffic.
    WANG Leye,born in 1987,associate professor,Ph.D supervisor,is a member of CCF(No.C9319S).His main research interests include ubiquitous computing,mobile crowdsensing and urban computing.

Abstract: Road information is closely related to the current traffic pattern of roads.Rich POI semantics can reveal the attributes of an area.Demographic data reveals the trend of population flow in an area.Considering the influence brought by the above external spatial features on the flow in spatio-temporal prediction can help the model accomplish more accurate prediction.Existing external spatial modeling methods usually focus on the input external spatial features,learn spatially relevant semantic representations through neural network mapping,and then fuse them with the final spatiotemporal flow representations.However,due to the heterogeneity between flow representations and spatial features,the existing external spatial feature modeling methods are often not highly scalable and can only target specific external spatial features or specific spatio-temporal models.To overcome the above problems,we propose a spatial context fusion network for traffic forecasting(SCFNet).Specifically,we introduce an attention mechanism based on information interaction to compute attention scores between spatio-temporal representations and external spatial features to achieve an efficient fusion of external spatial features and spatio-temporal representations,and we design a dynamic encoding method of time vectors to generate dynamic spatial feature semantics.SCFNet supports a mixture of different spatial static features such as regional demographic data,road information,and POI inputs.We conduct experiments on three real traffic datasets and demonstrated that SCFNet can significantly improve the prediction accuracy of various state-of-the-art spatiotemporal prediction methods such as(MTGNN,ASTGCN,and GraphWaveNet).

Key words: Spatio-temporal forecasting, External feature modeling, Point of interest, Road feature, Demographic, Attention me-chanism, Time-varying semantics

CLC Number: 

  • TP183
[1]CHEN Y,CHEN C,WU Q,et al.Spatial-temporal traffic congestion identification and correlation extraction using floating car data[J].Journal of Intelligent Transportation Systems,2021,25(3):263-280.
[2]DRAGOMIR G.A spatial-temporal data model for choosing optimal multimodal routes in urban areas[C]//2012 IEEE 8th International Conference on Intelligent Computer Communication and Processing.IEEE,2012.
[3]WU L,WANG M,WU D,et al.DynSTGAT:Dynamic spatial-temporal graph attention network for traffic signal control[C]//Proceedings of the 30th ACM International Conference on Information & Knowledge Management.2021.
[4]WU F,ZHU M,WANG Q,et al.Spatial-temporal visualization of city-wide crowd movement[J].Journal of Visualization,2017,20:183-194.
[5]FANG S,ZHANG Q,MENG G,et al.GSTNet:Global Spatial-Temporal Network for Traffic Flow Prediction[C]//IJCAI.2019.
[6]DENG D,SHAHABI C,DEMIRYUREK U,et al.Latent space model for road networks to predict time-varying traffic[C]//Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.2016:1525-1534.
[7]YUAN J,ZHENG Y,XIE X.Discovering regions of differentfunctions in a city using human mobility and POIs[C]//Proceedings of the 18th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining.2012:186-194.
[8]ZHENG C,FAN X,WEN C,et al.DeepSTD:Mining spatio-temporal disturbances of multiple context factors for citywide traffic flow prediction[J].IEEE Transactions on Intelligent Transportation Systems,2019,21(9):3744-3755.
[9]XIA T,LIN J,LI Y,et al.3dgcn:3-dimensional dynamic graph convolutional network for citywide crowd flow prediction[J].ACM Transactions on Knowledge Discovery from Data,2021,15(6):1-21.
[10]LIN Z Q,J.FENG J,LU Z Y,et al.Deepstn+:Context-aware spatial temporal neural network for crowd flow prediction in metropolis[C]//AAAI.2019:1020-1027.
[11]YUAN X M,HAN J C.WANG X,et al.A novel learning approach for citywide crowd flow prediction[C]//IEEE.2019:341-346.
[12]XING J,KONG X,XING W,et al.STGs:Construct spatial and temporal graphs for citywide crowd flow prediction[J].Applied Intelligence,2022,52(11):12272-12281.
[13]LI F,FENG J,YAN H,et al.Crowd flow prediction for irregular regions with semantic graph attention network[J].ACM Transactions on Intelligent Systems and Technology(TIST),2022,13(5):1-14.
[14]ZENG H,PENG Z,HUANG X H,et al.Deep spatio-temporal neural network based on interactive attention for traffic flow prediction[J].Applied Intelligence,2022,15(9):1-12.
[15]WANG K,LIU L B,LIU Y,et al.Urban regional functionguided traffic flow prediction[J].Information Sciences,2023,634:308-320.
[16]WU Z H,PAN S R,LONG G D,et al.Graph wavenet for deep spatial-temporal graph modeling[J].arXiv:1906.00121,2019.
[17]MA X,DAI Z,HE Z,et al.Learning traffic as images:A deep convolutional neural network for large-scale transportation network speed prediction[J].Sensors,2017,17(4):818.
[18]JIE L,YONG F.Prediction of time series data based on multi-time scale rnn [J].Computer Applications and Software,2018,35(7):33-37.
[19]LV Z,XU J,ZHENG K,et al.Lc-rnn:A deep learning model for traffic speed prediction[C]//IJCAI.2018.
[20]DO L N N,VU H L,VO B Q,et al.An effective spatial-temporal attention based neural network for traffic flow prediction[J].Transportation Research Part C:Emerging Technologies,2019,108:12-28.
[21]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.
[22]ZHENG C,FAN X,WANG C,et al.Gman:A graph multi-attention network for traffic prediction[C]//Proceedings of the AAAI Conference on Artificial Intelligence.2020:1234-1241.
[23]VASWANI A,SHAZEER N,PARMAR N,et al.Attention isall you need[C]//Proceedings of the 31st International Conference on Neural Information Processing SystemsDecember.2017:6000-6010.
[24]LIU X,YU H F,DHILON I,et al.Learning to encode position for transformer with continuous dynamical model[C]//International Conference on Machine Learning.PMLR,2020:6327-6335.
[25]WANG K,WANG P,HUANG Z,et al.A two-step model for predicting travel demand in expanding subways[J].IEEE Transactions on Intelligent Transportation Systems,2022,23(10):19534-19543.
[26]LIANG Y,OUYANG K,SUN J,et al.Fine-grained urban flow prediction[C]//Proceedings of the Web Conference 2021.2021:1833-1845.
[27]PAN Z,LIANG Y,WANG W,et al.Urban traffic prediction from spatio-temporal data using deep meta learning[C]//Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining.2019:1720-1730.
[28]WANG L,CHAI D,LIU X,et al.Exploring the generalizability of spatio-temporal traffic prediction:Meta-modeling and an analytic framework[J].IEEE Transactions on Knowledge and Data Engineering,2021,35(4):3870-3884.
[29]WU Z,PAN S,LONG G,et al.Connecting the dots:Multivariate time series forecasting with graph neural networks[C]//Proceedings of the 26th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining.2020:753-763.
[1] XIONG Keqin, RUAN Sijie, YANG Qianyu, XU Changwei , YUAN Hanning. Mobility Data-driven Location Type Inference Based on Crowd Voting [J]. Computer Science, 2025, 52(3): 169-179.
[2] SU Chang, PENG Shao-wen, XIE Xian-zhong, LIU Ning-ning. Study on Check-in Prediction Based on Deep Learning and Factorization Machine [J]. Computer Science, 2019, 46(5): 185-190.
[3] GUO Xiao-bo,ZHAO Shu-liang,LIU Jun-dan,ZHAO Jiao-jiao and WANG Chang-bin. Knowledge Presentation of Association Rules Based on Conceptual Graphs [J]. Computer Science, 2013, 40(8): 261-265.
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!