EEG-based multi-level stress classification with and without smoothing filter

Recently, multi-level stress assessment has become an active research subject. In this context, researchers typically develop models based on machine learning classifiers and features extracted from biosignals like electrocardiogram (ECG) or electroencephalogram (EEG). For that purpose, EEG power spectral density (PSD) is a recurrent feature owing to its high responsiveness and remarkable performance. However, PSD is usually smoothed to cope with its bursty nature, what may cause data leakage and hence call into question classification performance. In this study, our aim was twofold: first, to examine the effect of EEG-PSD smoothing in three-level stress classification, and second, to evaluate the practical viability of a two-level stress detector without smoothing. To this end, we conducted participants through a stress-relax session while recording their EEG. Then, we estimated the EEG-PSD and used the stress reported by the participants as labels for classification. Initially, we developed a three-level stress classifier and examined the effect of smoothing on its performance. We found that classification performance was directly proportional to smoothing intensity (F1-score 0.61-0.94), and also that when smoothing was not applied to features, classification performance was insufficient for practical applicability (AUC < 0.7). We link this behavior to train-test contamination due to smoothing. Subsequently, we attempted two-level stress classification without smoothing. In this case, performance met the criteria for practical applicability (AUC = 0.76). This suggest that performance enhancement in three-level stress classification was caused by data leakage produced by smoothing, and hence, to render realistic stress classifiers each epoch should be processed individually.

Automated summary

The study examined the impact of EEG-PSD smoothing on three-level stress classification and found that smoothing can lead to data leakage and affect classification performance. Two-level stress classification without smoothing met the criteria for practical applicability, suggesting individual processing of each epoch is necessary for realistic stress classifiers.