EEG quantization and entropy of multi-step transition probabilities for driver drowsiness detection via LSTM
Detecting driver drowsiness through electroencephalogram (EEG) poses challenges due to the complexity and variability of brain activity across different subjects. This study proposes a feature extraction pipeline combined with a Long Short-Term Memory (LSTM) network. EEG data from each electrode cha...
Saved in:
Published in | Computers in biology and medicine Vol. 196; no. Pt A; p. 110758 |
---|---|
Main Authors | , , |
Format | Journal Article |
Language | English |
Published |
United States
Elsevier Ltd
01.09.2025
|
Subjects | |
Online Access | Get full text |
Cover
Loading…
Summary: | Detecting driver drowsiness through electroencephalogram (EEG) poses challenges due to the complexity and variability of brain activity across different subjects. This study proposes a feature extraction pipeline combined with a Long Short-Term Memory (LSTM) network. EEG data from each electrode channel is normalized and quantized to discrete levels, where the probabilities of transitioning to other levels in the subsequent timesteps is modeled using multiple Hidden Markov Models (HMMs) for different timestep shifts. From the HMM emission probabilities, Shannon, Renyi, Tsallis, and Min entropy are extracted, forming a feature set that captures temporal channel information. These features are input into an LSTM network to classify alert or drowsy states. Monopolar and bipolar EEG montages are also investigated. Experiments on balanced and unbalanced EEG datasets show higher performance compared to existing machine learning and state-of-the-art deep learning methods. Subject-wise 5-fold and leave-one-out cross-validation achieved 91.23 % and 81.88 % accuracy on the balanced dataset, and 91.38 % and 80.58 % accuracy on the unbalanced dataset. Saliency analysis and ablation studies identifies key EEG channels, time-step shifts, and quantization levels contributing to drowsiness detection. The Oz channel produces the most discriminative features and offers state-of-the-art single-channel accuracy of 74.66 %. The feature extraction algorithm provides significant distinction between drowsy and alert classes, with transition Shannon entropy value being higher for the drowsy class. Reasons behind confidently correct and incorrect predictions and model criteria for detecting drowsiness are also explored. This work highlights the potential of using signal quantization with entropy of transition probabilities to extract meaningful features.
•A novel feature extraction method leveraging EEG quantization and multi-step entropy transition probabilities.•Comparative analysis of monopolar and bipolar EEG montages.•Comparative analysis of signal channel and multi-channel EEG signals.•Performance validation on balanced and unbalanced datasets with state-of-the-art accuracy metrics.•A comprehensive interpretability analysis using saliency maps.These highlights encapsulate the main contributions and findings of our research. |
---|---|
Bibliography: | ObjectType-Article-1 SourceType-Scholarly Journals-1 ObjectType-Feature-2 content type line 23 |
ISSN: | 0010-4825 1879-0534 1879-0534 |
DOI: | 10.1016/j.compbiomed.2025.110758 |