nav emailalert searchbtn searchbox tablepage yinyongbenwen piczone journalimg journalInfo journalinfonormal searchdiv searchzone qikanlogo popupnotification paper paperNew
2026, 02, v.58 17-24
基于多信息注意力对抗图卷积的公交车客流预测
基金项目(Foundation):
邮箱(Email):
DOI: 10.13705/j.issn.1671-6841.2024120
摘要:

针对公交车客流预测中时空依赖关系难以有效利用的问题,提出一种基于多信息注意力机制的动态自适应对抗图卷积网络客流预测模型。首先,利用时间特征编码器捕获不同时段客流之间的相似性,引入公交车站点的兴趣点(point of interest, POI)信息以辅助模型捕捉更多的节点特征。其次,采用动态建模时空依赖关系的方法完成对非欧几里得关系的建模,利用SimAM注意力模块捕获不同站点客流数据之间的整体差异性。在真实公交车客流数据集上的实验结果表明,相比最优基线模型,所提模型在预测未来12个时间步时的平均MAE和RMSE分别降低了0.34和0.33,展现了其在公交车客流预测中的有效性和优越性。

Abstract:

Aiming at the difficulty of utilizing spatiotemporal dependence relationship in bus passenger flow prediction effectively, a prediction model of passenger flow based on multiple information attention and dynamic adaptive adversarial graph convolutional network was proposed. Firstly, the time feature encoder was used to capture the similarity between passenger flows at different time periods, and point of interest(POI) information of bus stations was incorporated to enhance node feature extraction. Secondly, the dynamic modeling of spatiotemporal dependence was adopted to complete the modeling of non-Euclidean relationships, and the SimAM attention module was utilized to capture the overall differences in passenger flow data at different stations. The experimental results on real bus passenger flow data showed that compared with the best baseline model, the proposed model reduced the average MAE and RMSE of the next 12 time steps by 0.34 and 0.33, respectively, demonstrating its effectiveness and superiority in predicting bus passenger flow.

参考文献

[1] TAO S M,ZHANG H Y,YANG F,et al.Multiple information spatial-temporal attention based graph convolution network for traffic prediction[J].Applied soft computing,2023,136:110052.

[2] JIANG J Y,WU B P,CHEN L,et al.Dynamic adaptive and adversarial graph convolutional network for traffic forecasting [EB/OL].(2022-08-05)[2024-03-12].https://arxiv.org/pdf/2208.03063v1.

[3] LUO X L,ZHU C J,ZHANG D T,et al.STG4Traffic:a survey and benchmark of spatial-temporal graph neural networks for traffic prediction [EB/OL].(2023-07-02)[2024-03-12].https://arxiv.org/pdf/2307.00495.

[4] YANG L,ZHANG R Y,LI L,et al.SimAM:a simple,parameter-free attention module for convolutional neural networks[C]//Proceedings of the International Conference on Machine Learning.New York:ACM Press,2021:11863-11874.

[5] YANG J,ZHU J W,LIU B,et al.Short-term passenger flow prediction for urban railway transit based on combined model[J].Journal of transportation systems engineering and information technology,2019,19(3):119-125.

[6] YANG H F,DILLON T S,CHANG E,et al.Optimized configuration of exponential smoothing and extreme learning machine for traffic flow forecasting[J].IEEE transactions on industrial informatics,2019,15(1):23-34.

[7] LIPPI M,BERTINI M,FRASCONI P.Short-term traffic flow forecasting:an experimental comparison of time-series analysis and supervised learning[J].IEEE transactions on intelligent transportation systems,2013,14(2):871-882.

[8] LI C,WANG X D,CHENG Z W,et al.Forecasting bus passenger flows by using a clustering-based support vector regression approach[J].IEEE access,2020,8:19717-19725.

[9] VLAHOGIANNI E I,KARLAFTIS M G,GOLIAS J C.Optimized and meta-optimized neural networks for short-term traffic flow prediction:a genetic approach[J].Transportation research part C:emerging technologies,2005,13(3):211-234.

[10] ZHANG D,KABUKA M R.Combining weather condition data to predict traffic flow:a GRU-based deep learning approach[J].IET intelligent transport systems,2018,12(7):578-585.

[11] JIANG W W,LUO J Y,HE M,et al.Graph neural network for traffic forecasting:the research progress[J].ISPRS international journal of geo-information,2023,12(3):100.

[12] YU B,YIN H T,ZHU Z X.Spatio-temporal graph convolutional networks:a deep learning framework for traffic forecasting[C]//Proceedings of the 27th International Joint Conference on Artificial Intelligence.Piscataway:IEEE Press,2018:3634-3640.

[13] GUO S,LIN Y,FENG N,et al.Attention based spatial-temporal graph convolutional networks for traffic flow forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.Palo Alto:AAAI Press,2019:922-929.

[14] LI P,WANG S,ZHAO H T,et al.IG-net:an interaction graph network model for metro passenger flow forecasting[J].IEEE transactions on intelligent transportation systems,2023,24(4):4147-4157.

[15] 田钊,程钰婕,张乾钟,等.基于ASTLSTM的地铁乘客流量短时预测[J].郑州大学学报(理学版),2024,56(5):55-61.TIAN Z,CHENG Y J,ZHANG Q Z,et al.Short-term forecast of subway passenger flow based on ASTLSTM[J].Journal of Zhengzhou university (natural science edition),2024,56(5):55-61.

[16] LIU M Z,ZHU T Y,YE J C,et al.Spatio-temporal AutoEncoder for traffic flow prediction[J].IEEE transactions on intelligent transportation systems,2023,24(5):5516-5526.

[17] 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.

[18] CHOI J,CHOI H,HWANG J,et al.Graph neural controlled differential equations for traffic forecasting[C]//Proceedings of the AAAI Conference on Artificial Intelligence.Palo Alto:AAAI Press,2022:6367-6374.

[19] SONG C,LIN Y F,GUO S N,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.Palo Alto:AAAI Press,2020:914-921.

[20] WENG W C,FAN J,WU H F,et al.A decomposition dynamic graph convolutional recurrent network for traffic forecasting[J].Pattern recognition,2023,142:109670.

基本信息:

DOI:10.13705/j.issn.1671-6841.2024120

中图分类号:U495

引用信息:

[1]颜建强,赵仁琪,高原,等.基于多信息注意力对抗图卷积的公交车客流预测[J].郑州大学学报(理学版),2026,58(02):17-24.DOI:10.13705/j.issn.1671-6841.2024120.

发布时间:

2025-01-13

出版时间:

2025-01-13

网络发布时间:

2025-01-13

检 索 高级检索