-wave-based exploration of sensitive EEG features and classification of situation awareness

Abstract

The purpose of this study was to explore the electroencephalogram (EEG) features sensitive to situation awareness (SA) and then classify SA levels. Forty-eight participants were recruited to complete an SA standard test based on the multi-attribute task battery (MATB) II, and the corresponding EEG data and situation awareness global assessment technology (SAGAT) scores were recorded. The population with the top 25% of SAGAT scores was selected as the high-SA level (HSL) group, and the bottom 25% was the low-SA level (LSL) group. The results showed that (1) for the relative power of $\beta$ 1 (16–20Hz), $\beta$ 2 (20–24Hz) and $\beta$ 3 (24–30Hz), repeated measures analysis of variance (ANOVA) in three brain regions (Central Central-Parietal, and Parietal) × three brain lateralities (left, midline, and right) × two SA groups (HSL and LSL) showed a significant main effect for SA groups; post hoc comparisons revealed that compared with LSL, the above features of HSL were higher. (2) for most ratio features associated with $\beta$ 1 ∼ $\beta$ 3, ANOVA also revealed a main effect for SA groups. (3) EEG features sensitive to SA were selected to classify SA levels with small-sample data based on the general supervised machine learning classifiers. Five-fold cross-validation results showed that among the models with easy interpretability, logistic regression (LR) and decision tree (DT) presented the highest accuracy (both 92%), while among the models with hard interpretability, the accuracy of random forest (RF) was 88.8%, followed by an artificial neural network (ANN) of 84%. The above results suggested that (1) the relative power of $\beta$ 1 ∼ $\beta$ 3 and their associated ratios were sensitive to changes in SA levels; (2) the general supervised machine learning models all exhibited good accuracy (greater than 75%); and (3) furthermore, LR and DT are recommended by combining the interpretability and accuracy of the models.