An efficient K-NN approach for automatic drowsiness detection using single-channel EEG recording

Drowsy driving is a major cause of many traffic accidents. The aim of this work is to develop an automatic drowsiness detection system using an efficient k-nearest neighbors (K-NN) algorithm. First, the distribution of power in time-frequency space was obtained using short-time Fourier transform (STFT) and then, the mean value of power during time-segments of 0.5 second was calculated for each EEG subband. In addition, standard deviation (SD) and Shanon entropy related to each time-segment were computed from time-domain. Finally, 52 features were extracted. Random forest algorithm was applied over the extracted data, aiming to choose the most informative subset of features. A total of 11 features were selected in order to classify drowsiness and alertness. Kd-trees was used as the nearest neighbors search algorithm so as to have a fast classifier. Our experimental results show that drowsiness can be classified efficiently with 91% accuracy using the methods and materials proposed in this paper.

[1]  Chin-Teng Lin,et al.  Single channel wireless EEG device for real-time fatigue level detection , 2015, 2015 International Joint Conference on Neural Networks (IJCNN).

[2]  Bernhard Schölkopf,et al.  Improving the Accuracy and Speed of Support Vector Machines , 1996, NIPS.

[3]  Luc Devroye,et al.  On the Inequality of Cover and Hart in Nearest Neighbor Discrimination , 1981, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[4]  R. Suganya,et al.  Data Mining Concepts and Techniques , 2010 .

[5]  Gwen Littlewort,et al.  Drowsy Driver Detection Through Facial Movement Analysis , 2007, ICCV-HCI.

[6]  Shigeo Abe DrEng.  Pattern Classification , 2001, Springer London.

[7]  Vandana,et al.  Survey of Nearest Neighbor Techniques , 2010, ArXiv.

[8]  Piotr Indyk,et al.  Approximate nearest neighbors: towards removing the curse of dimensionality , 1998, STOC '98.

[9]  Harris Drucker,et al.  Comparison of learning algorithms for handwritten digit recognition , 1995 .

[10]  Jon Louis Bentley,et al.  An Algorithm for Finding Best Matches in Logarithmic Expected Time , 1977, TOMS.

[11]  Miad Faezipour,et al.  Efficient sleep stage classification based on EEG signals , 2014, IEEE Long Island Systems, Applications and Technology (LISAT) Conference 2014.

[12]  Mubarak Shah,et al.  Determining driver visual attention with one camera , 2003, IEEE Trans. Intell. Transp. Syst..

[13]  Franco P. Preparata,et al.  Computational Geometry , 1985, Texts and Monographs in Computer Science.

[14]  Cataldo Guaragnella,et al.  A visual approach for driver inattention detection , 2007, Pattern Recognit..

[15]  Adel Al-Jumaily,et al.  Fuzzy wavelet packet based feature extraction method for multifunction myoelectric control , 2007 .

[16]  Eric Laciar,et al.  Automatic detection of drowsiness in EEG records based on multimodal analysis. , 2014, Medical engineering & physics.

[17]  Robert F. Sproull,et al.  Refinements to nearest-neighbor searching ink-dimensional trees , 1991, Algorithmica.

[18]  Mercedes Atienza,et al.  Brain Spatial Microstates of Human Spontaneous Alpha Activity in Relaxed Wakefulness, Drowsiness Period, and REM Sleep , 2004, Brain Topography.

[19]  Leo Breiman,et al.  Random Forests , 2001, Machine Learning.

[20]  Mervyn V. M. Yeo,et al.  Can SVM be used for automatic EEG detection of drowsiness during car driving , 2009 .

[21]  Sunil Arya,et al.  An optimal algorithm for approximate nearest neighbor searching fixed dimensions , 1998, JACM.

[22]  J. L. Cantero,et al.  Human alpha oscillations in wakefulness, drowsiness period, and REM sleep: different electroencephalographic phenomena within the alpha band , 2002, Neurophysiologie Clinique/Clinical Neurophysiology.

[23]  Paul Stephen Rau,et al.  Drowsy Driver Detection and Warning System for Commercial Vehicle Drivers: Field Operational Test Design, Data Analyses, and Progress , 2005 .

[24]  Sun Yanfeng,et al.  Yawning Detection Based on Gabor Wavelets and LDA , 2009 .

[25]  Aeilko H. Zwinderman,et al.  Analysis of a sleep-dependent neuronal feedback loop: the slow-wave microcontinuity of the EEG , 2000, IEEE Transactions on Biomedical Engineering.

[26]  Chin-Teng Lin,et al.  Generalized EEG-Based Drowsiness Prediction System by Using a Self-Organizing Neural Fuzzy System , 2012, IEEE Transactions on Circuits and Systems I: Regular Papers.

[27]  Lingyu Yu,et al.  Title: Comparison of Short-time Fourier Transform and Wavelet Transform of Transient and Tone Burst Wave Propagation Signals For Structural Health Monitoring. , 2003 .

[28]  D. Sommer,et al.  Evaluation of fatigue monitoring technologies , 2010 .

[29]  Steven Skiena,et al.  The Algorithm Design Manual , 2020, Texts in Computer Science.

[30]  Cuntai Guan,et al.  An analysis on driver drowsiness based on reaction time and EEG band power , 2015, 2015 37th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[31]  Andrew W. Moore,et al.  An Investigation of Practical Approximate Nearest Neighbor Algorithms , 2004, NIPS.