Disentangled traffic forecasting via efficient graph neural network

Abstract

Listen

Translate article

Abstract Traffic prediction is important to intelligent transportation systems, but it poses significant challenges due to the complex spatio-temporal correlation of traffic data. To address these challenges, we decompose complex traffic time series into events and trends using discrete wavelet transform. However, the high-frequency information decomposed by wavelet transform contains noise, so we designed a module to denoise the high-frequency events. We model the spatio-temporal correlations of trends and events separately using two mutually independent spatio-temporal networks. We use causal dilation convolution with local receptive fields and attention with global receptive fields to capture the correlation of fluctuating time series and the correlation of stable time series, respectively. We capture spatial correlation using a GNN optimized by an attention mechanism. Finally, we merge useful information from volatile events into predictable trends using adaptive event fusion. Our model significantly outperforms nine existing traffic prediction methods across four real datasets.