Online Metro Origin-Destination Prediction via Heterogeneous Information Aggregation
Not on the map. Why?
Abstract (arXiv)
Metro origin-destination prediction is a crucial yet challenging time-series analysis task in intelligent transportation systems, which aims to accurately forecast two specific types of cross-station ridership, i.e., Origin-Destination (OD) one and Destination-Origin (DO) one. However, complete OD matrices of previous time intervals can not be obtained immediately in online metro systems, and conventional methods only used limited information to forecast the future OD and DO ridership separately. In this work, we proposed a novel neural network module termed Heterogeneous Information Aggregation Machine (HIAM), which fully exploits heterogeneous information of historical data (e.g., incomplete OD matrices, unfinished order vectors, and DO matrices) to jointly learn the evolutionary patterns of OD and DO ridership. Specifically, an OD modeling branch estimates the potential destinations of unfinished orders explicitly to complement the information of incomplete OD matrices, while a DO modeling branch takes DO matrices as input to capture the spatial-temporal distribution of DO ridership. Moreover, a Dual Information Transformer is introduced to propagate the mutual information among OD features and DO features for modeling the OD-DO causality and correlation. Based on the proposed HIAM, we develop a unified Seq2Seq network to forecast the future OD and DO ridership simultaneously. Extensive experiments conducted on two large-scale benchmarks demonstrate the effectiveness of our method for online metro origin-destination prediction. Our code is resealed at https://github.com/HCPLab-SYSU/HIAM.
Key facts. Cited by 11 papers in the map (data through 7 Oct 2026). Builds on 42 of its 83 references that are in the map. First posted on arXiv 2 Jul 2021.
Builds on
42 of 83 references.
References · 42
- 2017Attention Is All You Needcited by 59,932
- 2015Deep Residual Learning for Image Recognitioncited by 51,809
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 29,915
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2016Semi-Supervised Classification with Graph Convolutional Networkscited by 11,282
- 2017Graph Attention Networkscited by 8,854
- 2014Neural Machine Translation by Jointly Learning to Align and Translatecited by 4,530
- 2014Learning Phrase Representations using RNN Encoder-Decoder for Statistical Machine Translationcited by 4,016
- 2015Delving Deep into Rectifiers: Surpassing Human-Level Performance on ImageNet Classificationcited by 3,590
- 2014Sequence to Sequence Learning with Neural Networkscited by 2,974
30 more
- 2020Informer: Beyond Efficient Transformer for Long Sequence Time-Series Forecastingcited by 2,141
- 2016Convolutional Neural Networks on Graphs with Fast Localized Spectral Filteringcited by 2,085
- 2015Convolutional LSTM Network: A Machine Learning Approach for Precipitation Nowcastingcited by 1,428
- 2013Spectral Networks and Locally Connected Networks on Graphscited by 1,097
- 2019Graph WaveNet for Deep Spatial-Temporal Graph Modelingcited by 820
- 2018Image Transformercited by 770
- 2015Convolutional Networks on Graphs for Learning Molecular Fingerprintscited by 751
- 2019GMAN: A Graph Multi-Attention Network for Traffic Predictioncited by 433
- 2016Deep Spatio-Temporal Residual Networks for Citywide Crowd Flows Predictioncited by 330
- 2021Spectral Temporal Graph Neural Network for Multivariate Time-series Forecastingcited by 298
- 2020Spatial-Temporal Transformer Networks for Traffic Flow Forecastingcited by 179
- 2021Discrete Graph Structure Learning for Forecasting Multiple Time Seriescited by 169
- 2018Deep Multi-View Spatial-Temporal Network for Taxi Demand Predictioncited by 162
- 2020Deep Transformer Models for Time Series Forecasting: The Influenza Prevalence Casecited by 129
- 2018Revisiting Spatial-Temporal Similarity: A Deep Learning Framework for Traffic Predictioncited by 119
- 2020Knowledge-Guided Multi-Label Few-Shot Learning for General Image Recognitioncited by 69
- 2019STG2Seq: Spatial-temporal Graph to Sequence Model for Multi-step Passenger Demand Forecastingcited by 59
- 2019Bayesian Temporal Factorization for Multidimensional Time Series Predictioncited by 47
- 2020Cross-Domain Facial Expression Recognition: A Unified Evaluation Benchmark and Adversarial Graph Learningcited by 46
- 2019Predicting Citywide Crowd Flows in Irregular Regions Using Multi-View Graph Convolutional Networkscited by 40
- 2019Contextualized Spatial-Temporal Network for Taxi Origin-Destination Demand Predictioncited by 34
- 2019Tensor Completion for Weakly-dependent Data on Graph for Metro Passenger Flow Predictioncited by 30
- 2018Deep Sequence Learning with Auxiliary Information for Traffic Predictioncited by 29
- 2019Predicting origin-destination ride-sourcing demand with a spatio-temporal encoder-decoder residual multi-graph convolutional networkcited by 23
- 2018Attentive Crowd Flow Machinescited by 22
- 2020Pay Attention to Evolution: Time Series Forecasting with Deep Graph-Evolution Learningcited by 22
- 2019Learning Backtrackless Aligned-Spatial Graph Convolutional Networks for Graph Classificationcited by 13
- 2020SG-Net: Syntax Guided Transformer for Language Representationcited by 13
- 2020Deep Learning on Traffic Prediction: Methods, Analysis and Future Directionscited by 8
- 2020Short-term origin-destination demand prediction in urban rail transit systems: A channel-wise attentive split-convolutional neural network methodcited by 1
Built on by
Citing papers per year: 2021: 2; 2022: 0; 2023: 2; 2024: 4; 2025: 0; 2026: 3.
11 papers.
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Surveycited by 140
- 2021Road Network Guided Fine-Grained Urban Traffic Flow Inferencecited by 4
- 2021Completion and Augmentation based Spatiotemporal Deep Learning Approach for Short-Term Metro Origin-Destination Matrix Prediction under Limited Observable Datacited by 2
- 2023Long-term Wind Power Forecasting with Hierarchical Spatial-Temporal Transformercited by 2
- 2024ODMixer: Fine-grained Spatial-temporal MLP for Metro Origin-Destination Predictioncited by 2
- 2024Multi-Graph Inductive Representation Learning for Large-Scale Urban Rail Demand Prediction under Disruptionscited by 1
- 2024UMOD: A Novel and Effective Urban Metro Origin-Destination Flow Prediction Methodcited by 0
- 2024CSP-AIT-Net: A contrastive learning-enhanced spatiotemporal graph attention framework for short-term metro OD flow prediction with asynchronous inflow trackingcited by 0
- 2026Hierarchical Forecast Reconciliation for Urban Rail Transit Demand Prediction under Operational Disruptionscited by 0
- 2026Structure-Aware Graph Multi-Task Learning for Dynamic Sparse OD Demand Predictioncited by 0
- 2026Inferring Urban Mobility Interactions from Aggregated Dynamicscited by 0
- 2026Inferring Urban Mobility Interactions from Aggregated Dynamics7 Sep 2026
- 2026Structure-Aware Graph Multi-Task Learning for Dynamic Sparse OD Demand Prediction19 Jun 2026
- 2026Hierarchical Forecast Reconciliation for Urban Rail Transit Demand Prediction under Operational Disruptions5 Jun 2026
- 2024CSP-AIT-Net: A contrastive learning-enhanced spatiotemporal graph attention framework for short-term metro OD flow prediction with asynchronous inflow tracking2 Dec 2024
- 2024UMOD: A Novel and Effective Urban Metro Origin-Destination Flow Prediction Method8 Sep 2024
- 2024Multi-Graph Inductive Representation Learning for Large-Scale Urban Rail Demand Prediction under Disruptions28 Aug 2024
- 2024ODMixer: Fine-grained Spatial-temporal MLP for Metro Origin-Destination Prediction24 Apr 2024
- 2023Long-term Wind Power Forecasting with Hierarchical Spatial-Temporal Transformer30 May 2023
- 2023Spatio-Temporal Graph Neural Networks for Predictive Learning in Urban Computing: A Survey25 Mar 2023
- 2021Road Network Guided Fine-Grained Urban Traffic Flow Inference29 Sep 2021
- 2021Completion and Augmentation based Spatiotemporal Deep Learning Approach for Short-Term Metro Origin-Destination Matrix Prediction under Limited Observable Data9 Aug 2021