arXiv:2409.05933cs.LG2024-09

用自监督方法提升交通事故预测的泛化与实时性

Self-Supervised State Space Model for Real-Time Traffic Accident Prediction Using eKAN Networks

  • 设计自监督任务,自动学习时空模式,减少人工特征依赖
  • 提出eKamba模型,在保持高精度下降低计算开销
  • 适用于城市级实时交通风险预警系统部署

跨时空准确预测交通事件对公共安全至关重要。现有方法面临两大挑战:一是泛化能力弱,依赖人工构建多视角特征(如兴趣点分布、路网密度),难以跨城市扩展;二是实时性差,复杂模型导致计算成本过高。为此,我们提出SSL-eKamba框架,通过两个自监督辅助任务增强时空模式表征,提升泛化性;引入eKamba模型,重构KAN架构,采用可学习一元函数激活并结合选择性状态空间机制(Selective SSM),高效捕捉多变量关联,显著提升计算效率。在两个真实世界数据集上的实验表明,该框架持续优于现有最优基线。本方法也可为其他时空任务提供新思路。代码已开源。

原文摘要 · Abstract (English)

Accurate prediction of traffic accidents across different times and regions is vital for public safety. However, existing methods face two key challenges: 1) Generalization: Current models rely heavily on manually constructed multi-view structures, like POI distributions and road network densities, which are labor-intensive and difficult to scale across cities. 2) Real-Time Performance: While some methods improve accuracy with complex architectures, they often incur high computational costs, limiting their real-time applicability. To address these challenges, we propose SSL-eKamba, an efficient self-supervised framework for traffic accident prediction. To enhance generalization, we design two self-supervised auxiliary tasks that adaptively improve traffic pattern representation through spatiotemporal discrepancy awareness. For real-time performance, we introduce eKamba, an efficient model that redesigns the Kolmogorov-Arnold Network (KAN) architecture. This involves using learnable univariate functions for input activation and applying a selective mechanism (Selective SSM) to capture multi-variate correlations, thereby improving computational efficiency. Extensive experiments on two real-world datasets demonstrate that SSL-eKamba consistently outperforms state-of-the-art baselines. This framework may also offer new insights for other spatiotemporal tasks. Our source code is publicly available at http://github.com/KevinT618/SSL-eKamba.

交通预测自监督状态空间模型实时系统

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