A lightweight attention-based multi-frequency topology learning framework for driving fatigue detection

Abstract

Driving fatigue has been one of the major causes of traffic accident. Efficient and accurate detection of driving fatigue are a legitimate public concern. In this paper, we conduct the simulated driving experiments and an EEG-based driving fatigue detection framework integrating multilayer brain network and convolutional neural network (CNN) is developed. This lightweight attention-based multi-frequency topology learning (AMFTL) framework first captures the fatigue-related multi-frequency brain topological information and then feeds it into a CNN-based topology feature extraction (TFE) module to fully explore and integrate the critical topological features. The quantitative analysis results show that there are significant differences in brain topologies between the alert and fatigue states. And experimental results show that our proposed framework achieves an average detection accuracy of 94.71% for driving fatigue, which outperforms the current state-of-the-art methods. This proposed framework is expected to open new venues for EEG-based brain state analysis, and holds promising practical application potential.