arXiv:2507.03927cs.LG2025-07被引 4

用Mamba架构统一预测交通多变量,更准且更省参数。

MCST-Mamba: Multivariate Mamba-Based Model for Traffic Prediction

  • 基于Mamba构建多通道时空模型,原生处理速度、流量、占有率等多变量输入。
  • 在多个交通特征上同时预测,相比单变量模型提升整体预测精度。
  • 参数量更低,适合实际部署,尤其适合需要多维交通监控的系统。

精准交通预测对智能交通系统至关重要,可实现高效路径规划、缓解拥堵并主动调控交通。然而,动态道路状况、不同区域间变化的交通模式以及天气、事故等外部因素使得预测极具挑战。交通数据通常包含速度、流量、占有率等多个相互关联的测量值,但多数深度学习方法仅预测单一变量或需为每个变量单独建模,限制了跨通道联合模式的捕捉能力。为此,我们提出基于Mamba选择性状态空间架构的多通道时空(MCST)Mamba模型,原生支持多变量输入,并同时建模所有交通特征。该模型整合自适应时空嵌入,将时间序列与空间传感器交互分别由两个专用的Mamba模块处理,提升表征学习效果。不同于以往仅评估单一通道的方法,我们在所有交通特征上统一评估MCST-Mamba,更贴近实际拥堵形成机制。实验表明,该模型在保持较低参数量的同时,实现了优异的预测性能。

原文摘要 · Abstract (English)

Accurate traffic prediction plays a vital role in intelligent transportation systems by enabling efficient routing, congestion mitigation, and proactive traffic control. However, forecasting is challenging due to the combined effects of dynamic road conditions, varying traffic patterns across different locations, and external influences such as weather and accidents. Traffic data often consists of several interrelated measurements - such as speed, flow and occupancy - yet many deep-learning approaches either predict only one of these variables or require a separate model for each. This limits their ability to capture joint patterns across channels. To address this, we introduce the Multi-Channel Spatio-Temporal (MCST) Mamba model, a forecasting framework built on the Mamba selective state-space architecture that natively handles multivariate inputs and simultaneously models all traffic features. The proposed MCST-Mamba model integrates adaptive spatio-temporal embeddings and separates the modeling of temporal sequences and spatial sensor interactions into two dedicated Mamba blocks, improving representation learning. Unlike prior methods that evaluate on a single channel, we assess MCST-Mamba across all traffic features at once, aligning more closely with how congestion arises in practice. Our results show that MCST-Mamba achieves strong predictive performance with a lower parameter count compared to baseline models.

交通预测Mamba多变量建模时空模型

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。