POI-based Traffic Generation via Supervised Contrastive Learning on Reconstructed Graph
Keywords: Traffic Generation, Contrastive Learning, Graph Reconstruction
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 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 Traffic Generation model via Supervised Contrastive learning on \Reconstructed 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 structural 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.
Primary Area: learning on graphs and other geometries & topologies
Code Of Ethics: I acknowledge that I and all co-authors of this work have read and commit to adhering to the ICLR Code of Ethics.
Submission Guidelines: I certify that this submission complies with the submission instructions as described on https://iclr.cc/Conferences/2024/AuthorGuide.
Anonymous Url: I certify that there is no URL (e.g., github page) that could be used to find authors' identity.
No Acknowledgement Section: I certify that there is no acknowledgement section in this submission for double blind review.
Submission Number: 2735
Loading