Causal fMRI-Mamba: Causal State Space Model for Neural Decoding and Brain Task States Recognition

Abstract

Deep learning advances neural decoding in functional magnetic resonance imaging (fMRI) tasks with convolution and attention-based methods. However, these methods struggle with capturing global spatiotemporal information due to high dimensionality, noise and inter-individual difference of fMRI, which also increase computational complexity and prior bias. To this end, a novel causal state space model, Causal fMRI-Mamba, is proposed for neural decoding and task state mapping. It effectively captures global spatiotemporal information via eliminating local redundancies and capturing long-distance dependencies. Meanwhile, a causal representation framework is designed to extract invariant high-order features and disentangle related causal features, enhancing model performance. Furthermore, a dense connection module is expanded to prevent significant causal information loss in hidden states of inter layers. On the HCP brain task state classification task, Causal fMRI-Mamba achieves better performance and generalization than comparison methods.