A new approach to COVID-19 data mining: A deep spatial–temporal prediction model based on tree structure for traffic revitalization index

Abstract

The outbreak of the COVID-19 epidemic has had a huge impact on a global scale and its impact has covered almost all human industries. The Chinese government enacted a series of policies to restrict the transportation industry in order to slow the spread of the COVID-19 virus in early 2020. With the gradual control of the COVID-19 epidemic and the reduction of confirmed cases, the Chinese transportation industry has gradually recovered. The traffic revitalization index is the main indicator for evaluating the degree of recovery of the urban transportation industry after being affected by the COVID-19 epidemic. The prediction research of traffic revitalization index can help the relevant government departments to know the state of urban traffic from the macro level and formulate relevant policies. Therefore, this study proposes a deep spatial–temporal prediction model based on tree structure for the traffic revitalization index. The model mainly includes spatial convolution module, temporal convolution module and matrix data fusion module. The spatial convolution module builds a tree convolution process based on the tree structure that can contain directional features and hierarchical features of urban nodes. The temporal convolution module constructs a deep network for capturing temporal dependent features of the data in the multi-layer residual structure. The matrix data fusion module can perform multi-scale fusion of COVID-19 epidemic data and traffic revitalization index data to further improve the prediction effect of the model. In this study, experimental comparisons between our model and multiple baseline models are conducted on real datasets. The experimental results show that our model has an average improvement of 21%, 18%, and 23% in MAE, RMSE and MAPE indicators, respectively.