Research on Improved Traffic Flow Prediction Network Based on CapsNet

Bin Qiu () and Yun Zhao
Additional contact information
Bin Qiu: School of Information and Electronic Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, China
Yun Zhao: School of Information and Electronic Engineering, Zhejiang University of Science and Technology, Hangzhou 310023, China

Sustainability, 2022, vol. 14, issue 23, 1-17

Abstract: Traffic flow prediction is the basis and key to the realization of an intelligent transportation system. The road traffic flow prediction of city-level complex road network can be realized using traffic big data. In the traffic prediction task, the limitation of the convolutional neural network (CNN) for modeling the spatial relationship of the road network and the insufficient feature extraction of the shallow network make it impossible to accurately predict the traffic flow. In order to improve the prediction performance of the model, this paper proposes an improved capsule network (MCapsNet) based on capsule network (CapsNet). First, in the preliminary feature extraction stage, a depthwise separable convolutional block is added to expand the feature channel to enrich channel information. Subsequently, in order to strengthen the reuse of important features and suppress useless information, channel attention is used to selectively reinforce learning of extended channel information so that the network can extract a large number of high-dimensional important features and improve the ability of network feature learning and expression. At the same time, in order to alleviate the feature degradation during training and the channel collapse problem easily caused by deep convolution, a shortcut connection, and a modified linear bottleneck layer structure are added to the convolution layer. The bottleneck layer adds the depth convolution and channel attention connection to the residual block of the network. Finally, the deep local feature information extracted from the improved convolutional layer is vectorized into the form of a capsule, which can more accurately model the details of road network attributes and features and improve the model expression power and prediction performance. The network is tested on the Wenyi Road dataset and the public dataset SZ-taxi. Compared with other models, the evaluation indicators of MCapsNet are better than other models in the tests of different time periods and predictors. Compared with CapsNet, the RMSE index of MCapsNet is reduced by 10.50% in the full period of Wenyi Road, 4.66% in the peak period, 9.78% in the off-peak period, and 6.07% in the SZ-tax dataset. The experimental results verify the effectiveness of the model improvement.

Keywords: intelligent transportation; traffic flow prediction; attention mechanism; temporal-spatial correlation (search for similar items in EconPapers)
JEL-codes: O13 Q Q0 Q2 Q3 Q5 Q56 (search for similar items in EconPapers)
Date: 2022
References: View references in EconPapers View complete reference list from CitEc
Citations: View citations in EconPapers (1)

Downloads: (external link)
https://www.mdpi.com/2071-1050/14/23/15996/pdf (application/pdf)
https://www.mdpi.com/2071-1050/14/23/15996/ (text/html)

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:gam:jsusta:v:14:y:2022:i:23:p:15996-:d:989112

Access Statistics for this article

Sustainability is currently edited by Ms. Alexandra Wu

More articles in Sustainability from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().