Abstract:The driver state monitoring is becoming one of the research hotspots in the field of traffic and vehicle safety, which can ensure driving safety by monitoring the driver’s state. Therefore, this work makes an attempt to recognize driver's mental load and emotional states. However, the reliability and accuracy of driver status detection largely depend on the extracted features and the detection algorithm. The existing methods mainly improve accuracy by increasing the number of features, but for the problem with limited training samples, the increase in features may lead to an increase in model variance, and resulting in overfitting. In addition, single-model based methods may fall into local optimal solutions when performing local search. To alleviate these, we firstly propose an RFE-SHAP algorithm that improves the recursive feature elimination algorithm by using the Shapley Additive exPlanning value, and paired it with the eXtreme Gradient Boosting (XGBoost) algorithm to screen out the subset of features that best represent the driver's mental load and emotional state. Secondly, we proposed a multilayer stacking ensemble learning method based on different optimal feature subsets to further improve the accuracy of the driver’s mental load and emotional state recognition. For each base models used for integration learning, the corresponding optimal feature subsets are selected using the RFE-SHAP algorithm, and then the different optimal feature subsets are combined with multilayer stacking ensemble learning. The validation via experimental data demonstrates that the XGBoost-RFE-SHAP algorithm achieves superior model performance with fewer feature combinations than the classic algorithm. The proposed driver state detection model based on multiple optimum feature subsets and multilayer stacking integration methods would more accurately identify the driver's mental load and emotional state than the single model, commonly used machine learning and integrated learning algorithms.

Estimating the Vigilance of High-Speed Rail Drivers Using a Stacking Ensemble Learning Method

Driver vigilance detection for high-speed rail using fusion of multiple physiological signals and deep learning

Detection of Train Driver Fatigue and Distraction Based on Forehead EEG: A Time-Series Ensemble Learning Method

A Regression Method With Subnetwork Neurons for Vigilance Estimation Using EOG and EEG

Faster Single Model Vigilance Detection Based on Deep Learning

Dynamically Weighted Ensemble-based Prediction System for Adaptively Modeling Driver Reaction Time

An aggressive driving state recognition model using EEG based on stacking ensemble learning

Driver vigilance detection based on limited EEG signals

Ensemble Classifier for Driver's Fatigue Detection Based on a Single EEG Channel

Assessment of Vigilance Level during Work: Fitting a Hidden Markov Model to Heart Rate Variability

Estimation of Varying Reaction Times with RNN and Application to Human-like Autonomous Car-following Modeling

Regression Analysis of EEG Signals in Fatigue Driving Based on Ensemble Learning

Research on Brain Load Prediction Based on Machine Learning for High-Speed Railway Dispatching

Residual Attention Capsule Network for Multimodal EEG- and EOG-Based Driver Vigilance Estimation

Estimate vigilance level in driving simulation based on sparse representation

A Multilayer Stacking Method Base on RFE-SHAP Feature Selection Strategy for Recognition of Driver's Mental Load and Emotional State.

An Integrated Gaussian Mixture Model to Estimate Vigilance Level Based on EEG Recordings

Eeg-Based Vigilance Estimation Using Extreme Learning Machines

EEG-Fest: Few-shot based Attention Network for Driver's Vigilance Estimation with EEG Signals

Optimizing Driver Vigilance Recognition: Examining the Characterization and Cumulative Effect of Physiological Signals Across Manual and Automated Driving and Durations

Capsule Attention for Multimodal EEG-EOG Representation Learning With Application to Driver Vigilance Estimation