FlexiBrain: Resolution-Agnostic Voxel-Level Encoding for Native fMRI
On the map: State-space sequence models (S4, Mamba) → Physiological signal modeling
Abstract (arXiv)
The success of large-scale deep learning models in neuroscience is fundamentally constrained by severe data heterogeneity. Native fMRI data aggregated from diverse sources exhibit substantial variation in both spatial and temporal resolutions. Consequently, most existing frameworks rely on lengthy, rigid preprocessing pipelines that enforce uniformity across datasets. This practice introduces two critical limitations: (1) potential degradation of subject-specific anatomical information; (2) significant computational overhead, often requiring hours of processing per subject. Here, we propose FlexiBrain, a resolution-agnostic voxel-level encoding framework for native fMRI based on Mamba-JEPA. FlexiBrain defines patch sizes in real-world physical units and employs a dynamic patch resizing, thereby bypassing destructive spatial standardization while enabling direct ingestion of data in native space. We instantiate the framework using an efficient Mamba-JEPA backbone to model high-dimensional 4D fMRI signals. Across five diverse downstream neuroscience tasks, FlexiBrain consistently outperforms recent state-of-the-art methods, achieving gains of up to 12 percentage points without external data augmentation. Importantly, FlexiBrain functions as a seamless plug-in module, substantially reducing preprocessing costs and accelerating the development of robust voxel-level fMRI foundation models. Code is available at https://github.com/OneMore1/FlexiBrain.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Placed in Physiological signal modeling, one of 10 directions in State-space sequence models (S4, Mamba). Builds on 18 of its 34 references that are in the map. First posted on arXiv 9 Jun 2026.
Builds on
18 of 34 references.
References · 18
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 634
- 2023Reconstructing the Mind's Eye: fMRI-to-Image with Contrastive Learning and Diffusion Priorscited by 160
- 2022FlexiViT: One Model for All Patch Sizescited by 152
- 2024MindEye2: Shared-Subject Models Enable fMRI-To-Image With 1 Hour of Datacited by 109
- 2024Brain-JEPA: Brain Dynamics Foundation Model with Gradient Positioning and Spatiotemporal Maskingcited by 65
- 2023SwiFT: Swin 4D fMRI Transformercited by 36
- 2024BrainMass: Advancing Brain Network Analysis for Diagnosis with Large-scale Self-Supervised Learningcited by 32
- 2025Towards a general-purpose foundation model for fMRI analysiscited by 15
- 2025Brain Harmony: A Multimodal Foundation Model Unifying Morphology and Function into 1D Tokenscited by 15
- 2026Omni-fMRI: A Universal Atlas-Free fMRI Foundation Modelcited by 11
- 2026Brain-DiT: A Universal Multi-state fMRI Foundation Model with Metadata-Conditioned Pretrainingcited by 11
- 2025SLIM-Brain: A Data- and Training-Efficient Foundation Model for fMRI Data Analysiscited by 9
6 more
- 2022Explainable fMRI-based Brain Decoding via Spatial Temporal-pyramid Graph Convolutional Networkcited by 7
- 2024Uncovering cognitive taskonomy through transfer learning in masked autoencoder-based fMRI reconstructioncited by 7
- 2022Transfer learning to decode brain states reflecting the relationship between cognitive taskscited by 5
- 2025Voxel-Level Brain States Prediction Using Swin Transformercited by 5
- 2025DCA: Graph-Guided Deep Embedding Clustering for Brain Atlasescited by 4
- 2025Whole-brain Transferable Representations from Large-Scale fMRI Data Improve Task-Evoked Brain Activity Decodingcited by 2
Built on by
Citing papers per year: 2026: 6.
6 papers.
- 2026BrainWorld: A Structural-Prior-Conditioned Generative Model for Whole-Brain 4D fMRI Dynamicscited by 5
- 2026Beyond Single-Source Cognitive Taskonomy:Multi-Source Task Relations through fMRI Transfer Learningcited by 1
- 2026BrainTaskonomy: Learning How to Pretrain and What to Transfer in fMRI Foundation Modelscited by 1
- 2026Omni-Sleep: A Sleep Foundation Model via Hierarchical Contrastive Learning of CNS-ANS Dynamicscited by 0
- 2026A Scaling Study for fMRI Foundation Modelscited by 0
- 2026FlatClip: Reusing Image Foundation Models for fMRI Representation Learning via Cortical Flatmapscited by 0
- 2026FlatClip: Reusing Image Foundation Models for fMRI Representation Learning via Cortical Flatmaps25 Sep 2026
- 2026A Scaling Study for fMRI Foundation Models23 Sep 2026
- 2026BrainTaskonomy: Learning How to Pretrain and What to Transfer in fMRI Foundation Models9 Sep 2026
- 2026Omni-Sleep: A Sleep Foundation Model via Hierarchical Contrastive Learning of CNS-ANS Dynamics4 Jul 2026
- 2026Beyond Single-Source Cognitive Taskonomy:Multi-Source Task Relations through fMRI Transfer Learning24 Jun 2026
- 2026BrainWorld: A Structural-Prior-Conditioned Generative Model for Whole-Brain 4D fMRI Dynamics16 Jun 2026
Similar, not cited
Close in wording, no citation either way
- 2026Coarse-to-fine Hierarchical Architecture with Sequential Mamba for Brain Reconstruction
- 2024Hierarchical Spatio-Temporal State-Space Modeling for fMRI Analysis
- 2026Rhamba: Region-Aware Hybrid Attention-Mamba Framework for Self-Supervised Learning in Resting-State fMRI
- 2025FreqSelect: Frequency-Aware fMRI-to-Image Reconstruction
- 2025Sparse Autoencoders Bridge The Deep Learning Model and The Brain
- 2025BrainMT: A Hybrid Mamba-Transformer Architecture for Modeling Long-Range Dependencies in Functional MRI Data
- 2025fMRI-LM: Towards a Universal Foundation Model for Language-Aligned fMRI Understanding
- 2026NeurIPS: Neuro-anatomical Inductive Priors for Sphere-based Brain Decoding