InstanceEasyTL: An Improved Transfer-Learning Method for EEG-Based Cross-Subject Fatigue Detection

Sensors (Basel). 2020 Dec 17;20(24):7251. doi: 10.3390/s20247251.

Abstract

Electroencephalogram (EEG) is an effective indicator for the detection of driver fatigue. Due to the significant differences in EEG signals across subjects, and difficulty in collecting sufficient EEG samples for analysis during driving, detecting fatigue across subjects through using EEG signals remains a challenge. EasyTL is a kind of transfer-learning model, which has demonstrated better performance in the field of image recognition, but not yet been applied in cross-subject EEG-based applications. In this paper, we propose an improved EasyTL-based classifier, the InstanceEasyTL, to perform EEG-based analysis for cross-subject fatigue mental-state detection. Experimental results show that InstanceEasyTL not only requires less EEG data, but also obtains better performance in accuracy and robustness than EasyTL, as well as existing machine-learning models such as Support Vector Machine (SVM), Transfer Component Analysis (TCA), Geodesic Flow Kernel (GFK), and Domain-adversarial Neural Networks (DANN), etc.

Keywords: Electroencephalogram (EEG); InstanceEasyTL; cross-subject; fatigue driving; transfer learning.

MeSH terms

  • Automobile Driving*
  • Electroencephalography*
  • Fatigue / diagnosis*
  • Humans
  • Machine Learning*
  • Neural Networks, Computer
  • Support Vector Machine*