STR-GODEs: Spatial-Temporal-Ridership Graph ODEs for Metro Ridership Prediction

Abstract

The metro ridership prediction has always received extensive attention from governments and researchers. Recent works focus on designing complicated graph convolutional recurrent network architectures to capture spatial and temporal patterns. These works extract the information of spatial dimension well, but the limitation of temporal dimension still exists. We extended Neural ODE algorithms to the graph network and proposed the STR-GODEs network, which can effectively learn spatial, temporal, and ridership correlations without the limitation of dividing data into equal-sized intervals on the timeline. While learning the spatial relations and the temporal correlations, we modify the GODE-RNN cell to obtain the ridership feature and hidden states. Ridership information and its hidden states are added to the GODESolve to reduce the error accumulation caused by long time series in prediction. Extensive experiments on two large-scale datasets demonstrate the efficacy and robustness of our model.

References in corpus (21)

• Adam: A Method for Stochastic Optimization
• Semi-Supervised Classification with Graph Convolutional Networks
• Inductive Representation Learning on Large Graphs
• Neural Message Passing for Quantum Chemistry
• Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translation
• T-GCN: A Temporal Graph ConvolutionalNetwork for Traffic Prediction
• Spectral Networks and Locally Connected Networks on Graphs
• Convolutional Neural Networks on Graphs with Fast Localized Spectral Filtering
• Hierarchical Graph Representation Learning with Differentiable Pooling
• Adaptive Graph Convolutional Recurrent Network for Traffic Forecasting
• Neural Ordinary Differential Equations
• Link Prediction Based on Graph Neural Networks
• Deep Multi-View Spatial-Temporal Network for Taxi Demand Prediction
• Latent ODEs for Irregularly-Sampled Time Series
• Augmented Neural ODEs
• On Robustness of Neural Ordinary Differential Equations
• Model-based Reinforcement Learning for Semi-Markov Decision Processes with Neural ODEs
• STEER: Simple Temporal Regularization For Neural ODEs
• Neural Ordinary Differential Equations for Semantic Segmentation of Individual Colon Glands
• Stable Neural Flows
• Simple Video Generation using Neural ODEs