VIGNet: A Python repository from ksoh97

VIGNet: A Deep Convolutional Neural Network for EEG-based Driver Vigilance Estimation

This repository provides a TensorFlow implementation of the following paper:

VIGNet: A Deep Convolutional Neural Network for EEG-based Driver Vigilance Estimation
Wonjun Ko¹, Kwanseok Oh², Eunjin Jeon¹, and Heung-Il Suk^{1, 2}
(¹Department of Brain and Cognitive Engineering, Korea University)
(²Department of Artificial Intelligence, Korea University)
Official Version: https://ieeexplore.ieee.org/abstract/document/9061668
Presented in the 8th IEEE International Winter Conference on Brain-Computer Interface (BCI)

Abstract: Estimating driver fatigue is an important issue for traffic safety and user-centered brain–computer interface. In this paper, based on differential entropy (DE) extracted from electroencephalography (EEG) signals, we develop a novel deep convolutional neural network to detect driver drowsiness. By exploiting DE of EEG samples, the proposed network effectively extracts class-discriminative deep and hierarchical features. Then, a densely-connected layer is used for the final decision making to identify driver condition. To demonstrate the validity of our proposed method, we conduct classification and regression experiments using publicly available SEED-VIG dataset. Further, we also compare the proposed network to other competitive state-of-the-art methods with an appropriate statistical analysis. Furthermore, we inspect the real-world usability of our method by visualizing a change in the probability of driver status and confusion matrices.

Dependencies

Downloading datasets

To download SEED-VIG dataset

https://bcmi.sjtu.edu.cn/~seed/seed-vig.html

Usage

network.py contains the proposed deep learning architectures, utils.py contains functions used for experimental procedures, and experiment.py contains the main experimental functions.

Citation

If you find this work useful for your research, please cite our paper:

@inproceedings{ko2020vignet,
  title={Vignet: A deep convolutional neural network for eeg-based driver vigilance estimation},
  author={Ko, Wonjun and Oh, Kwanseok and Jeon, Eunjin and Suk, Heung-Il},
  booktitle={2020 8th International Winter Conference on Brain-Computer Interface (BCI)},
  pages={1--3},
  year={2020},
  organization={IEEE}
}

Acknowledgements