Enhancing Cognitive Workload Classification Using Integrated LSTM Layers and CNNs for fNIRS Data Analysis

Read original: arXiv:2407.15901 - Published 7/24/2024 by Mehshan Ahmed Khan, Houshyar Asadi, Mohammad Reza Chalak Qazani, Adetokunbo Arogbonlo, Siamak Pedrammehr, Adnan Anwar, Asim Bhatti, Saeid Nahavandi, Chee Peng Lim
Total Score

0

🏷️

Sign in to get full access

or

If you already have an account, we'll log you in

Overview

  • Functional near-infrared spectroscopy (fNIRS) is a non-invasive method to monitor brain activity by tracking changes in oxygenated and deoxygenated hemoglobin.
  • Machine learning techniques have been used to distinguish cognitive states from fNIRS data, but conventional methods have limitations in data preprocessing and only focus on differentiating between two levels of mental workload.
  • This paper explores the impact of integrating Long Short-Term Memory (LSTM) layers with Convolutional Neural Networks (CNNs) to address these issues and improve the accuracy of cognitive state classification.

Plain English Explanation

Functional near-infrared spectroscopy (fNIRS) is a technology that allows researchers to measure brain activity in a non-invasive way. It works by tracking changes in the levels of oxygen-rich and oxygen-poor blood in the brain, which are indicators of brain function.

Researchers have used various machine learning techniques to try to identify different cognitive states (mental activities) from fNIRS data. However, the traditional machine learning methods often require a lot of complex data preprocessing and are not very accurate at distinguishing between more than two levels of mental workload, such as just "low" and "high."

To address these limitations, this paper explores a new approach that combines two powerful machine learning techniques: Convolutional Neural Networks (CNNs) and Long Short-Term Memory (LSTMs). CNNs are good at extracting spatial features from data, while LSTMs can capture temporal patterns. By integrating these two methods, the researchers aimed to create a more effective model for classifying different cognitive states from fNIRS data.

The key idea is that the LSTM layers can help the model better understand the time-series nature of the fNIRS data, allowing it to make more accurate predictions about the user's mental state. This is an important advance, as previous studies have mainly focused on just distinguishing between "easy" and "difficult" tasks, rather than looking at more nuanced differences in cognitive load.

Technical Explanation

The researchers conducted experiments to assess the performance of Convolutional Neural Network (CNN) models with and without the addition of Long Short-Term Memory (LSTM) layers for classifying cognitive states from fNIRS data.

The CNN architecture was designed to capture spatial features from the fNIRS data, while the LSTM layers were incorporated to model the temporal dependencies in the signal. This integration was aimed at addressing the limitations of conventional CNN models, which can struggle to account for the time-series nature of physiological data like fNIRS.

The experimental results showed that the combined CNN-LSTM model achieved an accuracy of 97.92%, which represented a significant improvement over the standalone CNN model's accuracy of 97.40%. This suggests that the addition of LSTM layers enhanced the model's ability to learn meaningful representations from the fNIRS data, leading to more accurate classification of cognitive states.

The findings indicate that this hybrid deep learning approach is effective for cognitive load assessment using fNIRS, as it can capture both spatial and temporal features in the data. This is an important advancement over previous studies that have primarily focused on distinguishing between just two levels of mental workload.

Critical Analysis

The paper provides a compelling demonstration of how integrating LSTM layers with CNNs can improve the performance of cognitive state classification from fNIRS data. The authors acknowledge that most prior research in this area has been limited to differentiating between only two levels of cognitive load, such as "easy" and "difficult" tasks.

By exploring a more nuanced approach that can distinguish between multiple levels of mental workload, this research represents an important step forward in the field of neuroergonomics and cognitive assessment. The authors also note that the use of deep learning techniques, such as the CNN-LSTM model, can overcome the limitations of traditional machine learning methods that require complex data preprocessing.

