HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks

Zonghan Wu; Shirui Pan; Guodong Long; Jing Jiang; Xiaojun Chang; Chengqi Zhang

Connecting the Dots: Multivariate Time Series Forecasting with Graph Neural Networks

Abstract

Modeling multivariate time series has long been a subject that has attracted researchers from a diverse range of fields including economics, finance, and traffic. A basic assumption behind multivariate time series forecasting is that its variables depend on one another but, upon looking closely, it is fair to say that existing methods fail to fully exploit latent spatial dependencies between pairs of variables. In recent years, meanwhile, graph neural networks (GNNs) have shown high capability in handling relational dependencies. GNNs require well-defined graph structures for information propagation which means they cannot be applied directly for multivariate time series where the dependencies are not known in advance. In this paper, we propose a general graph neural network framework designed specifically for multivariate time series data. Our approach automatically extracts the uni-directed relations among variables through a graph learning module, into which external knowledge like variable attributes can be easily integrated. A novel mix-hop propagation layer and a dilated inception layer are further proposed to capture the spatial and temporal dependencies within the time series. The graph learning, graph convolution, and temporal convolution modules are jointly learned in an end-to-end framework. Experimental results show that our proposed model outperforms the state-of-the-art baseline methods on 3 of 4 benchmark datasets and achieves on-par performance with other approaches on two traffic datasets which provide extra structural information.

Code Repositories

nnzhan/MTGNN
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
traffic-prediction-on-expy-tky-1MTGNN
1 step MAE: 5.86
3 step MAE: 6.49
6 step MAE: 6.81
traffic-prediction-on-ne-bjMTGNN
12 steps MAE: 4.90
univariate-time-series-forecasting-onMTGNN (3 step)
RRSE: 0.0745
univariate-time-series-forecasting-onMTGNN (12 step)
RRSE: 0.0916
univariate-time-series-forecasting-onMTGNN (24 step)
RRSE: 0.0953
univariate-time-series-forecasting-onMTGNN (6 step)
RRSE: 0.0878

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp