Abstract:Cognitive workload is a crucial factor in tasks involving dynamic decision-making and other real-time and high-risk situations. Neuroimaging techniques have long been used for estimating cognitive workload. Given the portability, cost-effectiveness and high time-resolution of EEG as compared to fMRI and other neuroimaging modalities, an efficient method of estimating an individual’s workload using EEG is of paramount importance. Multiple cognitive, psychiatric and behavioral phenotypes have already been known to be linked with “functional connectivity”, i.e., correlations between different brain regions. In this work, we explored the possibility of using different model-free functional connectivity metrics along with deep learning in order to efficiently classify the cognitive workload of the participants. To this end, 64-channel EEG data of 19 participants were collected while they were doing the traditional n-back task. These data (after pre-processing) were used to extract the functional connectivity features, namely Phase Transfer Entropy (PTE), Mutual Information (MI) and Phase Locking Value (PLV). These three were chosen to do a comprehensive comparison of directed and non-directed model-free functional connectivity metrics (allows faster computations). Using these features, three deep learning classifiers, namely CNN, LSTM and Conv-LSTM were used for classifying the cognitive workload as low (1-back), medium (2-back) or high (3-back). With the high inter-subject variability in EEG and cognitive workload and recent research highlighting that EEG-based functional connectivity metrics are subject-specific, subject-specific classifiers were used. Results show the state-of-the-art multi-class classification accuracy with the combination of MI with CNN at 80.87%, followed by the combination of PLV with CNN (at 75.88%) and MI with LSTM (at 71.87%). The highest subject specific performance was achieved by the combinations of PLV with Conv-LSTM, and PLV with CNN with an accuracy of 97.92%, followed by the combination of MI with CNN (at 95.83%) and MI with Conv-LSTM (at 93.75%). The results highlight the efficacy of the combination of EEG-based model-free functional connectivity metrics and deep learning in order to classify cognitive workload. The work can further be extended to explore the possibility of classifying cognitive workload in real-time, dynamic and complex real-world scenarios.

EEG Based Dynamic Functional Connectivity Analysis in Mental Workload Tasks With Different Types of Information

Functional Brain Network Analysis Reveals Time-on-task Related Performance Decline

Dynamic functional connectivity correlates of mental workload

Evaluation of Mental Workload in Working Memory Tasks with Different Information Types Based on EEG.

Topological EEG-Based Functional Connectivity Analysis for Mental Workload State Recognition

Optimized electroencephalogram and functional near-infrared spectroscopy-based mental workload detection method for practical applications

Cross-Task Mental Workload Recognition Based on EEG Tensor Representation and Transfer Learning

Assessment of Mental Workload Based on Multi-Physiological Signals

Evidence for modulation of EEG microstates by mental workload levels and task types

Cognitive Load During Multitasking Can Be Accurately Assessed Based on Single Channel Electroencephalography Using Graph Methods

EEG-based Mental Workload Recognition Related to Multitasking

EEG-TNet: an End-To-End Brain Computer Interface Framework for Mental Workload Estimation

EEG/fNIRS Based Workload Classification Using Functional Brain Connectivity and Machine Learning

Mental workload estimation with electroencephalogram signals by combining multi-space deep models

Physiological Measures of Mental Workload: Evidence from Empirical Studies

Subject-Specific Cognitive Workload Classification Using EEG-Based Functional Connectivity and Deep Learning

Discriminative Analysis of Brain Functional Connectivity Patterns for Mental Fatigue Classification

Mental Workload Drives Different Reorganizations of Functional Cortical Connectivity Between 2D and 3D Simulated Flight Experiments

An EEG-based Mental Workload Estimator Trained on Working Memory Task Can Work Well under Simulated Multi-Attribute Task.

Recognition and Evaluation of Mental Workload in Different Stages of Perceptual and Cognitive Information Processing Using a Multimodal Approach.