bilalEthizo/BrainAge-ResNet18

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