A foundation-model approach to pediatric headache classification from resting-state fMRI.

Abstract

Headache is the most common neurological disorder in children and substantially affects quality of life. We investigated whether resting-state functional MRI (rs-fMRI) can support pediatric headache classification using machine learning. We encoded rs-fMRI data using NeuroSTORM, a recent foundation model, and fine-tuned it to distinguish healthy controls from children with headache and subsequently classify headache subtypes. We then compared NeuroSTORM with a standard neuroscience approach that uses functional connectivity (FC) matrices derived from brain activity as predictors. Using 189 rs-fMRI scans from 110 individuals collected across two visits (prevalence of any headache: 74%), NeuroSTORM achieved an area under the receiver operating characteristic curve (AUROC) of 0.82 (95% CI, 0.82-0.82) and an area under the precision-recall curve (AUPRC) of 0.93 (95% CI, 0.93-0.94) in discriminating headache from non-headache. In contrast, models trained on FC matrices showed limited performance (AUROC, 0.67 [95% CI, 0.67-0.67]; AUPRC, 0.85 [95% CI, 0.85-0.85]). In a multiclass setting, when tasked with classifying individuals as healthy controls, individuals with chronic migraine, or individuals with non-chronic headaches (e.g., post-viral headache, new daily persistent headache, post-traumatic headache), NeuroSTORM achieved a macro-AUROC of 0.69 (95% CI, 0.68-0.69). The results suggest this approach can distinguish chronic migraine, the most common headache syndrome, but has difficulty differentiating other headache subtypes from chronic migraine. Overall, under limited-data conditions, NeuroSTORM appears to capture latent rs-fMRI representations that transfer to headache-related tasks. The findings provide proof-of-concept for fMRI-based prediction of pediatric headache using a foundation model without relying on functional connectivity features and highlight the potential of this approach for subtype identification. Further development of such tools may help clinicians improve diagnosis based on brain activity and tailor treatment strategies to individual patients.