Calibration-Aware and Interpretable Graph Learning for Multi-Cohort Diffusion Connectome Brain-Age Modeling
Alexandra BadeaiD, Ines Poves Acle, Paula Méndez de Inza, Hanwen Lin, Robert J. AndersoniD, Kim G. JohnsoniD, Heather Elizabeth WhitsoniD, Allen W. SongiD, Cristian T. BadeaiD, Alzheimer's Disease Neuroimaging Initiative, The HABS-HD Study Team
Abstract
Brain-age models derived from diffusion MRI-based structural connectomes may provide imaging biomarkers of accelerated brain aging, but their biological interpretation and transportability across heterogeneous populations remain uncertain. We developed a calibration-aware and hierarchically interpretable graph-learning framework and evaluated it across four independent aging and Alzheimer's disease-related cohorts: ADNI, Duke/UNC ADRC, HABS-HD, and AD-DECODE. The analysis included 1,093 connectome sessions from 789 participants. Cohort-specific graph neural networks were trained using participant-grouped cross-validation across five imaging and multimodal feature configurations. Prediction performance varied more strongly across cohorts than across feature sets, with imaging-only out-of-fold mean absolute error ranging from 4.72 years in ADNI to 9.75 years in AD-DECODE. The imaging-only graph neural network was competitive with ridge, elastic-net, and gradient-boosted regression models trained on matched vectorized connectome features, but was not uniformly superior. Age-bias-corrected brain-age gap was most consistently associated with reduced diffusion-derived microstructural integrity and structural-network organization across cohorts. In longitudinal analyses, corrected brain-age gap showed moderate-to-good within-person preservation in ADNI and HABS-HD, with intraclass correlation coefficients of 0.67 and 0.81, respectively; higher baseline values also predicted subsequent microstructural and network deterioration in ADNI. Multiscale SHAP analysis identified distributed contributions from global graph topology, regional imaging features, edge-derived regional summaries, and individual structural connections involving thalamic, striatal, frontal, parietal, cerebellar, hippocampal, and entorhinal circuitry. External transfer was highly sensitive to cohort shift: across 12 off-diagonal train-test evaluations, median mean absolute error decreased from 17.39 to 8.22 years after target-cohort linear recalibration, whereas median Pearson correlation remained 0.17. Because recalibration used target-cohort chronological age, it was interpreted as a diagnostic sensitivity analysis rather than deployable external validation. Together, these findings support calibration-aware diffusion-connectome brain age as an interpretable imaging biomarker of structural brain aging and prospective microstructural and network vulnerability, while emphasizing the need for cohort-specific calibration before external application.
02 /Specification & Classification
Click any tag to browse matching papers- Citations
- 0
- Organism
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Enhanced prediction of cognitive function using aging-sensitive networks within the human structural connectome
James W. Madole, Stuart James Ritchie, … Elliot M. Tucker–Drob·bioRxiv·Dec 13, 2019
- Predicting brain-age from multimodal imaging data captures cognitive impairment
Franziskus Liem, Gael P. Varoquaux, … Daniel S. Margulies·NeuroImagePeer-reviewed·Nov 23, 2016
- A Multiverse Approach to Brain Age Prediction: Exploring the Impact of Modality, Bias Adjustment, and Atlas
Jonas Festor, Lars Smolders, Wouter De Baene·bioRxiv·Oct 13, 2026
- Population Graph GNNs for Brain Age Prediction
Kamilė Stankevičiūtė, Tiago Azevedo, … Píetro Lió·bioRxiv·Jun 26, 2020
- Probing Multiple Algorithms to Calculate Brain Age: Examining Reliability, Relations with Demographics, and Predictive Power
Eva Bacas, Isabella Kahhalé, … Jamie Lars Hanson·Human Brain MappingPeer-reviewed·Apr 5, 2023
- From a deep learning model back to the brain - inferring morphological markers and their relation to aging
Gidon Levakov, Gideon Rosenthal, … Galia Avidan·bioRxiv·Nov 5, 2019