Comprehensive Analysis on Feature Selection, Machine Learning and Deep Learning Algorithms to Detect Driver Drowsiness – An EEG Study
Document Type
Conference Proceeding
Publication Date
11-9-2023
Abstract
Drowsiness among drivers is one of the key reasons for traffic accidents which has been recognized as a global problem and this hazardous event causes physical injury, economic suffering, and even death to people. With the advancement of artificial intelligence, drowsiness can be detected from the brain signals and a prior alert can be generated to avoid such losses. For an optimal classification study, extensive analysis of major aspects such as optimal feature selection and choosing compatible classification algorithms should be done properly. This study focused on the classification of the drowsiness stage and active stage from continuous EEG signal (9 channels) starting with the implementation of feature selection algorithms to find the best temporal EEG features for feeding the classifiers and then undergoing different machine learning and deep learning algorithms. Recursive feature elimination and chi-square test, two popular feature selection algorithms, were utilized to find the most possible contributing features which were then fed to 17 machine learning models and 3 deep learning networks. After primary performance analysis selected machine learning models: K-Nearest Neighbor (KNN), Logistic Regression, Random Forest and Support Vector Machine (SVM) were chosen for hyperparametric tuning to achieve the highest performance. Among three deep learning algorithms such as long-short-term memory (LSTM), artificial neural network (ANN), and 1-dimensional convolutional network (1D-CNN), LSTM outperformed the rests with a classification accuracy of 87%. KNN and Random Forest yielded 90% accuracy with lower resource consumption than the deep learning models. The goal of this study is to help in the improvement of a drowsiness classification system that is effective in preventing accidents and reducing the risk of harm to people and property.
Identifier
85187723266 (Scopus)
ISBN
[9798400708343]
Publication Title
ACM International Conference Proceeding Series
External Full Text Location
https://doi.org/10.1145/3637732.3637791
First Page
147
Last Page
153
Recommended Citation
Ferdous, Talukdar Raian; Tushar, Md Abdul Kader; Rafi, Rifath Hasan; Asif, Nazmus Salehin; Palash, Torikul Islam; Arman, Md Adil; Saha, Debasish Kumar; and Islam, Muhammad Muinul, "Comprehensive Analysis on Feature Selection, Machine Learning and Deep Learning Algorithms to Detect Driver Drowsiness – An EEG Study" (2023). Faculty Publications. 1330.
https://digitalcommons.njit.edu/fac_pubs/1330
