arXiv:2605.15083cs.LGcs.AI2026-05

动态调整学习率,让模型更好预测交通事故伤情严重程度。

Novel Dynamic Batch-Sensitive Adam Optimiser for Vehicular Accident Injury Severity Prediction

  • 根据批次难易度动态调节学习率,提升训练稳定性。
  • 在事故伤情预测中达95.22%准确率,精度96.11%,显著优于主流优化器。
  • 适合处理不平衡、序列型数据,适用于实时交通应急响应场景。

优化器的选择对深度学习模型的效率和收敛速度有重要影响。然而,许多常用优化器在处理不平衡和序列数据时表现不佳,难以捕捉少数类模式。本文提出动态批次敏感的Adam优化器(DBS-Adam),通过梯度范数与批次损失的指数移动平均计算批次难度得分,动态调整学习率:对困难批次加大更新,对简单批次减少更新。将DBS-Adam与双向LSTM结合用于事故伤情严重程度预测,采用SMOTE-ENN重采样和焦点损失缓解类别不平衡。四种实验配置对比了基线模型与不同架构,结果表明,在五个随机种子下与AMSGrad、AdamW、AdaBound等先进优化器相比,DBS-Adam性能更具竞争力,精度提升具有统计显著性(p=0.020)。DBS-Adam实现测试准确率95.22%、精确率96.11%、召回率95.28%、F1分数95.39%、测试损失0.0086。该框架可支持实时事故严重性分类,助力紧急响应与道路安全干预,凸显其在不平衡序列数据学习中的价值。

原文摘要 · Abstract (English)

The choice of optimiser is important in deep learning, as it strongly influences model efficiency and speed of convergence. However, many commonly used optimisers encounter difficulties when applied to imbalanced and sequential datasets, limiting their ability to capture patterns of minority classes. In this study, we propose Dynamic Batch-Sensitive Adam (DBS-Adam), an optimiser that dynamically scales the learning rate using a batch difficulty score derived from exponential moving averages of gradient norms and batch loss. DBS-Adam improves training stability and accelerates convergence by increasing updates for difficult batches and reducing them for easier ones. We evaluate DBS-Adam by integrating it with Bi-Directional LSTM networks for accident injury severity prediction, addressing class imbalance through SMOTE-ENN resampling and Focal Loss. Four experimental configurations compare baseline Bi-LSTM models and alternative architectures to assess optimiser impact. Rigorous comparison against state-of-the-art optimisers (AMSGrad, AdamW, AdaBound) across five random seeds demonstrated DBS-Adam's competitive performance with statistically significant precision improvements (p=0.020). Results indicate that DBS-Adam outperforms standard optimisation approaches, achieving 95.22% test accuracy, 96.11% precision, 95.28% recall, 95.39% F1-score, and a test loss of 0.0086. The proposed framework enables effective real-time accident severity classification for targeted emergency response and road safety interventions, demonstrating the value of DBS-Adam for learning from imbalanced sequential data.

优化器事故预测序列数据不平衡数据

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