Singapore Institute of Technology
Browse

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.

History

Journal/Conference/Book title

IEEE Transactions on Cognitive and Developmental Systems

Publication date

2022-03-30

Usage metrics

    Categories

    No categories selected

    Exports

    RefWorks
    BibTeX
    Ref. manager
    Endnote
    DataCite
    NLM
    DC