Enhancing Urban Flow Prediction Via Mutual Reinforcement with Multi-Scale Regional Information

Xu Zhang,Mengxin Cao,Yongshun Gong,Xiaoming Wu,Xiangjun Dong,Ying Guo,Long Zhao,Chengqi Zhang
DOI: https://doi.org/10.1016/j.neunet.2024.106900
2024-01-01
Abstract:Intelligent Transportation Systems (ITS) are essential for modern urban development, with urban flow prediction being a key component. Accurate flow prediction optimizes routes and resource allocation, benefiting residents, businesses, and the environment. However, few methods address the spatial–temporal heterogeneity of urban flows. Existing methods typically capture spatial features solely from urban flows, but spatial feature sensitivity becomes a bottleneck when dealing with small or noisy datasets. To address this issue, we propose a method for urban flow prediction via mutual reinforcement with multi-scale regional information (MR-UFP). Firstly, we employ spatial–temporal random masking and spatial–temporal contrastive learning pre-training to directly mine spatial–temporal heterogeneity from historical flow data. Secondly, we transform the task of spatial feature extraction and embedding for urban flow prediction into a mutual reinforcement task by multi-scale region classification auxiliary task. The adaptive environment fusion module and real-time graph processing dynamically correlate regional and environmental features during flow prediction. To balance the mutual reinforcement of the two tasks, we design a joint loss function to optimize feature embedding and feedback correction, ensuring robust and accurate urban flow prediction. Extensive experiments on two real-world datasets demonstrate that MR-UFP outperforms baseline models, showcasing its robustness and effectiveness even with minimal data.
What problem does this paper attempt to address?