Skip to main content

POI-Based Traffic Generation via Supervised Contrastive Learning on Reconstructed Graph

  • Conference paper
  • First Online:
Database Systems for Advanced Applications (DASFAA 2024)

Abstract

Traffic flow generation problem under realistic scenarios has raised more and more attention in recent years. This problem aims at generating traffic flow without using historical traffic data. Since road network and Point of Interest (POI) data can provide a more comprehensive picture of traffic patterns, most previous methods use both or either of them to generate traffic flow. However, roadnet graph in real-world has bias and abnormal structure, which will influence the performance of traffic generation. Previous traffic generation models directly receive real-world roadnet graph with map-match POI data as input and then use an end-to-end loss for training, which could not model the complex relationship between POI and traffic in a proper way. Different from prior methods, we propose a novel POI-based \(\underline{{\textbf {T}}}\)raffic \(\underline{{\textbf {G}}}\)eneration model via \(\underline{{\textbf {S}}}\)upervised \(\underline{{\textbf {C}}}\)ontrastive learning on \(\underline{{\textbf {R}}}\)econstructed graph, termed as TG-SCR, which combines POI data and road network data to generate the distribution of traffic flows. Our model has two novel modules: a graph reconstruction module and a POI supervised contrastive module. The graph reconstruction module includes a k-NN graph builder and a k-NN graph aggregator, which is used to reconstruct the original roadnet graph into a k-NN graph and reform POI feature. The contrastive module is used to model the relationship between POI feature and traffic flow. Extensive experiments show that our model outperforms other baseline methods on four real-world datasets.

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

Access this chapter

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

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. Bai, L., Yao, L., Li, C., Wang, X., Wang, C.: Adaptive graph convolutional recurrent network for traffic forecasting. Adv. Neural. Inf. Process. Syst. 33, 17804–17815 (2020)

    Google Scholar 

  2. Breiman, L.: Random forests. Mach. Learn. 45, 5–32 (2001)

    Article  Google Scholar 

  3. Chen, T., Guestrin, C.: XGBoost: a scalable tree boosting system. In: Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 785–794 (2016)

    Google Scholar 

  4. Cortes, C., Vapnik, V.: Mach. Learn. Support-vector networks 20, 273–297 (1995)

    Google Scholar 

  5. Hamilton, W., Ying, Z., Leskovec, J.: Inductive representation learning on large graphs. Adv. Neural Inf. Process. Syst. 30 (2017)

    Google Scholar 

  6. Jiang, R., et al.: DeepCrowd: a deep model for large-scale citywide crowd density and flow prediction. IEEE Trans. Knowl. Data Eng. 35(1), 276–290 (2021)

    Google Scholar 

  7. Kipf, T.N., Welling, M.: Semi-supervised classification with graph convolutional networks. arXiv preprint arXiv:1609.02907 (2016)

  8. Ji, F., et al.: Dynamic graph convolutional recurrent network for traffic prediction: benchmark and solution. ACM Trans. Knowl. Discov. Data 17(1), 1–21 (2023)

    Article  Google Scholar 

  9. Liu, K.: Demands for bus ridesharing in Shanghai, China. IEEE Access 7, 74248–74262 (2019). https://doi.org/10.21227/2877-mk46

  10. Shao, E., et al.: DeepFlowGen: intention-aware fine grained crowd flow generation via deep neural networks. IEEE Trans. Knowl. Data Eng. 34(12), 5693–5707 (2021)

    Article  Google Scholar 

  11. Velickovic, P., et al.: Graph attention networks. STAT 1050(20), 10–48550 (2017)

    Google Scholar 

  12. Wang, G., Chen, X., Zhang, F., Wang, Y., Zhang, D.: Experience: understanding long-term evolving patterns of shared electric vehicle networks. In: The 25th Annual International Conference on Mobile Computing and Networking, pp. 1–12 (2019)

    Google Scholar 

  13. Xu, K., Hu, W., Leskovec, J., Jegelka, S.: How powerful are graph neural networks? arXiv preprint arXiv:1810.00826 (2018)

  14. Yu, H., Wu, Z., Wang, S., Wang, Y., Ma, X.: Spatiotemporal recurrent convolutional networks for traffic prediction in transportation networks. Sensors 17(7), 1501 (2017)

    Article  Google Scholar 

  15. Yuan, J., Zheng, Y., Xie, X., Sun, G.: Driving with knowledge from the physical world. In: Proceedings of the 17th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, pp. 316–324 (2011)

    Google Scholar 

  16. Yuan, J., et al.: T-drive: driving directions based on taxi trajectories. In: Proceedings of the 18th SIGSPATIAL International Conference on Advances in Geographic Information Systems, pp. 99–108 (2010)

    Google Scholar 

  17. Zhang, J., Zheng, Y., Qi, D., Li, R., Yi, X.: DNN-based prediction model for spatio-temporal data. In: Proceedings of the 24th ACM SIGSPATIAL International Conference on Advances in Geographic Information Systems, pp. 1–4 (2016)

    Google Scholar 

  18. Zhao, Z., Chen, W., Wu, X., Chen, P.C., Liu, J.: LSTM network: a deep learning approach for short-term traffic forecast. IET Intel. Transp. Syst. 11(2), 68–75 (2017)

    Article  Google Scholar 

Download references

Acknowledgements

This work was sponsored by National Key Research and Development Program of China under Grant No.2022YFB3904204, National Natural Science Foundation of China under Grant No. 62102246, 62272301, and Provincial Key Research and Development Program of Zhejiang under Grant No. 2021C01034. Part of the work was done when the students were doing internships at Yunqi Academy of Engineering.

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Guanjie Zheng .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2024 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Su, Z., Liu, Z., Ding, J., Zheng, G. (2024). POI-Based Traffic Generation via Supervised Contrastive Learning on Reconstructed Graph. In: Onizuka, M., et al. Database Systems for Advanced Applications. DASFAA 2024. Lecture Notes in Computer Science, vol 14850. Springer, Singapore. https://doi.org/10.1007/978-981-97-5552-3_15

Download citation

  • DOI: https://doi.org/10.1007/978-981-97-5552-3_15

  • Published:

  • Publisher Name: Springer, Singapore

  • Print ISBN: 978-981-97-5551-6

  • Online ISBN: 978-981-97-5552-3

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics