metadata
license: cc-by-nc-4.0
tags:
- brain-age
- neuroimaging
- mri
- 3d-cnn
- pytorch
BrainAge — Dual-Branch Brain Age Predictor (resnet18)
3D RESNET18 + tabular MLP trained on 6,050 healthy T1w brain MRIs (0–86 years).
Checkpoints
| File | Description |
|---|---|
brainage_resnet18.pt |
Phase A: lifespan pretraining (all ages) |
brainage_resnet18_finetune.pt |
Phase B: fine-tuned for pediatric (0–25 y) |
Training data
bilalEthizo/BrainAge-Golden-Preprocessed
Architecture
- Image: resnet18 3D CNN, input 128×144×112
- Tabular: MLP over 86-dim vector (70 regions + sex + site)
- Fusion: concat → 64 → 1 (regression in years)
- Loss: HuberLoss(delta=3.0)
- Training: fp16 mixed precision, cosine LR, age-bin weighted sampler