Adaptive Spatio-Temporal Graph Learning for Bus Station Profiling

Abstract

Understanding and managing public transportation systems require capturing complex spatio-temporal correlations within datasets. Existing studies often use predefined graphs in graph learning frameworks, neglecting shifted spatial and long-term temporal correlations, which are crucial in practical applications. To address these problems, we propose a novel bus station profiling framework to automatically infer the spatio-temporal correlations and capture the shifted spatial and long-term temporal correlations in the public transportation dataset. The proposed framework adopts and advances the graph learning structure through the following innovative ideas: (1) Designing an adaptive graph learning mechanism to capture the interactions between spatio-temporal correlations rather than relying on pre-defined graphs; (2) Modeling shifted correlation in shifted spatial graphs to learn fine-grained spatio-temporal features; (3) Employing self-attention mechanism to learn the long-term temporal correlations preserved in public transportation data. We conduct extensive experiments on three real-world datasets and exploit the learned profiles of stations for the station passenger flow prediction task. Experimental results demonstrate that the proposed framework outperforms all baselines under different settings and can produce meaningful bus station profiles.