Abstract: Recently, deep learning based city flow prediction has been extensively used in the establishment of smart cities. These methods are data-hungry, making them unscalable to areas lacking data. Although transfer learning can use data-rich source domains to assist target domain cities in city flow prediction, the performance of existing methods cannot meet the needs of actual use, because the long-distance road network connectivity is ignored. To solve this problem, we propose a transfer learning method based on spatiotemporal graph convolution, in which we construct a co-occurrence space between the source and target domains, and then align the mapping of the source and target domains’ data in this space, to achieve the transfer learning of the source city flow prediction model on the target domain. Specifically, a dynamic spatiotemporal graph convolution module along with a temporal encoder is devised to simultaneously capture the concurrent spatiotemporal features, which implies the inherent relationship among the road network structures, human travel habits, and city bike flow. Then, these concurrent features are leveraged as cross-city invariant representations and nonlinearly spanned to a co-occurrence space. The target domain features are thereby aligned with the source domain features in the co-occurrence space by using a Mahalanobis distance loss, to achieve cross-city bike flow prediction. The proposed method is evaluated on the public bike flow datasets in Chicago, New York, and Washington in 2015, and significantly outperforms state-of-the-art techniques.

基于时空图卷积的城市流迁移预测

[1]Ahmed MS, Cook AR, 1979. Analysis of freeway traffic time-series data by using Box-Jenkins techniques. Transp Res Rec, 773(722):1-9.

[2]De Maesschalck R, Jouan-Rimbaud D, Massart DL, 2000. The Mahalanobis distance. Chemom Intell Lab Syst, 50(1):1-18.

[3]Guo SN, Lin YF, Feng N, et al., 2019. Attention based spatial-temporal graph convolutional networks for traffic flow forecasting. 33^rd AAAI Conf on Artificial Intelligence, p.922-929.

[4]He ZW, Li Y, Zhang Y, et al., 2023. Ensemble-transfer-learning-based channel parameter prediction in asymmetric massive MIMO systems. Front Inform Technol Electron Eng, 24(2):275-288.

[5]Hu J, Lin XH, Wang C, 2022. DSTGCN: dynamic spatial-temporal graph convolutional network for traffic prediction. IEEE Sens J, 22(13):13116-13124.

[6]Huang DR, Deng ZP, Zhao L, et al., 2017. A short-term traffic flow forecasting method based on Markov chain and grey Verhulst model. 6^th Data Driven Control and Learning Systems, p.606-610.

[7]Huang YJ, Song XZ, Zhang SY, et al., 2021. Transfer learning in traffic prediction with graph neural networks. IEEE Int Intelligent Transportation Systems Conf, p.3732-3737.

[8]Kang GK, Gao JZ, Chiao S, et al., 2018. Air quality prediction: big data and machine learning approaches. Int J Environ Sci Dev, 9(1):8-16.

[9]Li JY, Guo FC, Wang YB, et al., 2020. Short-term traffic prediction with deep neural networks and adaptive transfer learning. IEEE 23^rd Int Conf on Intelligent Transportation Systems, p.1-6.

[10]Lippi M, Bertini M, Frasconi P, 2010. Collective traffic forecasting. European Conf on Machine Learning and Knowledge Discovery in Databases, p.259-273.

[11]Liu J, Guan W, 2004. A summary of traffic flow forecasting methods. J Highway Transp Res Devel, 21(3):82-85.

[12]Lv YS, Duan YJ, Kang WW, et al., 2015. Traffic flow prediction with big data: a deep learning approach. IEEE Trans Intell Transp Syst, 16(2):865-873.

[13]Ma XL, Dai Z, He ZB, et al., 2017. Learning traffic as images: a deep convolutional neural network for large-scale transportation network speed prediction. Sensors, 17(4):818.

[14]Mallick T, Balaprakash P, Rask E, et al., 2019. Graph-partitioning-based diffusion convolutional recurrent neural network for large-scale traffic forecasting. Transp Res Rec, 2674(9):473-488.

