With the development of brain-computer interface (BCI), electroencephalogram (EEG) is considered to be one of the best physiological signals to detect the fatigue state of drivers due to its advantages of extremely high time resolution and low use cost. However, since EEG is sensitive to noise/artifacts, and there is non-stationarity and low signal-to-noise ratio in inter-subject/intra-subject, the accurate estimation of driver drowsiness state remains challenging. Therefore, this paper proposes the EEG-based TSK fuzzy graph neural network (TSKG) to solve the regression problem of driver drowsiness estimation by combining transfer learning (TL), graph convolutional neural network (GCN), information theory, and TSK fuzzy neural network. TSKG extracts the relevant features in EEG through mutual information minimization and regards the fuzzy rules as graph structure data. The extracted relevant features and the original features are used to optimize the fuzzy rules through GCN. TSKG is tested on the fatigue driving dataset, and Root Mean Square Error (RMSE) and Correlation Coefficient (CC) are 0.1681 and 0.7118, respectively. Especially when dealing with difficult samples, TSKG has significantly better generalization performance.
Read full abstract