Self-supervision Spatiotemporal Part-Whole Convolutional Neural Network for Traffic Prediction

Linbo Zhai, Yong Yang, Shudian Song, Shuyue Ma, Xiumin Zhu and Feng Yang

Physica A: Statistical Mechanics and its Applications, 2021, vol. 579, issue C

Abstract: Traffic is a relatively broad concept, including transportation, travel, trade, and internet networks. It is a kind of method to analyze, model and give predictive results for a given sequence with temporal and spatial relations. Traffic forecasting has always been a hot issue for researchers. It is a non-stationary time series with a high degree of nonlinearity, and it is very challenging to accurately forecast it. We propose a novel self-supervision Spatiotemporal Part-Whole Convolutional Neural Network (STPWNet), which simultaneously captures the temporal and spatial correlations of the traffic sequence to accurately predict the traffic data at the next moment. In order to improve the inference accuracy and speed of the deep network, we designed a lightweight convolutional network module with a part-whole structure to improve the accuracy and speed of network prediction. Compared with traditional neural networks, STPWNet has fewer parameters, faster inference speed, and can produce good prediction performance on a variety of traffic data sets. Experiments show that our proposed network uses only a small number of parameters compared with other networks, and can achieve quite good performance. Our code is available on https://github.com/zhu-xm1/STPWNet.

Keywords: Traffic prediction; Deep learning; Convolutional neural network (search for similar items in EconPapers)
Date: 2021
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (2)

Downloads: (external link)
http://www.sciencedirect.com/science/article/pii/S0378437121004143
Full text for ScienceDirect subscribers only. Journal offers the option of making the article available online on Science direct for a fee of $3,000

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:eee:phsmap:v:579:y:2021:i:c:s0378437121004143

DOI: 10.1016/j.physa.2021.126141

Access Statistics for this article

Physica A: Statistical Mechanics and its Applications is currently edited by K. A. Dawson, J. O. Indekeu, H.E. Stanley and C. Tsallis

More articles in Physica A: Statistical Mechanics and its Applications from Elsevier
Bibliographic data for series maintained by Catherine Liu ().