Brain-OF: An Omnifunctional Foundation Model for fMRI, EEG and MEG
Organizations: INM-4, Forschungszentrum Jülich, Germany · Department of Computer Science 3 - Software Engineering, RWTH Aachen University, Germany · INM-7, Forschungszentrum Jülich, Germany · Institute of Systems Neuroscience, Heinrich Heine University, Germany ·
Abstract
Brain foundation models have achieved remarkable advances across a wide range of neuroscience tasks. However, most existing models are limited to a single functional modality, restricting their ability to exploit complementary spatiotemporal dynamics and the collective data scale across different neuroimaging techniques. This limitation largely arises from severe semantic heterogeneity and resolution discrepancies among modalities. To address these challenges, we propose Brain-OF, an omnifunctional brain foundation model jointly pretrained on fMRI, EEG and MEG, capable of handling both unimodal and multimodal inputs within a unified framework. To reconcile heterogeneous spatiotemporal resolutions, we introduce the Any-Resolution Neural Signal Sampler, which projects diverse brain signals into a shared semantic space. To further manage semantic shifts, the Brain-OF backbone integrates DINT attention with a Sparse Mixture of Experts, where shared experts capture modality-invariant representations and routed experts specialize in modality-specific semantics. Furthermore, to explicitly internalize the characteristics of neural activity through self-supervised learning, we propose Masked Temporal-Frequency Modeling, a dual-domain pretraining objective that jointly reconstructs brain signals in both the time and frequency domains. Brain-OF is pretrained on a large-scale corpus comprising around 40 datasets and demonstrates superior performance across diverse downstream tasks, highlighting the benefits of joint multimodal integration and dual-domain pretraining.
Figures & tables
| Downstream Tasks | Datasets | Modalities | #Chs/ROIs | Durations | #Subjects | #Samples | Labels |
|---|---|---|---|---|---|---|---|
| Emotion Recognition | SEED-V | EEG | 62 | 1s | 20 | 117,744 | 5-class |
| Seizure Detection | CHB-MIT | EEG | 22 | 10s | 22 | 326,993 | 2-class |
| Abnormal Detection | TUAB | EEG | 19 | 10s | 2,383 | 409,455 | 2-class |
| AD Diagnosis | ADNI | fMRI | 246 | 300s | 222 | 618 | 2-class |
| ADHD Identification | ADHD-200 | fMRI | 246 | 360s | 926 | 1,369 | 2-class |
| Brain Age Prediction | CamCAN | fMRI | 246 | 520s | 652 | 652 | regression |
| Methods | SEED-V | CHB-MIT | TUAB | ADNI | ADHD-200 | CamCAN | CamCAN | LEMON | LEMON |
|---|---|---|---|---|---|---|---|---|---|
| EEG | EEG | EEG | fMRI | fMRI | fMRI | MEG | fMRI | EEG | |
| Kappa(%) | BAC(%) | BAC(%) | BAC(%) | BAC(%) | MAE | MAE | BAC(%) | BAC(%) | |
| EEGNet | 10.06 1.43 | 56.58 1.06 | 76.42 0.36 | 60.22 2.01 | 57.66 2.03 | 15.57 1.79 | 11.60 0.47 | 59.12 2.10 | 78.84 2.89 |
| EEGConformer | 17.72 1.74 | 59.76 1.41 | 77.58 0.49 | 65.16 2.45 | 62.00 2.64 | 8.56 0.72 | 10.03 0.21 | 67.27 4.15 | 78.94 2.35 |
| ContraWR | 19.05 1.88 | 63.44 0.02 | 77.46 0.41 | 59.69 4.97 | 62.67 1.69 | 9.15 0.39 | 9.01 0.40 | 63.97 1.28 | 77.42 0.45 |
| CNN-Transformer | 20.72 1.83 | 63.89 0.67 | 77.77 0.22 | 62.27 4.87 | 61.94 0.43 | 16.00 0.10 | 15.69 0.09 | 64.83 1.54 | 78.68 0.77 |
| Methods | SEED-V | CHB-MIT | TUAB | ADNI | ADHD-200 | CamCAN | CamCAN | LEMON | LEMON |
|---|---|---|---|---|---|---|---|---|---|
| EEG | EEG | EEG | fMRI | fMRI | fMRI | MEG | fMRI | EEG | |
| Kappa(%) | BAC(%) | BAC(%) | BAC(%) | BAC(%) | MAE | MAE | BAC(%) | BAC(%) | |
| Brain-OF (Base) | 23.98 0.29 | 73.99 1.48 | 81.88 0.17 | 68.23 2.07 | 62.57 1.19 | 12.09 0.78 | 8.99 0.14 | 58.95 2.13 | 76.05 0.44 |
| Objective Ablation | |||||||||
| - w/o MTFM | 21.52 0.57 | 71.43 2.30 | 81.49 0.18 | 65.26 1.83 | 60.35 1.50 | 12.80 1.15 | 8.96 0.33 | 57.18 1.91 | 76.90 0.51 |
| Modality Ablation |
| Datasets | Metrics | Modality | Models | ||
|---|---|---|---|---|---|
| Base | Large | Huge | |||
| CamCAN | MAE | fMRI-only | 12.09 0.78 | 11.71 0.67 | 9.51 0.51 |
| MEG-only | 8.99 0.14 | 8.86 0.21 | 7.87 0.22 | ||
| Multimodal Fusion | 8.87 0.09 | 8.78 0.07 | 8.21 0.11 | ||
| LEMON | BAC(%) | fMRI-only | 58.95 2.13 | 62.85 4.29 | 61.79 1.29 |
| EEG-only | 76.05 0.44 | 78.04 1.59 | 80.39 0.59 | ||
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| Modalities | Datasets | Paradigms | SR/TR | #Chs/ROIs | Durations | #Participants | #Samples |
| EEG | FACED | task | 250Hz | 32 | 10s | 123 | 10,332 |
| PhysioNet-MI | task | 160Hz | 64 | 4s | 109 | 18,697 | |
| PREDICT | rest | 5,000Hz | 63 | 10s | 159 | 17,511 | |
| REEG-BACA | rest | 1,000Hz | 64 | 10s | 608 | 53,089 | |
| SEED | task | 1,000Hz | 62 | 10s | 15 | 14,895 | |
| SEED-DV | task | 1,000Hz | 62 | 10s | 20 | 7,644 |
| Dataset | Pretraining | Relation | Train Subjects in PT | Val./Test Overlap | OOD Level |
|---|---|---|---|---|---|
| SEED-V | Independent | Dataset + Subject | |||
| CHB-MIT | Independent | Dataset + Site + Subject | |||
| TUAB | Partial | Subject | |||
| ADNI | Independent | Dataset + Site + Subject | |||
| ADHD-200 | Independent | Dataset + Site + Subject | |||
| CamCAN | Partial | Subject |
| Models | Params (M) | GFLOPs | Architecture | Pretraining Modality |
|---|---|---|---|---|
| Task-Specific Models (Trained from scratch) | ||||
| EEGNet | 0.004 | 0.13 | CNN | ✗ |
| EEGConformer | 0.81 | 0.58 | Convolutional Transformer | ✗ |
| ContraWR | 1.57 | 0.28 | CNN | ✗ |
| CNN-Transformer | 3.16 | 0.33 | CNN + Transformer | ✗ |
| FFCL | 2.47 | 0.96 | CNN + LSTM | ✗ |
| Mask Ratios | TUAB (EEG) | ADNI (fMRI) | CamCAN (MEG) | ||||||
|---|---|---|---|---|---|---|---|---|---|
| BAC(%) | AUC-PR(%) | AUROC(%) | BAC(%) | AUC-PR(%) | AUROC(%) | MAE | R2(%) | MAPE(%) | |
| 81.02 0.12 | 87.17 0.09 | 88.67 0.06 | 61.10 2.57 | 69.84 2.36 | 67.54 2.47 | 11.93 0.38 | 32.70 3.92 | 23.25 0.77 | |
| 81.74 0.21 | 87.59 0.13 | 89.29 0.19 | 65.01 2.91 | 75.24 2.28 | 71.42 2.84 | 8.92 0.18 | 58.19 1.71 | 17.00 0.31 | |
| 81.82 0.31 | 87.64 0.20 | 89.45 0.27 | 63.74 2.76 | 72.87 3.12 | 70.85 3.31 | 8.83 0.19 | 57.02 1.94 | 16.50 0.48 | |
| 81.88 0.17 | 87.69 0.11 | 89.26 0.09 | 68.23 2.07 | 76.23 1.87 | 73.27 0.71 | 8.99 0.14 | 56.19 1.84 | 16.92 0.30 | |
| 81.51 0.09 | 87.46 0.06 | 89.14 0.09 | 67.86 2.62 | 76.77 3.17 | 74.53 4.71 | 9.45 0.40 | 53.27 3.62 | 18.06 0.51 | |
| Methods | BrainHarmonix-F | Brain-JEPA | Brain-OF Base |
|---|---|---|---|
| HAD (Top-10 Acc %) | 6.38 0.16 | 12.31 1.69 | 16.17 0.44 |
| BOLD5000 (BAC %) | 38.94 1.66 | 39.79 0.41 | 40.49 0.67 |
| Methods | CHB-MIT | TUAB | CamCAN |
|---|---|---|---|
| EEG | EEG | MEG | |
| BAC(%) | BAC(%) | MAE | |
| BrainOmni Base | 69.47 2.19 | 81.90 0.50 | 11.99 0.84 |
| Brain-OF Base | 73.99 1.48 | 81.88 0.17 | 8.99 0.14 |
| Methods | CamCAN | LEMON | ||||
|---|---|---|---|---|---|---|
| fMRI | MEG | Multimodal | fMRI | EEG | Multimodal | |
| MAE | MAE | MAE | BAC(%) | BAC(%) | BAC(%) | |
| BrainHarmonix | 14.81 0.49 | 8.73 0.09 | 11.61 0.75 | 63.11 1.32 | 77.95 2.36 | 74.43 0.98 |
| Brain-OF Base | 12.09 0.78 | 8.99 0.10 | 8.87 0.09 | 58.95 2.13 | 76.05 0.44 | 76.77 0.60 |
| Brain-OF Large | 11.71 0.67 | 8.86 0.21 | 8.78 0.07 | 62.85 4.29 | 78.04 1.59 | 78.32 1.64 |
| Brain-OF Huge | 9.51 0.51 | 7.87 0.22 | 8.21 0.11 | 61.79 1.29 | 80.39 0.59 | 81.10 0.54 |
| Hyperparameters | Brain-OF Base | Brain-OF Large | Brain-OF Huge | |
|---|---|---|---|---|
| Model Size | 47.5M | 331M | 1.7B | |
| Activated Parameters | 21.5M | 150M | 500M | |
| Patch size | 64 | |||
| Max sequence length | 2048 | |||
| Semantic space size | 128 | |||
| Padding token | Full zero | |||
| Datasets | Global bsz | Epochs | Warmup epochs | Lr | Warmup lr | Wd | Drop out | Drop path |
|---|---|---|---|---|---|---|---|---|
| SEED-V | 1,024 | 50 | 5 | 1e-4 | 1e-6 | 5.0 | 0.0 | 0.0 |
| CHB-MIT | 1,024 | 50 | 5 | 3e-5 | 3e-7 | 0.1 | 0.0 | 0.1 |
| TUAB | 1,024 | 50 | 5 | 7e-6 | 7e-8 | 0.1 | 0.0 | 0.0 |
| ADNI | 32 | 50 | 5 | 1.5e-4 | 1.5e-6 | 0.1 | 0.1 | 0.0 |
| ADHD-200 | 32 | 50 | 5 | 3e-5 | 3e-7 | 0.01 | 0.1 | 0.0 |
| CamCAN-fMRI | 32 | 100 | 5 | 7.5e-4 | 7.5e-6 | 0.1 | 0.1 | 0.0 |
| Datasets | Global bsz | Epochs | Warmup epochs | Lr | Warmup lr | Wd | Drop out | Drop path |
|---|---|---|---|---|---|---|---|---|
| SEED-V | 1,024 | 50 | 5 | 4e-4 | 4e-6 | 5.0 | 0.0 | 0.1 |
| CHB-MIT | 1,024 | 50 | 5 | 3e-5 | 3e-7 | 0.5 | 0.0 | 0.1 |
| TUAB | 1,024 | 50 | 5 | 3e-5 | 3e-7 | 0.1 | 0.0 | 0.3 |
| ADNI | 32 | 50 | 5 | 1e-4 | 1e-6 | 0.1 | 0.0 | 0.3 |
| ADHD-200 | 32 | 50 | 5 | 4e-5 | 4e-7 | 0.01 | 0.1 | 0.2 |
| CamCAN-fMRI | 32 | 100 | 5 | 3e-3 | 3e-5 | 0.1 | 0.1 | 0.1 |
| Datasets | Global bsz | Epochs | Warmup epochs | Lr | Warmup lr | Wd | Drop out | Drop path |
|---|---|---|---|---|---|---|---|---|
| SEED-V | 1,024 | 50 | 5 | 1e-4 | 1e-6 | 5.0 | 0.0 | 0.1 |
| CHB-MIT | 1,024 | 50 | 5 | 1e-4 | 1e-6 | 0.1 | 0.0 | 0.1 |
| TUAB | 1,024 | 50 | 5 | 7e-6 | 7e-8 | 0.1 | 0.0 | 0.1 |
| ADNI | 32 | 50 | 5 | 3e-4 | 3e-6 | 0.1 | 0.0 | 0.1 |
| ADHD-200 | 32 | 50 | 5 | 1e-4 | 1e-6 | 0.1 | 0.0 | 0.3 |
| CamCAN-fMRI | 32 | 100 | 5 | 5e-3 | 5e-5 | 0.1 | 0.0 | 0.1 |
| Model Variants | Total | Active | Comp. | Pretraining Cost | Inference (Batch=16) | ||
|---|---|---|---|---|---|---|---|
| Params | Params | GFLOPs | Time | # GPUs | Latency | Throughput | |
| Brain-OF Base | 47.5M | 21.5M | 11.29 | 71h 50m | 32 | 88 ms | 181.8 samples/s |
| Brain-OF Large | 331M | 150M | 61.01 | 82h 06m | 88 | 178 ms | 89.9 samples/s |
| Brain-OF Huge | 1.7B | 500M | 178.19 | 153h 03m | 224 | 297 ms | 53.8 samples/s |
| Models | Training Setting | TUAB | ADNI | CamCAN |
|---|---|---|---|---|
| EEG | fMRI | MEG | ||
| BAC (%) | BAC (%) | MAE | ||
| Brain-OF Base | Pretrained + Full FineTuning | 81.88 0.17 | 68.23 2.07 | 8.99 0.14 |
| Pretrained + Linear Probing | 79.14 0.24 | 65.22 1.47 | 10.92 0.22 | |
| No Pretraining + Full FineTuning | 77.59 0.27 | 63.39 2.36 | 9.69 1.07 | |
| Brain-OF Huge | Pretrained + Full FineTuning | 82.87 0.18 | 71.79 2.19 | 7.87 0.22 |
| Methods | SEED-V | TUAB | ADNI | ADHD-200 | CamCAN |
|---|---|---|---|---|---|
| EEG | EEG | fMRI | fMRI | MEG | |
| Kappa(%) | BAC(%) | BAC(%) | BAC(%) | MAE | |
| Brain-OF (Base) | 23.98 0.29 | 81.88 0.17 | 68.23 2.07 | 62.57 1.19 | 8.99 0.14 |
| w/ Dense FFN | 20.86 0.28 | 81.86 0.08 | 64.60 3.32 | 59.10 1.80 | 9.78 0.27 |
| w/ Vanilla Attn. | 22.82 0.38 | 81.85 0.09 | 64.32 2.82 | 62.82 1.76 | 9.50 0.53 |
| Methods | 20 dB | 10 dB | 5 dB | 0 dB | -5 dB |
|---|---|---|---|---|---|
| DINT Attention | 80.18 0.33 | 75.27 0.88 | 75.02 0.28 | 73.47 1.11 | 69.29 2.34 |
| Vanilla Attention | 78.93 0.40 | 74.20 1.06 | 72.71 1.21 | 70.53 1.86 | 63.96 3.47 |
| Methods | SEED-V | TUAB | ADNI | ADHD-200 | CamCAN | LEMON |
|---|---|---|---|---|---|---|
| EEG | EEG | fMRI | fMRI | MEG | fMRI | |
| Kappa(%) | BAC(%) | BAC(%) | BAC(%) | MAE | BAC(%) | |
| Full Finetuning | 23.98 0.29 | 81.88 0.17 | 68.23 2.07 | 62.57 1.19 | 8.99 0.14 | 58.95±2.13 |
| LoRA ( 2% Params) | 21.81 0.13 | 81.30 0.04 | 68.44 2.13 | 62.10 1.04 | 9.17 0.44 | 59.36 3.02 |