通过时空自监督学习,提升阿尔茨海默病预测模型在不同数据集和输入条件下的泛化能力。
SSL-AD: Spatiotemporal Self-Supervised Learning for Generalizability and Adaptability Across Alzheimer's Prediction Tasks and Datasets
- 设计新方法处理变长输入与时间间隔差异,增强模型适应性。
- 在7个任务中6个超越有监督学习,跨数据集表现稳定。
- 适合临床研究者用于少标注场景的疾病预测建模。
阿尔茨海默病是一种进行性神经退行性疾病,导致记忆丧失和认知衰退。尽管深度学习已在该领域广泛应用,但受限于标注数据稀缺、跨数据集泛化能力差,以及对输入扫描数量和时间间隔变化的不灵活。本研究将三种先进的时序自监督学习(SSL)方法适配至3D脑部MRI分析,并引入新扩展以应对变长输入并学习鲁棒的空间特征。我们整合了四个公开数据集,共包含3,161名患者用于预训练,评估模型在诊断分类、转换检测及未来转换预测等多任务上的表现。结果表明,结合时序顺序预测与对比学习的SSL模型,在七项下游任务中有六项优于有监督学习。模型展现出跨任务、跨输入图像数量与时间间隔的良好适应性和泛化能力,证明其在临床应用中的稳健表现。代码与模型已开源:https://github.com/emilykaczmarek/SSL-AD。
原文摘要 · Abstract (English)
Alzheimer's disease is a progressive, neurodegenerative disorder that causes memory loss and cognitive decline. While there has been extensive research in applying deep learning models to Alzheimer's prediction tasks, these models remain limited by lack of available labeled data, poor generalization across datasets, and inflexibility to varying numbers of input scans and time intervals between scans. In this study, we adapt three state-of-the-art temporal self-supervised learning (SSL) approaches for 3D brain MRI analysis, and add novel extensions designed to handle variable-length inputs and learn robust spatial features. We aggregate four publicly available datasets comprising 3,161 patients for pre-training, and show the performance of our model across multiple Alzheimer's prediction tasks including diagnosis classification, conversion detection, and future conversion prediction. Importantly, our SSL model implemented with temporal order prediction and contrastive learning outperforms supervised learning on six out of seven downstream tasks. It demonstrates adaptability and generalizability across tasks and number of input images with varying time intervals, highlighting its capacity for robust performance across clinical applications. We release our code and model publicly at https://github.com/emilykaczmarek/SSL-AD.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。