MAMBAxBrain: A Multi-task Neural Framework Linking Brain Functional Dynamics to Individual Fingerprints, Cognitive and Disease States
Xia, Y.; Arab, F.; Saha, U.; Sipes, B.; Gooden, G.; Chen, M.; Raj, A.
Show abstract
Functional magnetic resonance imaging (fMRI) contains rich individual, cognitive, and pathological information, yet no universal model exists for multi-task modeling of these dimensions. Here, we introduce MAMBAxBrain, a multi-task neural framework that integrates Mamba architecture with functional connectivity analysis to jointly model the temporal dynamics and spatial coordination of neural activity. MAMBAxBrain achieves high accuracy across four distinct fMRI objectives--brain fingerprinting, cognitive task decoding, reaction time prediction, and schizophrenia classification--consistently outperforming state-of-the-art methods with robust crosssession generalization. Interpretability analyses show that each task engages distinct, biologically plausible circuitry--from higher-order association cortex for identity to subcortical-motor loops for reaction time and disrupted control-sensory connectivity for schizophrenia. These findings inform a longstanding debate: rather than operating through wholly separate or entirely shared systems, the brain preferentially recruits task-specific circuits while retaining common representational structure across functions.
Matching journals
The top 4 journals account for 50% of the predicted probability mass.