Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Long-Range Transformers for Dynamic Spatiotemporal Forecasting
(Submitted on 24 Sep 2021 (this version), latest version 19 May 2022 (v2))
Abstract: Multivariate Time Series Forecasting (TSF) focuses on the prediction of future values based on historical context. In these problems, dependent variables provide additional information or early warning signs of changes in future behavior. State-of-the-art forecasting models rely on neural attention between timesteps. This allows for temporal learning but fails to consider distinct spatial relationships between variables. This paper addresses the problem by translating multivariate TSF into a novel spatiotemporal sequence formulation where each input token represents the value of a single variable at a given timestep. Long-Range Transformers can then learn interactions between space, time, and value information jointly along this extended sequence. Our method, which we call Spacetimeformer, scales to high dimensional forecasting problems dominated by Graph Neural Networks that rely on predefined variable graphs. We achieve competitive results on benchmarks from traffic forecasting to electricity demand and weather prediction while learning spatial and temporal relationships purely from data.
Submission history
From: Jake Grigsby [view email][v1] Fri, 24 Sep 2021 22:11:46 GMT (564kb,D)
[v2] Thu, 19 May 2022 18:46:35 GMT (748kb,D)
Link back to: arXiv, form interface, contact.