An Jianpeng, Cai Qing, Sun Xinlin, Li Mengyu, Ma Chao, Gao Zhongke
School of Electrical and Information Engineering, Tianjin University, Tianjin, 300072 China.
School of Artificial Intelligence, Tiangong University, Tianjin, 300387 China.
Cogn Neurodyn. 2024 Oct;18(5):3181-3194. doi: 10.1007/s11571-024-10141-w. Epub 2024 Jul 11.
Fatigue driving significantly contributes to global vehicle accidents and fatalities, making driver fatigue level estimation crucial. Electroencephalography (EEG) is a proven reliable predictor of brain states. With Deep Learning (DL) advancements, brain state estimation algorithms have improved significantly. Nonetheless, EEG's multi-domain nature and the intricate spatial-temporal-frequency correlations among EEG channels present challenges in developing precise DL models. In this work, we introduce an innovative Attention-based Cross-Frequency Graph Convolutional Network (ACF-GCN) for estimating drivers' reaction times using EEG signals from theta, alpha, and beta bands. This method utilizes a multi-head attention mechanism to detect long-range dependencies between EEG channels across frequencies. Concurrently, the transformer's encoder module learns node-level feature maps from the attention-score matrix. Subsequently, the Graph Convolutional Network (GCN) integrates this matrix with feature maps to estimate driver reaction time. Our validation on a publicly available dataset shows that ACF-GCN outperforms several state-of-the-art methods. We also explore the brain dynamics within the cross-frequency attention-score matrix, identifying theta and alpha bands as key influencers in fatigue estimating performance. The ACF-GCN method advances brain state estimation and provides insights into the brain dynamics underlying multi-channel EEG signals.
疲劳驾驶是导致全球交通事故和死亡的重要因素,因此对驾驶员疲劳程度进行评估至关重要。脑电图(EEG)已被证明是一种可靠的大脑状态预测指标。随着深度学习(DL)技术的进步,大脑状态估计算法有了显著改进。尽管如此,EEG的多域性质以及EEG通道之间复杂的时空频率相关性给开发精确的DL模型带来了挑战。在这项工作中,我们引入了一种创新的基于注意力的跨频率图卷积网络(ACF-GCN),用于利用来自θ、α和β波段的EEG信号估计驾驶员的反应时间。该方法利用多头注意力机制来检测不同频率下EEG通道之间的长程依赖关系。同时,Transformer的编码器模块从注意力得分矩阵中学习节点级特征图。随后,图卷积网络(GCN)将该矩阵与特征图进行整合,以估计驾驶员的反应时间。我们在一个公开可用数据集上的验证表明,ACF-GCN优于几种现有最先进的方法。我们还探索了跨频率注意力得分矩阵中的大脑动力学,确定θ和α波段是疲劳估计性能的关键影响因素。ACF-GCN方法推动了大脑状态估计的发展,并为多通道EEG信号背后的大脑动力学提供了见解。