用小型可解释模型预测城市交通事故严重程度,效果优于大语言模型。
RaX-Crash: A Resource Efficient and Explainable Small Model Pipeline with an Application to City Scale Injury Severity Prediction
- 构建统一特征表,用随机森林和XGBoost处理百万级交通数据。
- 模型准确率达0.78,显著高于小型语言模型的0.49~0.59。
- 可解释性分析揭示时间、地点和人为因素是伤情关键影响因子。
纽约市每年报告超十万起机动车碰撞事故,带来巨大伤亡与公共健康负担。本文提出RaX-Crash,一种资源高效且可解释的小型模型管道,用于在官方纽约市机动车碰撞数据集上进行结构化伤情严重度预测。该系统整合三张关联表格,包含数千万条记录,采用分片存储构建统一特征架构,并在工程化表格特征上训练紧凑的树模型集成(随机森林与XGBoost),与本地部署的小型语言模型(SLMs)结合文本摘要的提示结果进行对比。在时间上留出的测试集上,XGBoost与随机森林分别达到0.7828与0.7794的准确率,明显优于SLMs(0.594与0.496);类别不平衡分析显示,简单加权可提升致命伤召回率,代价仅小幅降低准确率;SHAP归因分析突出人类脆弱性、时间与位置为预测严重度的主要驱动因素。整体表明,可解释的小型模型集成仍是城市级伤情分析的强大基线,而将表格预测器与SLM生成叙事相结合的混合流程,在不牺牲可扩展性的前提下提升了沟通效率。
原文摘要 · Abstract (English)
New York City reports over one hundred thousand motor vehicle collisions each year, creating substantial injury and public health burden. We present RaX-Crash, a resource efficient and explainable small model pipeline for structured injury severity prediction on the official NYC Motor Vehicle Collisions dataset. RaX-Crash integrates three linked tables with tens of millions of records, builds a unified feature schema in partitioned storage, and trains compact tree based ensembles (Random Forest and XGBoost) on engineered tabular features, which are compared against locally deployed small language models (SLMs) prompted with textual summaries. On a temporally held out test set, XGBoost and Random Forest achieve accuracies of 0.7828 and 0.7794, clearly outperforming SLMs (0.594 and 0.496); class imbalance analysis shows that simple class weighting improves fatal recall with modest accuracy trade offs, and SHAP attribution highlights human vulnerability factors, timing, and location as dominant drivers of predicted severity. Overall, RaX-Crash indicates that interpretable small model ensembles remain strong baselines for city scale injury analytics, while hybrid pipelines that pair tabular predictors with SLM generated narratives improve communication without sacrificing scalability.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。