However, the paper does not delve into potential limitations or caveats of the proposed approach. For example, it would be helpful to understand how the model might perform with smaller or more noisy datasets, or how it would generalize to different cognitive tasks or populations. Additionally, the authors do not discuss any potential ethical considerations or privacy concerns related to the use of fNIRS and deep learning for cognitive assessment.

Further research could explore the robustness and real-world applicability of this approach, as well as investigate any potential societal implications. Nevertheless, this paper represents a valuable contribution to the field and demonstrates the potential for advanced deep learning techniques to enhance our understanding of cognitive processes.

Conclusion

This paper presents a novel approach to cognitive state classification using functional near-infrared spectroscopy (fNIRS) data. By integrating Long Short-Term Memory (LSTM) layers with Convolutional Neural Networks (CNNs), the researchers developed a deep learning model that can effectively capture both spatial and temporal features in the fNIRS signals.

The experimental results show that this hybrid CNN-LSTM architecture outperforms standalone CNN models, achieving an accuracy of 97.92% in classifying cognitive states. This is a significant improvement over previous studies that have mainly focused on distinguishing between just two levels of mental workload.

The findings of this research represent an important advancement in the field of neuroergonomics and cognitive assessment, as they demonstrate the potential for deep learning techniques to provide a more comprehensive and nuanced understanding of cognitive processes. While further research is needed to address potential limitations and explore real-world applications, this paper lays the groundwork for exciting future developments in this area.



This summary was produced with help from an AI and may contain inaccuracies - check out the links to read the original source documents!

Follow @aimodelsfyi on 𝕏 →

Related Papers

🏷️

Total Score

0

Enhancing Cognitive Workload Classification Using Integrated LSTM Layers and CNNs for fNIRS Data Analysis

Mehshan Ahmed Khan, Houshyar Asadi, Mohammad Reza Chalak Qazani, Adetokunbo Arogbonlo, Siamak Pedrammehr, Adnan Anwar, Asim Bhatti, Saeid Nahavandi, Chee Peng Lim

Functional near-infrared spectroscopy (fNIRS) is employed as a non-invasive method to monitor functional brain activation by capturing changes in the concentrations of oxygenated haemoglobin (HbO) and deoxygenated haemo-globin (HbR). Various machine learning classification techniques have been utilized to distinguish cognitive states. However, conventional machine learning methods, although simpler to implement, undergo a complex pre-processing phase before network training and demonstrate reduced accuracy due to inadequate data preprocessing. Additionally, previous research in cog-nitive load assessment using fNIRS has predominantly focused on differ-sizeentiating between two levels of mental workload. These studies mainly aim to classify low and high levels of cognitive load or distinguish between easy and difficult tasks. To address these limitations associated with conven-tional methods, this paper conducts a comprehensive exploration of the im-pact of Long Short-Term Memory (LSTM) layers on the effectiveness of Convolutional Neural Networks (CNNs) within deep learning models. This is to address the issues related to spatial features overfitting and lack of tem-poral dependencies in CNN in the previous studies. By integrating LSTM layers, the model can capture temporal dependencies in the fNIRS data, al-lowing for a more comprehensive understanding of cognitive states. The primary objective is to assess how incorporating LSTM layers enhances the performance of CNNs. The experimental results presented in this paper demonstrate that the integration of LSTM layers with Convolutional layers results in an increase in the accuracy of deep learning models from 97.40% to 97.92%.

Read more

7/24/2024

🏷️

Total Score

0

Functional near-infrared spectroscopy (fNIRS) and Eye tracking for Cognitive Load classification in a Driving Simulator Using Deep Learning

Mehshan Ahmed Khan, Houshyar Asadi, Mohammad Reza Chalak Qazani, Chee Peng Lim, Saied Nahavandi

