EEG-based Prediction of Driver's Cognitive Performance by Deep Convolutional Neural Network

Mehdi Hajinoroozi,Zijing Mao,Tzyy-Ping Jung,Chin-Teng Lin,Yufei Huang
DOI: https://doi.org/10.1016/j.image.2016.05.018
IF: 3.453
2016-01-01
Signal Processing Image Communication
Abstract:We considered the prediction of driver's cognitive states related to driving performance using EEG signals. We proposed a novel channel-wise convolutional neural network (CCNN) whose architecture considers the unique characteristics of EEG data. We also discussed CCNN-R, a CCNN variation that uses Restricted Boltzmann Machine to replace the convolutional filter, and derived the detailed algorithm. To test the performance of CCNN and CCNN-R, we assembled a large EEG dataset from 3 studies of driver fatigue that includes samples from 37 subjects. Using this dataset, we investigated the new CCNN and CCNN-R on raw EEG data and also Independent Component Analysis (ICA) decomposition. We tested both within-subject and cross-subject predictions and the results showed CCNN and CCNN-R achieved robust and improved performance over conventional DNN and CNN as well as other non-DL algorithms.
What problem does this paper attempt to address?