用脑龄预测模型做阿尔茨海默病影像分析,仅需1%参数就能跨数据集通用。
A Generalizable Feature Extractor for Alzheimer's-Related Brain MRI Tasks
- 冻结预训练3D CNN权重,用低秩适配仅调1%参数。
- 在多个任务中表现优异,最高AUC达0.964,跨数据集仍保持0.871。
- 适合资源受限的神经影像研究,尤其适合小样本场景。
当标注数据不足时,迁移学习可提供帮助,但其在神经影像中的有效性仍不明确,且是否可在新数据集上无需重训即适用尚不清楚。我们评估了一个紧凑的、基于脑龄预测的监督预训练模型,能否作为下游神经影像任务的通用基础模型。冻结一个已训练好的3D CNN(718万参数),使用低秩适配(LoRA)进行任务微调,仅需约1%额外可训练参数。在六个实验中验证泛化能力:在ADNI数据集上分类正常与痴呆,AUC达0.964;直接应用于未见过的OASIS-3数据集,无需重训,AUC为0.871;结合年龄和认知评分,区分稳定与进展性轻度认知障碍,AUC为0.828;预测淀粉样蛋白阳性,AUC为0.804;直接从T1w图像估计标准化海马体和白质低信号体积,决定系数分别为0.80和0.91,性能媲美大型U-Net网络。结果表明,经过精心训练的脑龄模型可作为阿尔茨海默病相关任务的有效基础模型,即使在严格数据约束下也能实现高效跨任务、跨数据集迁移。
原文摘要 · Abstract (English)
When there is not enough labeled data to properly train deep learning models, transfer learning can help. We still do not fully understand how effective it is in neuroimaging, especially for Alzheimer's disease research. It is also not clear if these transferred models can work on new datasets without being retrained for each specific task. We evaluate whether a compact, supervised pretrained model can serve as a reusable foundation model for downstream neuroimaging tasks. We freeze the 7.18 million weights of a 3D CNN previously trained for brain-age prediction, and adapt it to each task using Low-Rank Adaptation (LoRA), requiring only ~1% additional trainable parameters. We evaluate generalizability in six experiments. Adapting the model to classify cognitively normal versus Dementia on ADNI gave an AUC of 0.964 on held-out folds (Experiment #1). Applying that adapted model unchanged to OASIS-3, with no retraining, gave an AUC of 0.871 (Experiment #2). Reusing its output logit together with age and a cognitive score distinguished stable from progressing MCI with an AUC of 0.828 (Experiment #3). Adapting the same backbone to predict amyloid positivity from structural MRI gave an AUC of 0.804 (Experiment #4). Finally, the same approach estimated ICV-normalized hippocampal and white matter hypointensity volumes directly from the T1w image, with R^2 of 0.80 and 0.91 respectively, tasks normally addressed with much larger U-Net networks (Experiments #5 and #6). A compact model supervised on brain age can therefore serve as a reusable backbone, adapting to each task with ~1% additional parameters and transferring to an unseen cohort without any training. Our findings suggest that a carefully trained brain age model can serve as an effective foundation model for Alzheimer's related tasks, even under strict data constraints.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。