BrainNorm: A Foundation Model that knows Normal via Semantic Atlas Pretraining
Organizations: Indian Institute of Technology Hyderabad
Abstract
We introduce BrainNorm, a normative foundation model, trained and tested on ~66,000 T1-weighted structural MRI (T1w sMRI) scans. By leveraging language-image style contrastive pretraining on healthy cohorts across ages, BrainNorm learns a Semantic Atlas Latent space (SAL), where each scan is represented as a set of atlas-parcel embeddings. This yields parcel-specific healthy aging template trajectories that support age-consistent template matching and localized deviation scoring relative to a subject's chronological age. Across 6 downstream cohorts, BrainNorm demonstrates generalization evaluated across 25 task-setting combinations spanning age estimation, brain-age gap estimation, parcel identification, and single- & multi-disease classification tasks under direct inference, zero-shot, few-shot & full-data linear-probe settings. The resulting deviation patterns in SAL space enable zero-shot tasks for disease prediction using parcel-wise abnormalities. Fine-tuning on healthy-only cohorts of downstream datasets further improves the performance of various tasks. Across all classification tasks, linear probing on BrainNorm's frozen embeddings outperforms 9 baselines finetuned under end-to-end supervision. Furthermore, the localized deviations identified by BrainNorm across various neurodegenerative disorders closely align with established neurodegeneration pathology in clinical literature.
Figures & tables
| Type | Metric | Mode | UKB | MCSA | ADNI | AIBL | MIRIAD | NIFD | PPMI |
|---|---|---|---|---|---|---|---|---|---|
| T P | Acc@1 | No FT | 90.87 | 67.33 | |||||
| FT | 90.32 | 76.06 | |||||||
| Acc@5 | No FT | 99.02 | 85.19 | ||||||
| FT | 99.04 | 94.54 | |||||||
| P T | Acc@1 | No FT | 97.17 | 75.02 | |||||
| FT | 96.72 | 86.46 |
Appendix figures & tables22 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | #scans | #subjects | Age (y) | Gender | Field strength | Manufacturer |
| (min,max, ) | (M / F) | (1.5T / 3.0T) | (Siemens/GE/Philips) | |||
| UKB (CN) | 48,018 | 43,631 | 20,592 / 23,039 | 0 / 43,631 | 43,631 / 0 / 0 | |
| MCSA (CN) | 2,662 | 1,611 | 853 / 758 | 0 / 1,611 | 0 / 1,611 / 0 | |
| ADNI | 5,627 | 1,703 | 734 / 969 | 429 / 1,411 | 1,063 / 476 / 289 | |
| ADNI (MCI) | 5,308 | 687 | 408 / 279 | 275 / 512 | 389 / 258 / 134 | |
| AIBL | 1,084 | 599 | 255 / 343 | 107 / 541 | 599 / 0 / 0 |
| Dataset | Ablation | Text Parcel | Parcel Text | Age-MAE | Pearson- | ||
|---|---|---|---|---|---|---|---|
| Acc@1 | Acc@5 | Acc@1 | Acc@5 | ||||
| UKB | 1 | 90.87 | 99.02 | 97.17 | 99.80 | 4.14 | 0.85 |
| UKB | 2 | 90.99 | 99.09 | 97.99 | 99.99 | 4.98 | 0.77 |
| UKB | 3 | 0.48 | 2.34 | 0.32 | 2.05 | 7.74 | 0.44 |
| MCSA | 1 | 67.33 | 85.19 | 75.02 | 89.54 | 3.94 | 0.88 |
| MCSA | 2 | 70.53 | 86.60 | 74.19 | 88.12 | 6.78 | 0.84 |
| Dataset | Initialization | Text Parcel | Parcel Text | Age-MAE | Pearson- | ||
|---|---|---|---|---|---|---|---|
| Acc@1 | Acc@5 | Acc@1 | Acc@5 | ||||
| UKB | Random | 55.93 | 58.34 | 56.02 | 58.19 | 10.32 | 0.68 |
| UKB | Categorical | 99.99 | 99.99 | 99.98 | 99.99 | 10.05 | 0.73 |
| UKB | Learnable | 91.20 | 99.21 | 96.91 | 97.71 | 7.29 | 0.79 |
| UKB | Qwen3 | 90.87 | 99.02 | 97.17 | 99.80 | 4.14 | 0.85 |
| MCSA | Random | 15.37 | 17.80 | 14.97 | 17.65 | 17.40 | 0.67 |
| Dataset | Template | Text Parcel | Parcel Text | Age-MAE | Pearson- | ||
|---|---|---|---|---|---|---|---|
| Acc@1 | Acc@5 | Acc@1 | Acc@5 | ||||
| UKB | 1 | 90.87 | 99.02 | 97.17 | 99.80 | 4.14 | 0.85 |
| UKB | 2 | 86.10 | 97.87 | 94.49 | 99.66 | 4.47 | 0.85 |
| UKB | 3 | 85.96 | 97.89 | 93.43 | 99.58 | 3.99 | 0.85 |
| UKB | 4 | 80.39 | 96.57 | 93.48 | 99.45 | 3.91 | 0.85 |
| MCSA | 1 | 67.33 | 85.19 | 75.02 | 89.54 | 3.94 | 0.88 |
| Parcel Clustering | Age Ordering Within Parcel | |||||
|---|---|---|---|---|---|---|
| Encoder | Silhouette | Hit@1 | NMI | Spearman | Pair Order Acc | kNN Age MAE |
| Qwen3-Embedding-8B Yang et al. (2025) | 0.4637 | 0.9999 | 0.9476 | 0.9682 | 0.9311 | 0.5114 |
| MedEmbed-small Balachandran (2024) | 0.6667 | 1.0000 | 0.9962 | 0.5483 | 0.6922 | 1.9061 |
| MedSigLIP Sellergren et al. (2025) | 0.1224 | 0.7284 | 0.8790 | 0.2906 | 0.5971 | 6.0694 |
| BiomedCLIP Zhang et al. (2025b) | 0.6133 | 0.9999 | 0.9863 | 0.3816 | 0.6323 | 2.4691 |
| Hashing+SVD | 0.4440 | 0.9998 | 0.9939 | 0.0024 | 0.5010 | 3.4503 |
| Dataset | Mode | CN data | Text Parcel | Parcel Text | ||
|---|---|---|---|---|---|---|
| Acc@1 | Acc@5 | Acc@1 | Acc@5 | |||
| UKB | No FT | UKB | ||||
| FT | UKB | |||||
| MCSA | No FT | UKB | ||||
| FT | All | |||||
| ADNI | No FT | UKB | ||||
| Class | Parcel IDs |
|---|---|
| AD/MCI | 38, 39, 40, 41, 36, 37, 68, 69, 66, 67, 62, 63, 84, 85, 18, 19, 216, 217 |
| PD | 156, 157, 158, 159, 74, 75, 72, 73, 76, 77, 154, 155, 160, 161, 162, 163, 164, 165, 0, 1, 14, 15, 58, 59, 172, 173 |
| FTD | 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 82, 83, 86, 87, 80, 81, 84, 85, 32, 33, 146, 147, 148, 149, 150, 151, 24, 25, 26, 27, 28, 29, 30, 31, 20, 21, 152, 153 |
| Task | HBR | HBR | BrainNorm | BrainNorm |
|---|---|---|---|---|
| Direct Transfer | CN-only Adaptation | Direct Transfer | CN-only Fine-tuning | |
| AD/CN [ADNI] | 75.20 | 72.01 | 54.78 | 72.37 |
| AD/CN [MIRIAD] | 90.81 | 88.97 | 71.00 | 93.51 |
| AD/CN [AIBL] | 71.04 | 72.51 | 49.55 | 70.70 |
| sMCI/pMCI [ADNI] | 58.81 | 59.89 | 47.33 | 62.79 |
| PD/CN [PPMI] | 51.86 | 49.96 | 49.60 | 53.02 |
| Task | HBR BAcc | HBR AUC | BrainNorm BAcc | BrainNorm AUC | BAcc | AUC |
|---|---|---|---|---|---|---|
| AD/CN [ADNI] | 82.29 | 90.25 | 86.85 | 94.27 | +4.56 | +4.02 |
| AD/CN [AIBL] | 83.53 | 91.06 | 85.91 | 93.39 | +2.38 | +2.33 |
| AD/CN [MIRIAD] | 88.47 | 94.57 | 92.72 | 98.26 | +4.25 | +3.69 |
| sMCI/pMCI [ADNI] | 67.47 | 74.55 | 71.96 | 80.75 | +4.49 | +6.20 |
| PD/CN [PPMI] | 55.87 | 60.42 | 59.08 | 66.12 | +3.21 | +5.70 |
| FTD/CN [NIFD] | 70.14 | 75.42 | 81.82 | 89.15 | +11.68 | +13.73 |
| Dataset | Method | Mode | CN Train/FT data | MAE | RMSE | Pearson | |
| UKB | 3D-ViT | Full Train | UKB | ||||
| BrainIAC | Full FT | UKB | |||||
| BrainMVP | Full FT | UKB | |||||
| BrainHarmony | Full FT | UKB | |||||
| RadFM | Full FT | UKB | |||||
| NeuroVFM | Full FT | UKB |
| Dataset/Task | Method | Full FT | No FT | FT (own) | FT (all CN) |
|---|---|---|---|---|---|
| ADNI [AD/CN] | A | - | |||
| P+A | - | ||||
| BAG | - | ||||
| BrainIAC+BAG | - | - | - | ||
| BrainMVP+BAG | - | - | - | ||
| BrainHarmony+BAG | - | - | - |
| Method | Full FT | No FT+LP | CN-only FT (own)+LP | CN-only FT (all)+LP | ||||
| BACC | AUC | BACC | AUC | BACC | AUC | BACC | AUC | |
| ADNI | AD/CN | ||||||||
| 3D-ViT | – | – | – | – | – | – | ||
| BrainHarmony | – | – | – | – | – | – | ||
| BrainMVP | – | – | – | – | – | – | ||
| BrainIAC | – | – | – | – | – | – | ||
| Method | Full FT | No FT+LP | CN-only FT (own)+LP | CN-only FT (all)+LP | ||||
| BACC | AUC | BACC | AUC | BACC | AUC | BACC | AUC | |
| NIFD | FTD/CN | ||||||||
| 3D-ViT | – | – | – | – | – | – | ||
| BrainHarmony | – | – | – | – | – | – | ||
| BrainMVP | – | – | – | – | – | – | ||
| BrainIAC | – | – | – | – | – | – | ||
| Method | Full FT | No FT+LP | CN-only FT (own)+LP | CN-only FT (all)+LP | ||||
| BACC | AUC | BACC | AUC | BACC | AUC | BACC | AUC | |
| ADNI+NIFD+PPMI | AD/FTD/PD | ||||||||
| 3D-ViT | – | – | – | – | – | – | ||
| BrainHarmony | – | – | – | – | – | – | ||
| BrainMVP | – | – | – | – | – | – | ||
| BrainIAC | – | – | – | – | – | – | ||
| Method | Mode | |||||
| ADNI | [AD/CN] | ||||||
| BrainNorm_M1 | No FT + LP | |||||
| CN-only FT(own) + LP | ||||||
| CN-only FT (all) + LP | ||||||
| BrainNorm_M2 | No FT + LP | |||||
| CN-only FT(own) + LP | ||||||
| Method | Mode | |||||
|---|---|---|---|---|---|---|
| ADNI+NIFD+PPMI | [AD/FTD/PD] | ||||||
| BrainNorm_M1 | No FT + LP | |||||
| CN-only FT(own) + LP | ||||||
| CN-only FT (all) + LP | ||||||
| BrainNorm_M2 | No FT + LP | |||||
| CN-only FT(own) + LP | ||||||
| Parcel ID | Atlas | Tissue_Class | Label ID | Region |
|---|---|---|---|---|
| 0 | AAL3 | Gray_Matter | 1 | Left Precentral Gyrus |
| 1 | AAL3 | Gray_Matter | 2 | Right Precentral Gyrus |
| 2 | AAL3 | Gray_Matter | 3 | Left Superior Frontal Gyrus |
| 3 | AAL3 | Gray_Matter | 4 | Right Superior Frontal Gyrus |
| 4 | AAL3 | Gray_Matter | 5 | Left Middle Frontal Gyrus |
| 5 | AAL3 | Gray_Matter | 6 | Right Middle Frontal Gyrus |
Medical Imaging Foundation Models
Medical imaging foundation models learn reusable representations for tasks across clinical images, modalities, and institutions.
See all 234 papers