A Continuous Encoder-Decoder Method for Spatial-Temporal Forecasting

Lei Liu,Yanming Shen,Heng Qi
DOI: https://doi.org/10.1109/icpads56603.2022.00113
2022-01-01
Abstract:Spatial-temporal forecasting (e.g., traffic flow forecasting) plays a vital role in various applications and has important research significance. Most existing spatial-temporal forecasting methods improve the forecasting accuracy by stacking discrete modules. However, the discontinuous hidden state trajectories between discrete modules may lead to high numerical errors, parameter redundancy, and model complexity. Neural Controlled Differential Equations (NCDEs) can solve the above issues effectively, which have a mechanism that can continuously adjust the hidden state trajectories according to controlled signals. In this paper, we propose a novel Spatial-Temporal Continuous Encoder-Decoder (STCED) method for spatial-temporal forecasting. Specifically, we first propose an overall spatial-temporal continuous encoder-decoder architecture based on NCDEs, which can not only promote the spatial-temporal message passing, but also capture the periodicity without introducing multiple modules or extra parameters. Then, we refactor the core of NCDEs based on Fast Weight Programmers (FWPs) and Graph Neural Networks (GNNs) to overcome the limitations that previous NCDEs cannot capture long-range spatial-temporal dependencies. We conduct experiments on two representative spatial-temporal datasets, demonstrating the effectiveness and superiority of our proposed algorithm.
What problem does this paper attempt to address?