End-to-end fatigue driving EEG signal detection model based on improved temporal-graph convolution network

Abstract

Fatigue driving is one of the leading causes of traffic accidents, so fatigue driving detection technology plays a crucial role in road safety. The physiological information-based fatigue detection methods have the advantage of objectivity and accuracy. Among many physiological signals, EEG signals are considered to be the most direct and promising ones. Most traditional methods are challenging to train and do not meet real-time requirements. To this end, we propose an end-to-end temporal and graph convolution-based (MATCN-GT) fatigue driving detection algorithm. The MATCN-GT model consists of a multi-scale attentional temporal convolutional neural network block (MATCN block) and a graph convolutional-Transformer block (GT block). Among them, the MATCN block extracts features directly from the original EEG signal without a priori information, and the GT block processes the features of EEG signals between different electrodes. In addition, we design a multi-scale attention module to ensure that valuable information on electrode correlations will not be lost. We add a Transformer module to the graph convolutional network, which can better capture the dependencies between long-distance electrodes. We conduct experiments on the public dataset SEED-VIG, and the accuracy of the MATCN-GT model reached 93.67%, outperforming existing algorithms. Furthermore, compared with the traditional graph convolutional neural network, the GT block has improved the accuracy rate by 3.25%. The accuracy of the MATCN block on different subjects is higher than the existing feature extraction methods.