Volume 26 Issue 6
Jun.  2026
Turn off MathJax
Article Contents
ZHAO Xia, SHI Zhuo-ya, LI Zhi-hong, LIU Jian-feng, WU Meng-lin, LI Chen-ji. Forecasting ride-hailing demand via contextual spatiotemporal cross-attention mechanism[J]. Journal of Traffic and Transportation Engineering, 2026, 26(6): 186-197. doi: 10.19818/j.cnki.1671-1637.2026.075
Citation: ZHAO Xia, SHI Zhuo-ya, LI Zhi-hong, LIU Jian-feng, WU Meng-lin, LI Chen-ji. Forecasting ride-hailing demand via contextual spatiotemporal cross-attention mechanism[J]. Journal of Traffic and Transportation Engineering, 2026, 26(6): 186-197. doi: 10.19818/j.cnki.1671-1637.2026.075

Forecasting ride-hailing demand via contextual spatiotemporal cross-attention mechanism

doi: 10.19818/j.cnki.1671-1637.2026.075
Funds:

National Natural Science Foundation of China 52402377

Natural Science Foundation of Beijing 8252005

More Information
  • Corresponding author: LI Zhi-hong, professor, PhD, E-mail: lizhihong@bucea.edu.cn
  • Received Date: 2025-04-01
  • Accepted Date: 2025-09-26
  • Rev Recd Date: 2025-08-24
  • Publish Date: 2026-06-28
  • To accurately characterize passengers' temporal preferences for frequent travel within specific geographic areas, or their spatial preferences for frequently visited locations within given time windows, this paper proposed a ride-hailing demand prediction model named ST-BiAformer (Spatiotemporal Bidirectional Association Transformer), which integrates a contextual spatiotemporal cross-attention mechanism. The model captured sequential dependencies in travel demand by constructing a contextual temporal correlation module; designed a spatiotemporal cross-attention mechanism to cross-extract spatial (or temporal) dependencies within specific temporal (or spatial) contexts, thereby revealing recurrent travel demand patterns along targeted spatiotemporal dimensions; and further developed a spatiotemporal fusion module to enhance the model's performance in both single-step and multi-step demand forecasting. The proposed model was evaluated on multiple datasets through a series of comparative, ablation, and robustness experiments, assessing its predictive performance under various spatiotemporal scenarios. Experimental results demonstrate that, compared with the optimal baseline model, the mean absolute error and root mean square error are reduced by 7.21% and 5.54%, respectively, demonstrating improved overall prediction accuracy. In the 5 min, 10 min, and 15 min forecasting tasks, both error metrics decrease by 1% - 7%, indicating sound single-step and multi-step predictive capability. The model achieves the best performance only when all three constituent submodules work jointly, enabling comprehensive modeling of travel demand dependencies along both contextual temporal and spatiotemporal correlation dimensions. The proposed model is expected to precisely match users' ride-hailing demands across heterogeneous spatiotemporal scenarios at different time periods or locations, thereby providing technical support for enhancing supply-demand scheduling in urban ride-hailing services.

     

  • loading
  • [1]
    LI Zhi-hong, SHEN Tian-yu, WEN Yan-jie, et al. Order demand prediction and anomaly-point identification for online car-hailing orders based on hybrid machine learning frame-work[J]. Journal of Transport Information and Safety, 2023, 41(3): 157-165, 174.
    [2]
    ZHAO X, CUI M Y, LEVINSON D. Exploring temporal variability in travel patterns on public transit using big smart card data[J]. Environment and Planning B: Urban Analytics and City Science, 2023, 50(1): 198-217. doi: 10.1177/23998083221089662
    [3]
    WEN Y J, LI Z H, WANG X Y, et al. Traffic demand prediction based on spatial-temporal guided multi graph Sand-wich-Transformer[J]. Information Sciences, 2023, 643: 119269. doi: 10.1016/j.ins.2023.119269
    [4]
    CUI Jian-xun, YAO Jia, ZHAO Bo-yuan. Review on short-term traffic flow prediction methods based on deep learning[J]. Journal of Traffic and Transportation Engineering, 2024, 24(2): 50-64. doi: 10.19818/j.cnki.1671-1637.2024.02.003
    [5]
    LIANG G J, KINTAK U, NING X, et al. Semantics-aware dynamic graph convolutional network for traffic flow fore-casting[J]. IEEE Transactions on Vehicular Technology, 2023, 72(6): 7796-7809. doi: 10.1109/TVT.2023.3239054
    [6]
    XIA M R, JIN D W, CHEN J Y. Short-term traffic flow prediction based on graph convolutional networks and fede-rated learning[J]. IEEE Transactions on Intelligent Trans-portation Systems, 2023, 24(1): 1191-1203. doi: 10.1109/TITS.2022.3179391
    [7]
    HOCHREITER S, SCHMIDHUBER J. Long short-term memory[J]. Neural Computation, 1997, 9(8): 1735-1780. doi: 10.1162/neco.1997.9.8.1735
    [8]
    CHO K, VAN MERRIENBOER B, GULCEHRE C, et al. Learning phrase representations using RNN encoder-decoder for statistical machine translation[C]//ACL. Proceedings of the 2014 conference on empirical methods in natural language processing (EMNLP). Baltimore: ACL, 2014: 1724-1734.
    [9]
    GRAVES A, SCHMIDHUBER J. Framewise phoneme classi-fication with bidirectional LSTM and other neural network architectures[J]. Neural Networks, 2005, 18(5): 602-610.
    [10]
    LI Z H, XU H, GAO X L, et al. Fusion attention mecha-nism bidirectional LSTM for short-term traffic flow prediction[J]. Journal of Intelligent Transportation Systems, 2024, 28(4): 511-524. doi: 10.1080/15472450.2022.2142049
    [11]
    JIANG J W, HAN C K, ZHAO W X, et al. PDFormer: Propagation delay-aware dynamic long-range transformer for traffic flow prediction[J]. Proceedings of the AAAI Confe-rence on Artificial Intelligence, 2023, 37(4): 4365-4373. doi: 10.1609/aaai.v37i4.25556
    [12]
    VASWANI A, SHAZEER N, PARMAR N, et al. Atten-tion is all you need[J]. Advances in Neural Information Processing Systems, 2017, 30.
    [13]
    ZHOU H Y, ZHANG S H, PENG J Q, et al. Informer: Beyond efficient transformer for long sequence time-series forecasting[J]. Proceedings of the AAAI Conference on Arti-ficial Intelligence, 2021, 35(12): 11106-11115. doi: 10.1609/aaai.v35i12.17325
    [14]
    WU H, XU J, WANG J, et al. Autoformer: Decomposition transformers with auto-correlation for long-term series forecasting[J]. Advances in Neural Information Processing Systems, 2021, 34: 22419-22430.
    [15]
    YAO Jun-feng, HE Rui, SHI Tong-tong, et al. Review on machine learning-based traffic flow prediction methods[J]. Journal of Traffic and Transportation Engineering, 2023, 23(3): 44-67. doi: 10.19818/j.cnki.1671-1637.2023.03.003
    [16]
    LIU K, CHEN Z J, YAMAMOTO T, et al. Exploring the impact of spatiotemporal granularity on the demand prediction of dynamic ride-hailing[J]. IEEE Transactions on Intelligent Transportation Systems, 2023, 24(1): 104-114. doi: 10.1109/TITS.2022.3216016
    [17]
    YAN H Y, MA X L, PU Z Y. Learning dynamic and hierar-chical traffic spatiotemporal features with transformer[J]. IEEE Transactions on Intelligent Transportation Systems, 2022, 23(11): 22386-22399. doi: 10.1109/TITS.2021.3102983
    [18]
    HUO G Y, ZHANG Y, WANG B Y, et al. Hierarchical spatio-temporal graph convolutional networks and trans-former network for traffic flow forecasting[J]. IEEE Tran-sactions on Intelligent Transportation Systems, 2023, 24(4): 3855-3867. doi: 10.1109/TITS.2023.3234512
    [19]
    LI M Z, ZHU Z X. Spatial-temporal fusion graph neural networks for traffic flow forecasting[J]. Proceedings of the AAAI Conference on Artificial Intelligence, 2021, 35(5): 4189-4196. doi: 10.1609/aaai.v35i5.16542
    [20]
    WU Z H, PAN S R, LONG G D, et al. Graph WaveNet for deep spatial-temporal graph modeling[C]//IJCAI. Pro-ceedings of the Twenty-Eighth International Joint Conference on Artificial Intelligence. Macao: International Joint Confe-rences on Artificial Intelligence Organization, 2019: 1907-1913.
    [21]
    WANG Q Y, WANG S H, ZHUANG D Y, et al. Uncer-tainty quantification of spatiotemporal travel demand with probabilistic graph neural networks[J]. IEEE Transactions on Intelligent Transportation Systems, 2024, 25(8): 8770-8781. doi: 10.1109/TITS.2024.3367779
    [22]
    ZHANG H P, DONG H H, YANG Z Q. TSGDiff: Traffic state generative diffusion model using multi-source infor-mation fusion[J]. Transportation Research Part C: Emerging Technologies, 2025, 174: 105081. doi: 10.1016/j.trc.2025.105081
    [23]
    ZHOU Q S, ZHANG Y F, MAKRIDIS M A, et al. Mo-GERNN: An inductive traffic predictor for unobserved loca-tions[J]. Transportation Research Part C: Emerging Tech-nologies, 2025, 174: 105080. doi: 10.1016/j.trc.2025.105080
    [24]
    LIU XUN, LEI F Y, XIA G Q, et al. AdjMix: Simplifying and attending graph convolutional networks[J]. Complex & Intelligent Systems, 2022, 8(2): 1005-1014.
    [25]
    YU B, YIN H T, ZHU Z X. Spatio-temporal graph convo-lutional networks: A deep learning framework for traffic forecasting[C]//IJCAI. Proceedings of the Twenty-Seventh International Joint Conference on Artificial Intelligence. Stockholm: International Joint Conferences on Artificial Intelligence Organization, 2018: 3634-3640.
    [26]
    LV M Q, HONG Z X, CHEN L, et al. Temporal multi-graph convolutional network for traffic flow prediction[J]. IEEE Transactions on Intelligent Transportation Systems, 2021, 22(6): 3337-3348. doi: 10.1109/TITS.2020.2983763
    [27]
    WEI S Q, FENG S Y, YANG H. Multi-view spatial-tem-poral graph convolutional network for traffic prediction[J]. IEEE Transactions on Intelligent Transportation Systems, 2024, 25(8): 9572-9586. doi: 10.1109/TITS.2024.3364759
    [28]
    LI G Y, ZHONG S H, DENG X D, et al. A lightweight and accurate spatial-temporal transformer for traffic forecasting[J]. IEEE Transactions on Knowledge and Data Engineer-ing, 2023, 35(11): 10967-10980. doi: 10.1109/TKDE.2022.3233086
    [29]
    WANG M, ZHANG Y, ZHAO X, et al. Traffic origin-desti-nation demand prediction via multichannel hypergraph convolutional networks[J]. IEEE Transactions on Computa-tional Social Systems, 2024, 11(4): 5496-5509. doi: 10.1109/TCSS.2024.3372856
    [30]
    WEN Y J, XU P, LI Z H, et al. RPConvformer: A novel Transformer-based deep neural networks for traffic flow prediction[J]. Expert Systems with Applications, 2023, 218: 119587. doi: 10.1016/j.eswa.2023.119587

Catalog

    Article Metrics

    Article views (50) PDF downloads(8) Cited by()
    Related

    /

    DownLoad:  Full-Size Img  PowerPoint
    Return
    Return