WSG-CNN.pdf (3.22 MB)
Download file

Wavelet Based Sampling and Graph Convolutional Neural Network: A Deep Learning Method for Traffic Data Learning and Forecasting

Download (3.22 MB)
posted on 2023-05-23, 15:17 authored by Xiaohong Nian, Qidong Cai, Yao HuangYao Huang, Hongyun Xiong

Traffic network data is characterized by high complex spatiotemporal dependence and heterogeneity, which poses great challenges for traffic forecasting. To address this challenge, we propose a novel model framework, named Wavelet Based Sampling and Graph Convolutional Neural Network (WSG-CNN). To solve the problem of complex spatiotemporal dependencies in the data, we propose a method that combines sampling convolution and gated temporal convolution network (TCN) to extract temporal features. This model uses sampling convolution to improve the Receptive field of shallow network of TCN model, while using TCN to reduce the number of convolution layers, thereby effectively enhancing the ability of the model to extract temporal features by preserving the original temporal features while also deeply fusing them. We also integrate graph convolution network into the model to effectively handle the spatiotemporal dependencies of the data. Traffic data has heterogeneity, characterized by periodic and sudden features. To separate these two features from the frequency domain, we employ wavelet transform and add them to the original data for training. This expands the explicit features of the original traffic data and effectively improves the predictive ability of the model. Experiments are conducted on six public datasets, which demonstrates that our model achieves state-of-the-art performance over baseline models both on traffic flow forecasting and traffic speed forecasting. Our code and datasets are available on Github.


National Natural Science Foundation of China

National Natural Science Foundation of China

Find out more...


Email Address of Submitting Author

ORCID of Submitting Author


Submitting Author's Institution

Central South University

Submitting Author's Country

  • China

Usage metrics