CrashFormer: A Multimodal Architecture to Predict the Risk of Crash

📄 arXiv: 2402.05151v1 📥 PDF

作者: Amin Karimi Monsefi, Pouya Shiri, Ahmad Mohammadshirazi, Nastaran Karimi Monsefi, Ron Davies, Sobhan Moosavi, Rajiv Ramnath

分类: cs.LG, cs.AI

发布日期: 2024-02-07

备注: The paper is accepted In 1st ACM SIGSPATIAL International Workshop on Advances in Urban-AI (UrbanAI 23), November 13, 2023, Hamburg, Germany

DOI: 10.1145/3615900.3628769


💡 一句话要点

提出CrashFormer以解决交通事故预测问题

🎯 匹配领域: 支柱九:具身大模型 (Embodied Foundation Models)

关键词: 交通事故预测 多模态架构 机器学习 数据融合 公共安全

📋 核心要点

  1. 现有事故预测方法在可推广性、可重复性和实际应用方面存在诸多不足,限制了其有效性。
  2. 本文提出的CrashFormer通过整合历史事故、天气、地图图像和人口统计信息等多模态数据,提供了一种新的事故风险预测方法。
  3. 在10个主要美国城市的实地实验中,CrashFormer的F1-score平均提高了1.8%,显示出其优越的性能。

📝 摘要(中文)

减少交通事故是全球公共安全的重要问题。事故预测是提高交通安全的关键,能够在事故发生前采取主动措施,并为安全政策、法规和针对性干预提供信息。尽管过去几十年有许多关于事故预测的研究,但由于输入数据或问题表述的限制,许多研究在可推广性、可重复性或实际应用方面存在不足。为了解决这些问题,本文提出了CrashFormer,这是一种多模态架构,利用相对易于获取的输入数据,如事故历史、天气信息、地图图像和人口统计信息。该模型能够以合理的频率(每六小时)预测5.161平方公里地理位置的未来事故风险。经过在美国10个主要城市的广泛实地实验,CrashFormer在使用“稀疏”输入数据时,平均F1-score比现有的序列和非序列模型提高了1.8%。

🔬 方法详解

问题定义:本文旨在解决交通事故预测中的数据可获取性和模型有效性问题。现有方法往往依赖于复杂或难以获取的数据,导致其在实际应用中的局限性。

核心思路:CrashFormer的核心思路是利用多模态输入数据,结合历史事故、天气、地图图像和人口统计信息,以提高事故风险预测的准确性和实用性。通过这种方式,模型能够在较短的时间内提供可靠的预测结果。

技术框架:CrashFormer由五个主要模块组成:序列编码器用于处理历史事故和天气数据,图像编码器用于处理地图图像数据,原始数据编码器用于处理人口统计信息,特征融合模块用于聚合编码特征,分类器则接受聚合后的数据并进行预测。

关键创新:CrashFormer的主要创新在于其多模态架构设计,能够有效整合不同类型的数据源,从而克服了传统模型在数据单一性上的局限。与现有方法相比,CrashFormer在输入数据的多样性和模型的适应性上具有显著优势。

关键设计:在模型设计中,采用了适合不同数据类型的编码器,确保信息的有效提取和融合。同时,模型的损失函数经过优化,以提高预测的准确性和稳定性。

🖼️ 关键图片

fig_0
fig_1
fig_2

📊 实验亮点

在10个主要美国城市的实地实验中,CrashFormer在使用“稀疏”输入数据时,平均F1-score比现有的序列和非序列模型提高了1.8%。这一结果表明,CrashFormer在事故风险预测方面具有显著的性能优势,能够更好地应对实际应用中的挑战。

🎯 应用场景

CrashFormer的研究成果在交通安全领域具有广泛的应用潜力。通过准确预测事故风险,该模型可以为交通管理部门提供决策支持,帮助制定更有效的安全政策和干预措施,从而降低交通事故发生率,提升公共安全水平。未来,该技术还可以扩展到智能交通系统和自动驾驶车辆的安全监控中。

📄 摘要(原文)

Reducing traffic accidents is a crucial global public safety concern. Accident prediction is key to improving traffic safety, enabling proactive measures to be taken before a crash occurs, and informing safety policies, regulations, and targeted interventions. Despite numerous studies on accident prediction over the past decades, many have limitations in terms of generalizability, reproducibility, or feasibility for practical use due to input data or problem formulation. To address existing shortcomings, we propose CrashFormer, a multi-modal architecture that utilizes comprehensive (but relatively easy to obtain) inputs such as the history of accidents, weather information, map images, and demographic information. The model predicts the future risk of accidents on a reasonably acceptable cadence (i.e., every six hours) for a geographical location of 5.161 square kilometers. CrashFormer is composed of five components: a sequential encoder to utilize historical accidents and weather data, an image encoder to use map imagery data, a raw data encoder to utilize demographic information, a feature fusion module for aggregating the encoded features, and a classifier that accepts the aggregated data and makes predictions accordingly. Results from extensive real-world experiments in 10 major US cities show that CrashFormer outperforms state-of-the-art sequential and non-sequential models by 1.8% in F1-score on average when using ``sparse'' input data.