Singapore Institute of Technology

File(s) stored somewhere else

Please note: Linked content is NOT stored on Singapore Institute of Technology and we can't guarantee its availability, quality, security or accept any liability.

Cognitive Workload Estimation Using Variational Auto Encoder & Attention-based Deep Model

journal contribution
posted on 2023-09-26, 04:24 authored by Debashis Das Chakladar, Sumalyo Datta, Partha Pratim Roy, Vinod Achutavarrier PrasadVinod Achutavarrier Prasad

The estimation of cognitive workload using electroencephalogram (EEG) is an emerging research area. However, due to poor spatial resolution issues, features obtained from EEG signals often lead to poor classification results. As a good generative model, the variational autoencoder (VAE) extracts the noise-free robust features from the latent space that lead to better classification performance. The spatial attention-based method [convolutional block attention module (CBAM)] can improve the spatial resolution of EEG signals. In this article, we propose an effective VAE-CBAM-based deep model for estimating cognitive states from topographical videos. Topographical videos of four different conditions [baseline (BL), low workload (LW), medium workload (MW), and high workload (HW)] of the mental arithmetic task are taken for the experiment. Initially, the VAE extracts localized features from input images (extracted from topographical video), and CBAM infers the spatial–channel-level’s attention features from those localized features. Finally, the deep CNN-BLSTM model effectively learns those attention-based spatial features in a timely distributed manner to classify the cognitive state. For four-class and two-class classifications, the proposed model achieves 83.13% and 92.09% classification accuracy, respectively. The proposed model enhances the future research scope of attention-based studies in EEG applications.


Journal/Conference/Book title

IEEE Transactions on Cognitive and Developmental Systems

Publication date


Usage metrics


    No categories selected


    Ref. manager