A new feature selection approach for driving fatigue EEG detection with a modified machine learning algorithm

Yun Zheng, Yuliang Ma, Jared Cammon, Songjie Zhang, Jianhai Zhang, Yingchun Zhang

Research output: Contribution to journalArticlepeer-review

2 Scopus citations


This study aims to identify new electroencephalography (EEG) features for the detection of driving fatigue. The most common EEG feature in driving fatigue detection is the power spectral density (PSD) of five frequency bands, i.e., alpha, beta, gamma, delta, and theta bands. PSD has proved to be useful, however its flaw is that it covers much implicit information of the time domain. In this study we propose a new approach, which combines ensemble empirical mode decomposition (EEMD) and PSD, to explore new EEG features for driving fatigue detection. Through EEMD we get a series of intrinsic mode function (IMF) components, from which we can extract PSD features. We used six features to compare with the proposed features, including the PSD of five frequency bands, PSD of empirical mode decomposition (EMD)-IMF components, PSD, permutation entropy (PE), sample entropy (SE), and fuzzy entropy (FE) of EEMD-IMF components, and common spatial pattern. Feature overlap ratio and multiple machine learning methods were applied to evaluate these feature extraction approaches. The results show that the classification accuracy and overlap ratio of experiments based on IMF's energy spectrum is far superior to other features. Through channel optimization and a comparison of accuracy, we conclude that our new feature selection approach has a better performance based on the modified hierarchical extreme learning machine algorithm with Particle Swarm Optimization (PSO–H-ELM) classifier, which has the highest average accuracy of 97.53%.

Original languageEnglish (US)
Article number105718
JournalComputers in Biology and Medicine
StatePublished - Aug 2022


  • Driving fatigue
  • Electroencephalography
  • Ensemble empirical mode decomposition (EEMD)
  • Extreme learning machines
  • Particle swarm optimization
  • Power spectral density (PSD)

ASJC Scopus subject areas

  • Computer Science Applications
  • Health Informatics


Dive into the research topics of 'A new feature selection approach for driving fatigue EEG detection with a modified machine learning algorithm'. Together they form a unique fingerprint.

Cite this