Multidomain EEG Representation Learning With Orthogonal Mapping and Attention-Based Fusion for Cognitive Load Classification

Abstract

We propose a new representation learning solution for the classification of cognitive load based on electroencephalogram (EEG). Our method integrates both time and frequency domains by first passing the raw EEG signals through the convolutional encoder to obtain the time domain representations. Next, we measure the power spectral density for all five EEG frequency bands and generate the channel power values as 2-D images referred to as multispectral topography maps. These multispectral topography maps are then fed to a separate encoder to obtain the representations in frequency domain. Our solution employs a multidomain attention module that maps these domain-specific embeddings onto a shared embedding space to emphasize more on important interdomain relationships to enhance the representations for cognitive load classification. In addition, we incorporate an orthogonal projection constraint during the training of our method to effectively increase the interclass distances while improving intraclass clustering. This enhancement allows efficient discrimination between different cognitive states and aids in better grouping of similar states within the feature space. We validate the effectiveness of our model through extensive experiments on two public EEG datasets, CL-Drive and CLARE for cognitive load classification. Our results demonstrate the superiority of our multidomain approach over the traditional single-domain techniques. Moreover, we conduct ablation and sensitivity analyses to assess the impact of various components of our method. Finally, robustness experiments on different amounts of added noise demonstrate the stability of our method compared to other state-of-the-art solutions.