SE-MAConvLSTM: A deep learning framework for short-term traffic flow prediction combining Squeeze-and-Excitation Network and Multi-Attention Convolutional LSTM Network

Rong Zhu, Jie Tang, Xuansen He, Xianlai Zhou, Xiaohui Huang, Fengyun Wu and Songli Chen

PLOS ONE, 2024, vol. 19, issue 12, 1-19

Abstract: Traffic flow prediction is an important part of transportation management and planning. For example, accurate demand prediction of taxis and online car-hailing can reduce the waste of resources caused by empty cars. The prediction of public bicycle flow can be more reasonable to plan the release and deployment of public bicycles. There are three difficulties in traffic flow prediction to achieve higher accuracy. Firstly, more accurately to capture the spatio-temporal correlation existing in historical flow data. Secondly, the weight of each channel in the traffic flow data at the same time interval affects the prediction results. Thirdly, the proportion of closeness, period and trend of traffic flow data affects the prediction results. In this paper, we design a deep learning algorithm for short-term traffic flow prediction, called SE-MAConvLSTM. First, we designed Spatio-Temporal Feature Extraction Module (STFEM), which is composed of Convolutional Neural Network (CNN), Squeeze-and-Excitation Network (SENet), Residual Network (ResNet) and Convolutional LSTM Network (ConvLSTM) to solve the above two problems mentioned. In addition, we design multi-attention modules (MAM) to model the closeness, period and trend of traffic flow data to solve the third problem mentioned above. Finally, the aggregation module was used to integrate the output of the last time interval in STFEM and the output of the multi-attention module. Experiments are carried out on two real data sets, and the results show that the proposed model reduces RMSE by 4.5% and 3.7% respectively compared with the best baseline model.

Date: 2024
References: Add references at CitEc
Citations:

Downloads: (external link)
https://journals.plos.org/plosone/article?id=10.1371/journal.pone.0312601 (text/html)
https://journals.plos.org/plosone/article/file?id= ... 12601&type=printable (application/pdf)

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:plo:pone00:0312601

DOI: 10.1371/journal.pone.0312601

Access Statistics for this article

More articles in PLOS ONE from Public Library of Science
Bibliographic data for series maintained by plosone ().