We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:

Download:

Current browse context:

eess

Change to browse by:

References & Citations

Bookmark

(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Electrical Engineering and Systems Science > Signal Processing

Title: EEG-based Cross-Subject Driver Drowsiness Recognition with an Interpretable Convolutional Neural Network

Abstract: In the context of electroencephalogram (EEG)-based driver drowsiness recognition, it is still challenging to design a calibration-free system, since EEG signals vary significantly among different subjects and recording sessions. Many efforts have been made to use deep learning methods for mental state recognition from EEG signals. However, existing work mostly treats deep learning models as black-box classifiers, while what have been learned by the models and to which extent they are affected by the noise in EEG data are still underexplored. In this paper, we develop a novel convolutional neural network combined with an interpretation technique that allows sample-wise analysis of important features for classification. The network has a compact structure and takes advantage of separable convolutions to process the EEG signals in a spatial-temporal sequence. Results show that the model achieves an average accuracy of 78.35% on 11 subjects for leave-one-out cross-subject drowsiness recognition, which is higher than the conventional baseline methods of 53.40%-72.68% and state-of-the-art deep learning methods of 71.75%-75.19%. Interpretation results indicate the model has learned to recognize biologically meaningful features from EEG signals, e.g., Alpha spindles, as strong indicators of drowsiness across different subjects. In addition, we also explore reasons behind some wrongly classified samples with the interpretation technique and discuss potential ways to improve the recognition accuracy. Our work illustrates a promising direction on using interpretable deep learning models to discover meaningful patterns related to different mental states from complex EEG signals.
Subjects: Signal Processing (eess.SP); Machine Learning (cs.LG); Neural and Evolutionary Computing (cs.NE); Neurons and Cognition (q-bio.NC)
Journal reference: IEEE Transactions on Neural Networks and Learning Systems, 2022
DOI: 10.1109/TNNLS.2022.3147208
Cite as: arXiv:2107.09507 [eess.SP]
  (or arXiv:2107.09507v4 [eess.SP] for this version)

Submission history

From: Jian Cui [view email]
[v1] Sun, 30 May 2021 14:47:20 GMT (5532kb)
[v2] Fri, 5 Nov 2021 01:46:13 GMT (1959kb)
[v3] Wed, 24 Nov 2021 13:51:52 GMT (1962kb)
[v4] Fri, 18 Feb 2022 02:27:49 GMT (1963kb)

Link back to: arXiv, form interface, contact.