用两阶段框架提升分心驾驶行为定位精度与效率
A Two-stage Transformer Framework for Temporal Localization of Distracted Driver Behaviors
- 先用VideoMAE提取特征,再用自掩码注意力+多尺度模块精确定位
- 大模型达92.67% mAP,轻量版仅需101.85 GFLOPs/段
- 适合车队管理、交通检查等实时监控场景
从车载视频流中识别危险驾驶行为对提升道路安全至关重要,但现有时间动作定位方法难以兼顾准确率与计算效率。本文提出一种专为驾驶员监控设计的两阶段框架,适用于交通安检或车队管理等周期性检查场景。方法结合VideoMAE特征提取与增强型自掩码注意力(AMA)检测器,并引入空间金字塔池化-快速(SPPF)模块以捕捉多尺度时间特征。实验表明,在特征提取阶段,ViT-Giant主干网络在测试集上达到88.09% Top-1准确率;而基于ViT的轻量变体虽降至82.55%准确率,但计算开销大幅降低(每片段仅101.85 GFLOPs,相较巨模型的1584.06 GFLOPs)。在下游定位任务中,所有配置均因SPPF集成而性能提升,其中ViT-Giant + SPPF模型取得92.67%的峰值mAP,轻量配置亦表现稳健。
原文摘要 · Abstract (English)
The identification of hazardous driving behaviors from in-cabin video streams is essential for enhancing road safety and supporting the detection of traffic violations and unsafe driver actions. However, current temporal action localization techniques often struggle to balance accuracy with computational efficiency. In this work, we develop and evaluate a temporal action localization framework tailored for driver monitoring scenarios, particularly suitable for periodic inspection settings such as transportation safety checkpoints or fleet management assessment systems. Our approach follows a two-stage pipeline that combines VideoMAE-based feature extraction with an Augmented Self-Mask Attention (AMA) detector, enhanced by a Spatial Pyramid Pooling-Fast (SPPF) module to capture multi-scale temporal features. Experimental results reveal a distinct trade-off between model capacity and efficiency. At the feature extraction stage, the ViT-Giant backbone delivers higher representations with 88.09% Top-1 test accuracy, while the ViT-based variant proves to be a practical alternative, achieving 82.55% accuracy with significantly lower computational fine-tuning costs (101.85 GFLOPs/segment compared to 1584.06 GFLOPs/segment for Giant). In the downstream localization task, the integration of SPPF consistently improves performance across all configurations. Notably, the ViT-Giant + SPPF model achieves a peak mAP of 92.67%, while the lightweight ViT-based configuration maintains robust results.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。