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

Journal: Sensors (Basel, Switzerland)
Published Date:

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.

Authors

  • Hong Zeng
    School of Computer Science and Technology, Hangzhou Dianzi University, China.
  • Jiaming Zhang
    School of Computer Science and Technology, Hangzhou Dianzi University, Hangzhou, Zhejiang 310018, China.
  • Wael Zakaria
    Department of Mathematics-Computer Science, Faculty of Science, Ain Shams University, Abbassia, Cairo 11435, Egypt.
  • Fabio Babiloni
    BrainSigns SRL, Rome, Italy.
  • Borghini Gianluca
    Industrial NeuroScience Lab, University of Rome "La Sapienza", 00161 Rome, Italy.
  • Xiufeng Li
    School of Computer Science and Technology, Hangzhou Dianzi University, China.
  • Wanzeng Kong
    School of Computer Science and Technology, Hangzhou Dianzi University, Hangzhou, Zhejiang 310018, China.