无需校准,实时自适应脑电检测困倦,提升跨人泛化能力。
Calibration-Free EEG-based Driver Drowsiness Detection with Online Test-Time Adaptation
- 在线测试时自适应调整批归一化参数,动态适配个体差异。
- 在模拟驾驶数据集上达81.73%平均F1分数,比最优基线高11.73%。
- 适合实际场景中无校准需求的车载困倦监测系统部署。
困倦驾驶是交通事故的重要诱因,促使研究者探索基于脑电图(EEG)的困倦检测系统。然而,心理和生理因素导致的脑电信号固有变异,使系统需繁琐校准。尤其个体间脑电信号差异引发领域偏移问题,难以推广至未见受试者。为此,本文提出一种新型驾驶员困倦检测框架,利用在线测试时自适应(TTA)方法动态调整目标受试者分布。所提方法更新批归一化(BN)层中的可学习参数,同时保留预训练归一化统计量,实现测试时有效适应。引入记忆库动态管理流式脑电片段,依据负能量得分与持续时间选择可靠样本。此外,通过原型学习确保对随时间变化的分布偏移具有鲁棒性。在模拟环境中采集的持续注意力驾驶数据集上验证,困倦通过单调车道保持任务中的反应延迟时间估计。实验表明,本方法优于所有基线,平均F1分数达81.73%,较最佳TTA基线提升11.73%。结果表明,该方法显著增强非独立同分布场景下脑电困倦检测系统的适应性。
原文摘要 · Abstract (English)
Drowsy driving is a growing cause of traffic accidents, prompting recent exploration of electroencephalography (EEG)-based drowsiness detection systems. However, the inherent variability of EEG signals due to psychological and physical factors necessitates a cumbersome calibration process. In particular, the inter-subject variability of EEG signals leads to a domain shift problem, which makes it challenging to generalize drowsiness detection models to unseen target subjects. To address these issues, we propose a novel driver drowsiness detection framework that leverages online test-time adaptation (TTA) methods to dynamically adjust to target subject distributions. Our proposed method updates the learnable parameters in batch normalization (BN) layers, while preserving pretrained normalization statistics, resulting in a modified configuration that ensures effective adaptation during test time. We incorporate a memory bank that dynamically manages streaming EEG segments, selecting samples based on their reliability determined by negative energy scores and persistence time. In addition, we introduce prototype learning to ensure robust predictions against distribution shifts over time. We validated our method on the sustained-attention driving dataset collected in a simulated environment, where drowsiness was estimated from delayed reaction times during monotonous lane-keeping tasks. Our experiments show that our method outperforms all baselines, achieving an average F1-score of 81.73\%, an improvement of 11.73\% over the best TTA baseline. This demonstrates that our proposed method significantly enhances the adaptability of EEG-based drowsiness detection systems in non-i.i.d. scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。