We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

cs.LG

Change to browse by:

References & Citations

DBLP - CS Bibliography

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo

Computer Science > Machine Learning

Title: ST-MambaSync: The Complement of Mamba and Transformers for Spatial-Temporal in Traffic Flow Prediction

Abstract: Accurate traffic flow prediction is crucial for optimizing traffic management, enhancing road safety, and reducing environmental impacts. Existing models face challenges with long sequence data, requiring substantial memory and computational resources, and often suffer from slow inference times due to the lack of a unified summary state. This paper introduces ST-MambaSync, an innovative traffic flow prediction model that combines transformer technology with the ST-Mamba block, representing a significant advancement in the field. We are the pioneers in employing the Mamba mechanism which is an attention mechanism integrated with ResNet within a transformer framework, which significantly enhances the model's explainability and performance. ST-MambaSync effectively addresses key challenges such as data length and computational efficiency, setting new benchmarks for accuracy and processing speed through comprehensive comparative analysis. This development has significant implications for urban planning and real-time traffic management, establishing a new standard in traffic flow prediction technology.
Comments: 11 pages. arXiv admin note: substantial text overlap with arXiv:2404.13257
Subjects: Machine Learning (cs.LG); Artificial Intelligence (cs.AI)
MSC classes: 53A45
ACM classes: I.2.0
Cite as: arXiv:2404.15899 [cs.LG]
  (or arXiv:2404.15899v3 [cs.LG] for this version)

Submission history

From: Zhiqi Shao [view email]
[v1] Wed, 24 Apr 2024 14:41:41 GMT (995kb,D)
[v2] Fri, 26 Apr 2024 04:05:09 GMT (996kb,D)
[v3] Thu, 9 May 2024 06:48:37 GMT (1292kb,D)

Link back to: arXiv, form interface, contact.