Convolutional Neural Network for Drowsiness Detection Using EEG Signals

Sensors (Basel). 2021 Mar 3;21(5):1734. doi: 10.3390/s21051734.

Abstract

Drowsiness detection (DD) has become a relevant area of active research in biomedical signal processing. Recently, various deep learning (DL) researches based on the EEG signals have been proposed to detect fatigue conditions. The research presented in this paper proposes an EEG classification system for DD based on DL networks. However, the proposed DD system is mainly realized into two procedures; (i) data acquisition and (ii) model analysis. For the data acquisition procedure, two key steps are considered, which are the signal collection using a wearable Emotiv EPOC+ headset to record 14 channels of EEG, and the signal annotation. Furthermore, a data augmentation (DA) step has been added to the proposed system to overcome the problem of over-fitting and to improve accuracy. As regards the model analysis, a comparative study is also introduced in this paper to argue the choice of DL architecture and frameworks used in our DD system. In this sense, The proposed DD protocol makes use of a convolutional neural network (CNN) architecture implemented using the Keras library. The results showed a high accuracy value (90.42%) in drowsy/awake discrimination and revealed the efficiency of the proposed DD system compared to other research works.

Keywords: EEG signals; Emotiv EPOC+; awake/drowsy states; classification; convolutional neural networks; data augmentation; deep learning; drowsiness detection.

MeSH terms

  • Electroencephalography*
  • Neural Networks, Computer
  • Signal Processing, Computer-Assisted
  • Wakefulness*