[15]Mallick T, Balaprakash P, Rask E, et al., 2021. Transfer learning with graph neural networks for short-term highway traffic forecasting. 25^th Int Conf on Pattern Recognition, p.10367-10374.

[16]Miao H, Shen JX, Cao JN, et al., 2023. MBA-STNet: Bayes-enhanced discriminative multi-task learning for flow prediction. IEEE Trans Knowl Data Eng, 35(7):7164-7177.

[17]Peng H, Wang HF, Du BW, et al., 2020. Spatial temporal incidence dynamic graph neural networks for traffic flow forecasting. Inform Sci, 521:277-290.

[18]Seng DW, Lv FS, Liang ZY, et al., 2021. Forecasting traffic flows in irregular regions with multi-graph convolutional network and gated recurrent unit. Front Inform Technol Electron Eng, 22(9):1179-1193.

[19]Shao W, Jin ZL, Wang S, et al., 2022. Long-term spatio-temporal forecasting via dynamic multiple-graph attention. 31^st Int Joint Conf on Artificial Intelligence, p.2225-2232.

[20]Shao ZZ, Zhang Z, Wang F, et al., 2022. Spatial-temporal identity: a simple yet effective baseline for multivariate time series forecasting. Proc 31^st ACM Int Conf on Information & Knowledge Management, p.4454-4458.

[21]Smola AJ, Schölkopf B, 2004. A tutorial on support vector regression. Stat Comput, 14(3):199-222.

[22]Tascikaraoglu A, 2018. Evaluation of spatio-temporal forecasting methods in various smart city applications. Renew Sustain Energy Rev, 82:424-435.

[23]Van Der Voort M, Dougherty M, Watson S, 1996. Combining Kohonen maps with ARIMA time series models to forecast traffic flow. Transp Res Part C Emerg Technol, 4(5):307-318.

[24]Wang B, Yan Z, Lu J, et al., 2018. Road traffic flow prediction using deep transfer learning. Proc 13^th Int FLINS Conf, p.331-338.

[25]Wang BW, Wang JS, 2022. ST-MGAT: spatio-temporal multi-head graph attention network for traffic prediction. Phys A, 603:127762.

[26]Wang LY, Geng X, Ma XJ, et al., 2019. Cross-city transfer learning for deep spatio-temporal prediction. Proc 28^th Int Joint Conf on Artificial Intelligence, p.1893-1899.

[27]Wang SZ, Miao H, Li JY, et al., 2022. Spatio-temporal knowledge transfer for urban crowd flow prediction via deep attentive adaptation networks. IEEE Trans Intell Transp Syst, 23(5):4695-4705.

[28]Williams BM, Hoel LA, 2003. Modeling and forecasting vehicular traffic flow as a seasonal ARIMA process: theoretical basis and empirical results. J Transp Eng, 129(6):664-672.

[29]Xiang SM, Nie FP, Zhang CS, 2008. Learning a Mahalanobis distance metric for data clustering and classification. Patt Recog, 41(12):3600-3612.

[30]Xu ZY, Kang Y, Cao Y, 2023. High-resolution urban flows forecasting with coarse-grained spatiotemporal data. IEEE Trans Artif Intell, 4(2):315-327.

[31]Yao ZX, Xia SC, Li Y, et al., 2023. Transfer learning with spatial-temporal graph convolutional network for traffic prediction. IEEE Trans Intell Transp Syst, 24(8):8592-8605.

[32]Yu B, Yin HT, Zhu ZX, 2018. Spatio-temporal graph convolutional networks: a deep learning framework for traffic forecasting. Proc 27^th Int Joint Conf on Artificial Intelligence, p.3634-3640.

[33]Zhao L, Song YJ, Zhang C, et al., 2020. T-GCN: a temporal graph convolutional network for traffic prediction. IEEE Trans Intell Transp Syst, 21(9):3848-3858.