arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.00543cs.CV

基于掩码图像建模自监督学习的新型驾驶员分心行为检测方法

A Novel Driver Distraction Behavior Detection Method Based on Self-supervised Learning with Masked Image Modeling

Shaanxi Kangfu Hospital(陕西康复医院)

Yingzhi Zhang, Taiguo Li, Chao Li, Xinghong Zhou

AI总结:

针对驾驶员分心检测标注成本高、语义建模与泛化不足的问题,提出基于 MIM 和改进 Swin Transformer 的自监督方法,准确率达 99.60%。

摘要:

驾驶员分心每年都会导致大量交通事故,造成经济损失和人员伤亡。目前,商用车的自动化水平远未达到完全无人驾驶,驾驶员在车辆操作和控制中仍发挥着重要作用。因此,驾驶员分心行为检测对道路安全至关重要。当前,驾驶员分心检测主要依赖传统卷积神经网络(CNN)和监督学习方法。然而,仍存在标注数据集成本高、捕获高层语义信息能力有限以及泛化性能较弱等挑战。为解决这些问题,本文提出一种基于掩码图像建模的新型自监督学习方法,用于驾驶员分心行为检测。首先,引入掩码图像建模(MIM)自监督学习框架,以解决数据集标注造成的严重人力和物力消耗问题。其次,采用 Swin Transformer 作为编码器。通过重构 Swin Transformer block,并调整各个阶段中窗口多头自注意力(W-MSA)和移位窗口多头自注意力(SW-MSA)检测头的数量分布来提升性能,从而使模型更加轻量化。最后,结合多种数据增强策略与最佳随机掩码策略,以增强模型的识别和泛化能力。在一个大规模驾驶员分心行为数据集上的测试结果表明,本文提出的自监督学习方法达到了 99.60% 的准确率,接近先进监督学习方法的优异性能。代码已公开发布于 github.com/Rocky1salady-killer/SL-DDBD。

英文摘要:

Driver distraction causes a significant number of traffic accidents every year, resulting in economic losses and casualties. Currently, the level of automation in commercial vehicles is far from completely unmanned, and drivers still play an important role in operating and controlling the vehicle. Therefore, driver distraction behavior detection is crucial for road safety. At present, driver distraction detection primarily relies on traditional convolutional neural networks (CNN) and supervised learning methods. However, there are still challenges such as the high cost of labeled datasets, limited ability to capture high-level semantic information, and weak generalization performance. In order to solve these problems, this paper proposes a new self-supervised learning method based on masked image modeling for driver distraction behavior detection. Firstly, a self-supervised learning framework for masked image modeling (MIM) is introduced to solve the serious human and material consumption issues caused by dataset labeling. Secondly, the Swin Transformer is employed as an encoder. Performance is enhanced by reconfiguring the Swin Transformer block and adjusting the distribution of the number of window multi-head self-attention (W-MSA) and shifted window multi-head self-attention (SW-MSA) detection heads across all stages, which leads to model more lightening. Finally, various data augmentation strategies are used along with the best random masking strategy to strengthen the model's recognition and generalization ability. Test results on a large-scale driver distraction behavior dataset show that the self-supervised learning method proposed in this paper achieves an accuracy of 99.60%, approximating the excellent performance of advanced supervised learning methods. Our code is publicly available at github.com/Rocky1salady-killer/SL-DDBD.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