Mamba-Based Macro–MicroSpatio-Temporal Model for Traffic Flow Prediction

Traffic flow prediction plays an important role in intelligent transportation systems. However, accurately modeling traffic dynamics remains challenging due to complex temporal correlations and spatial interactions across road networks. In this work, we propose a Mamba-based macro–micro spatio-temporal model for traffic flow prediction. Unlike graph-based approaches that rely on predefined adjacency matrices to model spatial relationships, our method treats sensor nodes as sequence elements and applies Mamba blocks along the spatial dimension. Through the global receptive field of the structured state space model, spatial dependencies are implicitly learned without requiring explicit graph structures. The proposed architecture consists of stacked spatio-temporal blocks, each composed of two Macro Feature Blocks and one Micro Feature Block. The Macro Feature Blocks are designed to capture global temporal dependencies and spatial interactions across all nodes, while the Micro Feature Block focuses on modeling localized spatio-temporal patterns at a finer granularity. By applying structured state space modeling along both temporal and spatial dimensions, the model is able to capture long-range temporal dependencies and global spatial correlations without relying on explicit graph structures. Experiments conducted on four real-world datasets demonstrate that the proposed model achieves competitive or improved performance compared with existing baseline methods under standard evaluation metrics.

Authors

Institutions

Publication Details

Journal
Electronics
Published
2026-03-23
DOI
https://doi.org/10.3390/electronics15061327
Primary Topic
Traffic Prediction and Management Techniques
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Mamba-Based Macro–MicroSpatio-Temporal Model for Traffic Flow Prediction

Fayang Lan, Haoning Lv, Weijie Xiu
Electronics
Traffic Prediction and Management Techniques
article

Mamba-Based Macro–MicroSpatio-Temporal Model for Traffic Flow Prediction

Fayang Lan, Haoning Lv, Weijie Xiu
article en

Abstract

Traffic flow prediction plays an important role in intelligent transportation systems. However, accurately modeling traffic dynamics remains challenging due to complex temporal correlations and spatial interactions across road networks. In this work, we propose a Mamba-based macro–micro spatio-temporal model for traffic flow prediction. Unlike graph-based approaches that rely on predefined adjacency matrices to model spatial relationships, our method treats sensor nodes as sequence elements and applies Mamba blocks along the spatial dimension. Through the global receptive field of the structured state space model, spatial dependencies are implicitly learned without requiring explicit graph structures. The proposed architecture consists of stacked spatio-temporal blocks, each composed of two Macro Feature Blocks and one Micro Feature Block. The Macro Feature Blocks are designed to capture global temporal dependencies and spatial interactions across all nodes, while the Micro Feature Block focuses on modeling localized spatio-temporal patterns at a finer granularity. By applying structured state space modeling along both temporal and spatial dimensions, the model is able to capture long-range temporal dependencies and global spatial correlations without relying on explicit graph structures. Experiments conducted on four real-world datasets demonstrate that the proposed model achieves competitive or improved performance compared with existing baseline methods under standard evaluation metrics.

ElectronicsVol. 15(6)
North China University of Technology (CN), Guizhou Institute of Technology (CN)
Sustainable cities and communities
Openalex Percentile: Top 11%
Traffic Prediction and Management Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.