用对比学习提升脑龄预测精度,揭示阿尔茨海默病与帕金森病患者脑龄差距。
MRI-Based Brain Age Estimation with Supervised Contrastive Learning of Continuous Representation
- 采用监督对比学习和RNC损失函数,捕捉脑结构变化的连续性特征。
- 在小样本下实现4.27年MAE和0.93的R²,优于传统深度回归方法。
- 通过Grad-RAM可视化解释结果,发现疾病严重程度与脑龄差相关。
基于MRI的脑龄估计旨在根据神经解剖特征评估个体的生物脑龄。多种因素(如神经退行性疾病)可加速脑衰老,其测量或可作为临床生物标志物。尽管深度学习回归近年受到关注,但现有方法常难以捕捉神经形态变化的连续性,导致特征表示与性能受限。本文首次将监督对比学习结合最近提出的Rank-N-Contrast(RNC)损失用于标准T1w结构MRI的脑龄估计,并采用Grad-RAM进行结果可视化解释。实验表明,该方法在有限训练样本下取得4.27年均绝对误差(MAE)和0.93的$R^2$,显著优于同主干网络的常规深度回归,且表现优于或相当主流方法(后者使用更大训练数据)。此外,Grad-RAM揭示了使用RNC损失时更精细的年龄相关特征。作为探索性研究,我们应用该方法估算阿尔茨海默病与帕金森病患者的生物脑龄与日历脑龄差异,发现二者间存在相关性,验证其在神经退行性疾病中作为生物标志物的潜力。
原文摘要 · Abstract (English)
MRI-based brain age estimation models aim to assess a subject's biological brain age based on information, such as neuroanatomical features. Various factors, including neurodegenerative diseases, can accelerate brain aging and measuring this phenomena could serve as a potential biomarker for clinical applications. While deep learning (DL)-based regression has recently attracted major attention, existing approaches often fail to capture the continuous nature of neuromorphological changes, potentially resulting in sub-optimal feature representation and results. To address this, we propose to use supervised contrastive learning with the recent Rank-N-Contrast (RNC) loss to estimate brain age based on widely used T1w structural MRI for the first time and leverage Grad-RAM to visually explain regression results. Experiments show that our proposed method achieves a mean absolute error (MAE) of 4.27 years and an $R^2$ of 0.93 with a limited dataset of training samples, significantly outperforming conventional deep regression with the same ResNet backbone while performing better or comparably with the state-of-the-art methods with significantly larger training data. Furthermore, Grad-RAM revealed more nuanced features related to age regression with the RNC loss than conventional deep regression. As an exploratory study, we employed the proposed method to estimate the gap between the biological and chronological brain ages in Alzheimer's Disease and Parkinson's disease patients, and revealed the correlation between the brain age gap and disease severity, demonstrating its potential as a biomarker in neurodegenerative disorders.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。