Motion simulators allow researchers to safely investigate the interaction of drivers with a vehicle. However, many studies that use driving simulator data to predict cognitive load only employ two levels of workload, leaving a gap in research on employing deep learning methodologies to analyze cognitive load, especially in challenging low-light conditions. Often, studies overlook or solely focus on scenarios in bright daylight. To address this gap and understand the correlation between performance and cognitive load, this study employs functional near-infrared spectroscopy (fNIRS) and eye-tracking data, including fixation duration and gaze direction, during simulated driving tasks in low visibility conditions, inducing various mental workloads. The first stage involves the statistical estimation of useful features from fNIRS and eye-tracking data. ANOVA will be applied to the signals to identify significant channels from fNIRS signals. Optimal features from fNIRS, eye-tracking and vehicle dynamics are then combined in one chunk as input to the CNN and LSTM model to predict workload variations. The proposed CNN-LSTM model achieved 99% accuracy with neurological data and 89% with vehicle dynamics to predict cognitive load, indicating potential for real-time assessment of driver mental state and guide designers for the development of safe adaptive systems.

Read more

8/14/2024

🚀

Total Score

0

Cognitive Networks and Performance Drive fMRI-Based State Classification Using DNN Models

Murat Kucukosmanoglu, Javier O. Garcia, Justin Brooks, Kanika Bansal

Deep neural network (DNN) models have demonstrated impressive performance in various domains, yet their application in cognitive neuroscience is limited due to their lack of interpretability. In this study we employ two structurally different and complementary DNN-based models, a one-dimensional convolutional neural network (1D-CNN) and a bidirectional long short-term memory network (BiLSTM), to classify individual cognitive states from fMRI BOLD data, with a focus on understanding the cognitive underpinnings of the classification decisions. We show that despite the architectural differences, both models consistently produce a robust relationship between prediction accuracy and individual cognitive performance, such that low performance leads to poor prediction accuracy. To achieve model explainability, we used permutation techniques to calculate feature importance, allowing us to identify the most critical brain regions influencing model predictions. Across models, we found the dominance of visual networks, suggesting that task-driven state differences are primarily encoded in visual processing. Attention and control networks also showed relatively high importance, however, default mode and temporal-parietal networks demonstrated negligible contribution in differentiating cognitive states. Additionally, we observed individual trait-based effects and subtle model-specific differences, such that 1D-CNN showed slightly better overall performance, while BiLSTM showed better sensitivity for individual behavior; these initial findings require further research and robustness testing to be fully established. Our work underscores the importance of explainable DNN models in uncovering the neural mechanisms underlying cognitive state transitions, providing a foundation for future work in this domain.

Read more

9/4/2024

📈

Total Score

0

Predicting cognitive load in immersive driving scenarios with a hybrid CNN-RNN model

Mehshan Ahmed Khan, Houshyar Asadi, Mohammad Reza Chalak Qazani, Adetokunbo Arogbonlo, Saeid Nahavandi, Chee Peng Lim

One debatable issue in traffic safety research is that cognitive load from sec-ondary tasks reduces primary task performance, such as driving. Although physiological signals have been extensively used in driving-related research to assess cognitive load, only a few studies have specifically focused on high cognitive load scenarios. Most existing studies tend to examine moderate or low levels of cognitive load In this study, we adopted an auditory version of the n-back task of three levels as a cognitively loading secondary task while driving in a driving simulator. During the simultaneous execution of driving and the n-back task, we recorded fNIRS, eye-tracking, and driving behavior data to predict cognitive load at three different levels. To the best of our knowledge, this combination of data sources has never been used before. Un-like most previous studies that utilize binary classification of cognitive load and driving in conditions without traffic, our study involved three levels of cognitive load, with drivers operating in normal traffic conditions under low visibility, specifically during nighttime and rainy weather. We proposed a hybrid neural network combining a 1D Convolutional Neural Network and a Recurrent Neural Network to predict cognitive load. Our experimental re-sults demonstrate that the proposed model, with fewer parameters, increases accuracy from 99.82% to 99.99% using physiological data, and from 87.26% to 92.02% using driving behavior data alone. This significant improvement highlights the effectiveness of our hybrid neural network in accurately pre-dicting cognitive load during driving under challenging conditions.

Read more

8/14/2024