Skip to main content

Advertisement

Log in

Transfer learning with a spatiotemporal graph convolution network for city flow prediction

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

  • Research Article
  • Published:
Frontiers of Information Technology & Electronic Engineering Aims and scope Submit manuscript

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.

摘要

最近,基于深度学习的城市流量预测被广泛应用于智慧城市的建设。由于这些方法通常对数据量要求较高,因此难以扩展到数据匮乏的城市。虽然迁移学习可以利用数据丰富的源城市协助目标城市进行城市流量预测,但由于忽略了长距离路网的连通性,因此现有方法的性能无法满足实际使用的需要。为解决这个问题,提出一种基于时空图卷积的迁移预测方法,即在源城市和目标城市之间构建一个共现空间,然后在共现空间中对源城市和目标城市流量数据进行映射对齐,从而实现源城市流量预测模型在目标城市上的迁移预测。具体来说,我们设计了一个动态时空图卷积模块和一个时间编码器,以同时捕捉流量的时间特征和空间特征,这些特征揭示了道路网络结构、人类出行习惯和城市流量之间的内在关联。然后,将这些特征作为跨城市不变表示被非线性映射到共现空间。通过优化马氏距离损失,目标城市特征与源城市特征在共现空间中对齐,从而实现跨城市自行车流量预测。在2015年芝加哥、纽约和华盛顿的公共自行车流量数据集上对所提出的方法进行评估,结果表明该方法明显优于目前最先进的技术。

This is a preview of subscription content, log in via an institution to check access.

Access this article

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Price excludes VAT (USA)
Tax calculation will be finalised during checkout.

Instant access to the full article PDF.

Similar content being viewed by others

Data availability

The data that support the findings of this study are available from the corresponding authors upon reasonable request.

References

Download references

Author information

Authors and Affiliations

Authors

Contributions

Binkun LIU and Zhenyi XU designed the research. Binkun LIU processed the data and drafted the paper. Yang CAO helped organize the paper. Yu KANG and Yunbo ZHAO helped in data control and project management. Yang CAO and Zhenyi XU revised and finalized the paper. Yu KANG and Zhenyi XU provided the funding acquisition.

Corresponding authors

Correspondence to Yang Cao  (曹洋) or Zhenyi Xu  (许镇义).

Ethics declarations

All the authors declare that they have no conflict of interest.

Additional information

Project supported by the National Natural Science Foundation of China (Nos. 62103124 and 62033012), the Major Special Science and Technology Project of Anhui Province, China (No. 202003a07020009), and the Open Project Program of Key Laboratory of Ministry of Education of System Control and Information Processing, China (No. SCIP20230109)

Rights and permissions

Reprints and permissions

About this article

Check for updates. Verify currency and authenticity via CrossMark

Cite this article

Liu, B., Kang, Y., Cao, Y. et al. Transfer learning with a spatiotemporal graph convolution network for city flow prediction. Front Inform Technol Electron Eng 26, 79–92 (2025). https://doi.org/10.1631/FITEE.2300571

Download citation

  • Received:

  • Accepted:

  • Published:

  • Issue Date:

  • DOI: https://doi.org/10.1631/FITEE.2300571

Key words

关键词

CLC number