Title Assessment of mental workload using a transformer network and two prefrontal EEG channels: an unparameterized approach /
Authors Beiramvand, Matin ; Shahbakhti, Mohammad ; Karttunen, Nina ; Koivula, Reijo ; Turunen, Jari ; Lipping, Tarmo
DOI 10.1109/TIM.2024.3395312
Full Text Download
Is Part of IEEE transactions on instrumentation and measurement.. Piscataway, NJ : IEEE. 2024, vol. 73, art. no. 4007010, p. 1-10.. ISSN 0018-9456. eISSN 1557-9662
Keywords [eng] energy ; entropy ; mental workload ; transformer network (TN) ; wearable EEG device
Abstract [eng] Despite promising results reported in the literature for mental workload assessment using electroencephalography (EEG), most of the proposed methods rely on employing multiple EEG channels, limiting their practicality. However, the advent of wearable EEG technology provides the possibility of mental workload assessment for real-life applications. Yet, a few studies that considered consumer-oriented EEG headsets for mental workload assessment only used a single database for validating the proposed methods, overlooking the potential for portability. In this research, we studied 60 recordings of participants playing a three-level n-back game, utilizing data from two EEG devices, Enobio and Muse, with distinctive characteristics such as sampling rate and channel configuration. Following the denoising of the EEG signals, we segmented the signals and applied the discrete wavelet transform (DWT) to decompose them into subbands. Then, we extracted Shannon entropy (SE) and wavelet log energy (WLE) features from all subbands. Subsequently, we fed the extracted features into five classifiers: support vector machine (SVM), k-nearest neighbors (kNNs), multilayer perceptron (MLP), AdaBoost, and the transformer network (TN). In comparing the results across all classifiers, the TN demonstrated superiority by achieving highest mean accuracy for DatabaseM (88%) and Database E (85%). Given the consistent outcomes achieved with the TN classifier across both databases and utilizing a three-level n-back game, our findings indicate that the proposed method holds promise for real-life applications.
Published Piscataway, NJ : IEEE
Type Journal article
Language English
Publication date 2024
CC license CC license description