Understanding Cognitive Fatigue from fMRI Scans with Self-supervised Learning
This work addresses cognitive fatigue prediction for traumatic brain injury patients and healthy individuals, representing an incremental advance in neuroimaging analysis.
The paper tackled predicting cognitive fatigue levels from fMRI scans by proposing a spatio-temporal CNN-LSTM model pre-trained with MoCo on BOLD5000 and fine-tuned on a novel dataset of TBI patients and healthy controls, achieving state-of-the-art results.
Functional magnetic resonance imaging (fMRI) is a neuroimaging technique that records neural activations in the brain by capturing the blood oxygen level in different regions based on the task performed by a subject. Given fMRI data, the problem of predicting the state of cognitive fatigue in a person has not been investigated to its full extent. This paper proposes tackling this issue as a multi-class classification problem by dividing the state of cognitive fatigue into six different levels, ranging from no-fatigue to extreme fatigue conditions. We built a spatio-temporal model that uses convolutional neural networks (CNN) for spatial feature extraction and a long short-term memory (LSTM) network for temporal modeling of 4D fMRI scans. We also applied a self-supervised method called MoCo (Momentum Contrast) to pre-train our model on a public dataset BOLD5000 and fine-tuned it on our labeled dataset to predict cognitive fatigue. Our novel dataset contains fMRI scans from Traumatic Brain Injury (TBI) patients and healthy controls (HCs) while performing a series of N-back cognitive tasks. This method establishes a state-of-the-art technique to analyze cognitive fatigue from fMRI data and beats previous approaches to solve this problem.