Benchmarking EEG-based Cross-dataset Driver Drowsiness Recognition with Deep Transfer Learning.

Abstract

It usually takes a long time to collect data for calibration when using electroencephalography (EEG) for driver drowsiness monitoring. Cross-dataset recognition is desirable since it can significantly save the calibration time when an existing dataset is used. However, the recognition accuracy is affected by the distribution drift problem caused by different experimental environments when building different datasets. In order to solve the problem, we propose a deep transfer learning model named Entropy-Driven Joint Adaptation Network (EDJAN), which can learn useful information from source and target domains simultaneously. An entropy-driven loss function is used to promote clustering of target-domain representations and an individual-level domain adaptation technique is proposed to alleviate the distribution discrepancy problem of test subjects. We use two public driving datasets SEEG-VIG and SADT to test the model on the cross-dataset setting. The proposed model achieved an accuracy of 83.3% when SADT is used as source domain and SEED-VIG is used as target domain and 76.7% accuracy on the reverse setting, which is higher than the other SOTA methods. The results are further analyzed with both global and local interpretation methods. Our work illuminates a promising direction of using EEG for calibration-free driver drowsiness recognition.